mirrors/SingleApplication
1
0
Fork 0
mirror of https://github.com/itay-grudev/SingleApplication.git synced 2025-04-21 04:24:43 +08:00
Code Activity

Improved the connectToPrimary() method and fixed issue with instanceStarted() not getting emitted

Browse Source
This commit is contained in:
Itay Grudev 2020-09-09 02:28:07 +01:00
parent eca5803665
commit 3e83f5ce13
3 changed files with 46 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
singleapplication.cpp
View File

@ -85,7 +85,7 @@ SingleApplication::SingleApplication( int &argc, char *argv[], bool allowSeconda
abortSafely(); abortSafely();
} }
} else { } else {
qCritical() << "SingleApplication: Unable create block."; qCritical() << "SingleApplication: Unable to create block.";
abortSafely(); abortSafely();
} }
} }
@ -238,7 +238,8 @@ bool SingleApplication::sendMessage( const QByteArray &message, int timeout )
if( isPrimary() ) return false; if( isPrimary() ) return false;
// Make sure the socket is connected // Make sure the socket is connected
d->connectToPrimary( timeout, SingleApplicationPrivate::Reconnect ); if( ! d->connectToPrimary( timeout, SingleApplicationPrivate::Reconnect ) )
return false;
d->socket->write( message ); d->socket->write( message );
bool dataWritten = d->socket->waitForBytesWritten( timeout ); bool dataWritten = d->socket->waitForBytesWritten( timeout );
@ -254,7 +255,7 @@ void SingleApplication::abortSafely()
{ {
Q_D( SingleApplication ); Q_D( SingleApplication );
qCritical() << d->memory->errorString(); qCritical() << "SingleApplication: " << d->memory->error() << d->memory->errorString();
delete d; delete d;
::exit( EXIT_FAILURE ); ::exit( EXIT_FAILURE );
} }

37
singleapplication_p.cpp
View File

@ -208,32 +208,40 @@ void SingleApplicationPrivate::startSecondary()
instanceNumber = inst->secondary; instanceNumber = inst->secondary;
} }
void SingleApplicationPrivate::connectToPrimary( int msecs, ConnectionType connectionType ) bool SingleApplicationPrivate::connectToPrimary( int timeout, ConnectionType connectionType )
{ {
QElapsedTimer time;
time.start();
// Connect to the Local Server of the Primary Instance if not already // Connect to the Local Server of the Primary Instance if not already
// connected. // connected.
if( socket == nullptr ){ if( socket == nullptr ){
socket = new QLocalSocket(); socket = new QLocalSocket();
} }
// If already connected - we are done; if( socket->state() == QLocalSocket::ConnectedState ) return true;
if( socket->state() == QLocalSocket::ConnectedState )
return;
// If not connect if( socket->state() != QLocalSocket::ConnectedState ){
if( socket->state() == QLocalSocket::UnconnectedState ||
socket->state() == QLocalSocket::ClosingState ){ while( true ){
randomSleep();
if( socket->state() != QLocalSocket::ConnectingState )
socket->connectToServer( blockServerName ); socket->connectToServer( blockServerName );
if( socket->state() == QLocalSocket::ConnectingState ){
socket->waitForConnected( timeout - time.elapsed() );
} }
// Wait for being connected // If connected break out of the loop
if( socket->state() == QLocalSocket::ConnectingState ){ if( socket->state() == QLocalSocket::ConnectedState ) break;
socket->waitForConnected( msecs );
// If elapsed time since start is longer than the method timeout return
if( time.elapsed() >= timeout ) return false;
}
} }
// Initialisation message according to the SingleApplication protocol // Initialisation message according to the SingleApplication protocol
if( socket->state() == QLocalSocket::ConnectedState ){
// Notify the parent that a new instance had been started;
QByteArray initMsg; QByteArray initMsg;
QDataStream writeStream(&initMsg, QIODevice::WriteOnly); QDataStream writeStream(&initMsg, QIODevice::WriteOnly);
@ -259,8 +267,9 @@ void SingleApplicationPrivate::connectToPrimary( int msecs, ConnectionType conne
socket->write( header ); socket->write( header );
socket->write( initMsg ); socket->write( initMsg );
socket->flush(); socket->flush();
socket->waitForBytesWritten( msecs ); if( socket->waitForBytesWritten( timeout - time.elapsed() )) return true;
}
return false;
} }
quint16 SingleApplicationPrivate::blockChecksum() quint16 SingleApplicationPrivate::blockChecksum()

2
singleapplication_p.h
View File

@ -75,7 +75,7 @@ public:
void initializeMemoryBlock(); void initializeMemoryBlock();
void startPrimary(); void startPrimary();
void startSecondary(); void startSecondary();
void connectToPrimary(int msecs, ConnectionType connectionType ); bool connectToPrimary(int msecs, ConnectionType connectionType );
quint16 blockChecksum(); quint16 blockChecksum();
qint64 primaryPid(); qint64 primaryPid();
QString primaryUser(); QString primaryUser();