Files
o3de/Gems/Atom/Tools/AtomToolsFramework/Code/Source/Communication/LocalServer.cpp
T
guthadam fae33e9235 ATOM-15439 Implement basic local socket and server for IPC in material editor and other tools
This replaces grid hub usage in the material editor. It allows material editor and other tools to intercommunicate on the local host.  This will allow enforcing that there is only one instance of the material editor running.  Opening a second instance will forward command line options to the first instance running a local server.

https://jira.agscollab.com/browse/ATOM-15439
https://jira.agscollab.com/browse/ATOM-13742
2021-05-02 17:08:21 -05:00

109 lines
3.7 KiB
C++

/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
#include <AtomToolsFramework/Communication/LocalServer.h>
#include <AzCore/Debug/Trace.h>
#include <QString>
namespace AtomToolsFramework
{
LocalServer::LocalServer()
{
AZ_TracePrintf("AtomToolsFramework::LocalServer", "Creating local server\n");
m_server.setSocketOptions(QLocalServer::WorldAccessOption);
QObject::connect(&m_server, &QLocalServer::newConnection, this, [this]() { AddConnection(m_server.nextPendingConnection()); });
}
LocalServer::~LocalServer()
{
Disconnect();
}
bool LocalServer::Connect(const QString& serverName)
{
Disconnect();
m_serverName = serverName;
AZ_TracePrintf("AtomToolsFramework::LocalServer", "Starting: %s\n", m_serverName.toUtf8().constData());
if (m_server.listen(m_serverName))
{
AZ_TracePrintf("AtomToolsFramework::LocalServer", "Started: %s\n", m_serverName.toUtf8().constData());
return true;
}
if (m_server.serverError() == QAbstractSocket::AddressInUseError)
{
AZ_TracePrintf("AtomToolsFramework::LocalServer", "Restarting: %s\n", m_serverName.toUtf8().constData());
Disconnect();
AZ_TracePrintf("AtomToolsFramework::LocalServer", "Starting: %s\n", m_serverName.toUtf8().constData());
if (m_server.listen(m_serverName))
{
return true;
}
}
AZ_TracePrintf("AtomToolsFramework::LocalServer", "Starting failed: %s\n", m_serverName.toUtf8().constData());
Disconnect();
return false;
}
void LocalServer::Disconnect()
{
AZ_TracePrintf("AtomToolsFramework::LocalServer", "Disconnecting: %s\n", m_serverName.toUtf8().constData());
QLocalServer::removeServer(m_serverName);
}
bool LocalServer::IsConnected() const
{
return m_server.isListening();
}
void LocalServer::SetReadHandler(LocalServer::ReadHandler handler)
{
m_readHandler = handler;
}
void LocalServer::AddConnection(QLocalSocket* connection)
{
AZ_TracePrintf("AtomToolsFramework::LocalServer", "Connection added: %s\n", m_serverName.toUtf8().constData());
QObject::connect(connection, &QLocalSocket::readyRead, this, [this, connection]() { ReadFromConnection(connection); });
QObject::connect(connection, &QLocalSocket::disconnected, this, [this, connection]() { DeleteConnection(connection); });
}
void LocalServer::ReadFromConnection(QLocalSocket* connection)
{
if (connection)
{
AZ_TracePrintf("AtomToolsFramework::LocalServer", "Data received: %s\n", m_serverName.toUtf8().constData());
QByteArray buffer = connection->readAll();
if (m_readHandler)
{
m_readHandler(buffer);
}
}
}
void LocalServer::DeleteConnection(QLocalSocket* connection)
{
if (connection)
{
AZ_TracePrintf("AtomToolsFramework::LocalServer", "Deleting connection: %s\n", m_serverName.toUtf8().constData());
connection->deleteLater();
}
}
} // namespace AtomToolsFramework
#include <AtomToolsFramework/Communication/moc_LocalServer.cpp>