/// <summary>
/// Called when a new client is connected via TMQ protocol
/// </summary>
public async Task <TmqServerSocket> Connected(ITwinoServer server, IConnectionInfo connection, ConnectionData data)
{
string clientId;
bool found = data.Properties.TryGetValue(TmqHeaders.CLIENT_ID, out clientId);
if (!found || string.IsNullOrEmpty(clientId))
{
clientId = _server.ClientIdGenerator.Create();
}
//if another client with same unique id is online, do not accept new client
MqClient foundClient = _server.FindClient(clientId);
if (foundClient != null)
{
await connection.Socket.SendAsync(await _writer.Create(MessageBuilder.Busy()));
return(null);
}
//creates new mq client object
MqClient client = new MqClient(_server, connection, _server.MessageIdGenerator, _server.Options.UseMessageId);
client.Data = data;
client.UniqueId = clientId.Trim();
client.Token = data.Properties.GetStringValue(TmqHeaders.CLIENT_TOKEN);
client.Name = data.Properties.GetStringValue(TmqHeaders.CLIENT_NAME);
client.Type = data.Properties.GetStringValue(TmqHeaders.CLIENT_TYPE);
//connecting client is a MQ server
string serverValue = data.Properties.GetStringValue(TmqHeaders.TWINO_MQ_SERVER);
if (!string.IsNullOrEmpty(serverValue) && (serverValue.Equals("1") || serverValue.Equals("true", StringComparison.InvariantCultureIgnoreCase)))
{
if (_server.ServerAuthenticator == null)
{
return(null);
}
bool accepted = await _server.ServerAuthenticator.Authenticate(_server, client);
if (!accepted)
{
return(null);
}
client.IsInstanceServer = true;
_server.SlaveInstances.Add(new SlaveInstance
{
ConnectedDate = DateTime.UtcNow,
Client = client,
RemoteHost = client.Info.Client.Client.RemoteEndPoint.ToString().Split(':')[0]
});
await client.SendAsync(MessageBuilder.Accepted(client.UniqueId));
}
//connecting client is a producer/consumer client
else
{
//authenticates client
if (_server.Authenticator != null)
{
client.IsAuthenticated = await _server.Authenticator.Authenticate(_server, client);
if (!client.IsAuthenticated)
{
await client.SendAsync(MessageBuilder.Unauthorized());
return(null);
}
}
//client authenticated, add it into the connected clients list
_server.AddClient(client);
//send response message to the client, client should check unique id,
//if client's unique id isn't permitted, server will create new id for client and send it as response
await client.SendAsync(MessageBuilder.Accepted(client.UniqueId));
if (_server.ClientHandler != null)
{
await _server.ClientHandler.Connected(_server, client);
}
}
return(client);
}
/// <summary>
/// Called when a new client is connected via TMQ protocol
/// </summary>
public async Task <TmqServerSocket> Connected(ITwinoServer server, IConnectionInfo connection, ConnectionData data)
{
string clientId;
bool found = data.Properties.TryGetValue(TmqHeaders.CLIENT_ID, out clientId);
if (!found || string.IsNullOrEmpty(clientId))
{
clientId = _server.ClientIdGenerator.Create();
}
//if another client with same unique id is online, do not accept new client
MqClient foundClient = _server.FindClient(clientId);
if (foundClient != null)
{
await connection.Socket.SendAsync(await _writer.Create(MessageBuilder.Busy()));
return(null);
}
if (_server.Options.ClientLimit > 0 && _server.GetOnlineClients() >= _server.Options.ClientLimit)
{
return(null);
}
//creates new mq client object
MqClient client = new MqClient(_server, connection, _server.MessageIdGenerator, _server.Options.UseMessageId);
client.Data = data;
client.UniqueId = clientId.Trim();
client.Token = data.Properties.GetStringValue(TmqHeaders.CLIENT_TOKEN);
client.Name = data.Properties.GetStringValue(TmqHeaders.CLIENT_NAME);
client.Type = data.Properties.GetStringValue(TmqHeaders.CLIENT_TYPE);
//authenticates client
if (_server.Authenticator != null)
{
client.IsAuthenticated = await _server.Authenticator.Authenticate(_server, client);
if (!client.IsAuthenticated)
{
await client.SendAsync(MessageBuilder.Unauthorized());
return(null);
}
}
//client authenticated, add it into the connected clients list
_server.AddClient(client);
//send response message to the client, client should check unique id,
//if client's unique id isn't permitted, server will create new id for client and send it as response
await client.SendAsync(MessageBuilder.Accepted(client.UniqueId));
if (_server.ClientHandler != null)
{
await _server.ClientHandler.Connected(_server, client);
}
return(client);
}
