/// <summary>
/// Constructor
/// </summary>
/// <param name="commLayer">Communication layer to use (TCP)</param>
/// <param name="settings">Broker settings</param>
public MqttBroker(IMqttCommunicationLayer commLayer, MqttSettings settings)
{
// MQTT broker settings
this.settings = settings;
// MQTT communication layer
this.commLayer = commLayer;
this.commLayer.ClientConnected += commLayer_ClientConnected;
// create managers (publisher, subscriber, session and UAC)
this.subscriberManager = new MqttSubscriberManager();
this.sessionManager = new MqttSessionManager();
this.publisherManager = new MqttPublisherManager(this.subscriberManager, this.sessionManager);
this.uacManager = new MqttUacManager();
this.PreClients = new HashSetTS <MqttClient>();
this.Clients = new DictionaryTS <string, MqttClient>();
}
/// <summary>
/// Stop broker
/// </summary>
public void Stop()
{
running = false;
this.commLayer.Stop();
this.publisherManager.Stop();
// close connection with all clients, in transit ones..
foreach (var client in this.PreClients)
{
client.Close();
}
PreClients.Clear();
PreClients = null;
//and "real" ones
foreach (var client in this.Clients.Values)
{
client.Close();
}
Clients.Clear();
Clients = null;
}
//------------------------------------------------------------------------------------------------------------------------
public GenericRsp HandleGraphDeploymentReq(GraphDeploymentReq req, bool SupressSave)
{
lock (locker)
{
var res = new GenericRsp();
try
{
//get graphkey
var graphkey = (GraphKey)req.GraphKey;
if (graphkey.IsInvalid)
{
res.IsSuccess = false;
res.Message = "Invalid GraphKey";
return res;
}
if (graphkey.NodeId != this.Node.NodeKey.NodeID)
{
res.IsSuccess = false;
res.Message = "Invalid NodeID in GraphKey";
return res;
}
//collect sets
var sets = Node.BeginActiveThingsUpdate();
//deploy or undeploy?
if (req.IsDeployed)
{
//deserialize graph descriptor
GraphDescriptor graphDescriptor;
try { graphDescriptor = GraphBuilder.GetGraphDescriptorFromJson(req.GraphDescriptor, false); }
catch (Exception ex)
{
DebugEx.Assert(ex, "Could not deserialize graph descriptor");
res.IsSuccess = false;
res.Message = "Could not deserialize graph descriptor";
return res;
}
graphDescriptor.GraphKey = graphkey;
//Build Graph
Graph graph;
try { graph = BuildGraph(graphDescriptor); }
catch (Exception ex)
{
DebugEx.Assert(ex, "Could not build graph from graph descriptor (unhandled exception)");
res.IsSuccess = false;
res.Message = "Could not build graph from graph descriptor (unhandled exception)";
return res;
}
if (graph == null)
{
DebugEx.Assert("Could not build graph from graph descriptor");
res.IsSuccess = false;
res.Message = "Could not build graph from graph descriptor";
return res;
}
//set key
graph.GraphKey = graphkey;
//do i already have it?
if (Graphs.ContainsKey(graphkey))
{
//invalidate graph
_GraphManager.InvalidateGraph(graphkey);
//remove information
Graphs.Remove(graphkey);
}
//try deploy graph
try
{
Exception exception;
var depres = graph.OnDeploy(true, null, out exception);
if (!depres)
{
res.IsSuccess = false;
res.Message = "Graph OnDeploy() failed. Message : " + (exception?.Message ?? "null");
return res;
}
}
catch (Exception ex)
{
DebugEx.Assert(ex, "Graph OnDeploy() failed");
res.IsSuccess = false;
res.Message = "Graph OnDeploy() failed. Message : " + ex.Message;
return res;
}
//add to lookup
var gi = new GraphInfo()
{
GraphKey = graphkey,
GraphDescriptor = graphDescriptor,
GraphDescriptorString = req.GraphDescriptor,
Graph = graph,
};
Graphs.Add(graphkey, gi);
//save!
if (IsInitialized && !SupressSave)
Save();
//associate block keys
foreach (var thingblock in graph.Blocks.OfType<BaseThings>())
{
//Add to Thing2Block set
{
//find block key set
var set = ThingKey2BlockKey.TryGetOrDefault(thingblock.ThingKey);
if (set == null)
{
set = new HashSetTS<BlockKey>();
ThingKey2BlockKey.Add(thingblock.ThingKey, set);
}
//add to loockup set
set.Add(thingblock.BlockKey);
}
//Add only for thingIn
if (thingblock.IsThingIn)
{
//find block key set
var set = ThingKey2ThingInBlockKey.TryGetOrDefault(thingblock.ThingKey);
if (set == null)
{
set = new HashSetTS<BlockKey>();
ThingKey2ThingInBlockKey.Add(thingblock.ThingKey, set);
}
//add to loockup set
set.Add(thingblock.BlockKey);
}
}
//Handle Deploy
res.IsSuccess = true;
res.Message = "Graph Deployed Successfully";
}
else
{
//Handle UnDeploy
if (Graphs.ContainsKey(graphkey) == false)
{
res.IsSuccess = true;
res.Message = "Graph Undeployed Successfully (was not deployed)";
}
else
{
//get graph
var gi = Graphs[graphkey];
var graph = gi.Graph;
//inform graph
Exception exception;
try
{
if (graph.OnUndeploy(null, out exception) == false)
DebugEx.Assert(exception, "Graph OnUndeploy failed");
}
catch (Exception ex) { DebugEx.Assert(ex, "Graph OnUndeploy failed"); }
//invalidate graph
_GraphManager.InvalidateGraph(graphkey);
//remove information
Graphs.Remove(graphkey);
//save!
if (IsInitialized && !SupressSave)
Save();
//disassociate block keys
if (graph != null)
foreach (var thingblock in gi.Graph.Blocks.OfType<BaseThings>())
{
//remove from thing2block
{
var set = ThingKey2BlockKey.TryGetOrDefault(thingblock.ThingKey);
if (set != null)
set.Remove(thingblock.BlockKey);
}
//remove from thignIn2block
if (thingblock.IsThingIn)
{
var set = ThingKey2ThingInBlockKey.TryGetOrDefault(thingblock.ThingKey);
if (set != null)
set.Remove(thingblock.BlockKey);
}
}
//done
res.IsSuccess = true;
res.Message = "Graph Undeployed Successfully";
}
}
//finish update
Node.EndActiveThingsUpdate(sets);
}
catch (Exception ex)
{
res.IsSuccess = false;
res.Message = "Unhandled exception in GraphDeploymentReq(). Message=" + ex.Message;
}
finally
{
//begin activation state snapshot
var sets = Node.BeginActiveThingsUpdate();
//update active ports/things
var activeThings = ThingKey2BlockKey.Where(kv => kv.Value.Count > 0)
.Select(kv => Node.Things.TryGetOrDefaultReadOnly(kv.Key))
.WhereNotNull().ToHashSetTS();
var activeThingKeys = activeThings.Select(t => (ThingKey)t.ThingKey).ToHashSetTS();
var activePorts = activeThings.SelectMany(t => t.Ports).ToHashSetTS();
var activePortsKeys = activePorts.Select(p => (PortKey)p.PortKey).ToHashSetTS();
//update sets
Interlocked.Exchange(ref _ActiveThings, activeThings);
Interlocked.Exchange(ref _ActivePorts, activePorts);
Interlocked.Exchange(ref _ActiveThingKeys, activeThingKeys);
Interlocked.Exchange(ref _ActivePortKeys, activePortsKeys);
//trigger node thing activation update
Node.EndActiveThingsUpdate(sets);
}
//return result msg
return res;
}
}
//------------------------------------------------------------------------------------------------------------------------
public GenericRsp HandleGraphDeploymentReq(GraphDeploymentReq req, bool SupressSave)
{
lock (locker)
{
var res = new GenericRsp();
try
{
//get graphkey
var graphkey = (GraphKey)req.GraphKey;
if (graphkey.IsInvalid)
{
res.IsSuccess = false;
res.Message = "Invalid GraphKey";
return(res);
}
if (graphkey.NodeId != this.Node.NodeKey.NodeID)
{
res.IsSuccess = false;
res.Message = "Invalid NodeID in GraphKey";
return(res);
}
//collect sets
var sets = Node.BeginActiveThingsUpdate();
//deploy or undeploy?
if (req.IsDeployed)
{
//deserialize graph descriptor
GraphDescriptor graphDescriptor;
try { graphDescriptor = GraphBuilder.GetGraphDescriptorFromJson(req.GraphDescriptor, false); }
catch (Exception ex)
{
DebugEx.Assert(ex, "Could not deserialize graph descriptor");
res.IsSuccess = false;
res.Message = "Could not deserialize graph descriptor";
return(res);
}
graphDescriptor.GraphKey = graphkey;
//Build Graph
Graph graph;
try { graph = BuildGraph(graphDescriptor); }
catch (Exception ex)
{
DebugEx.Assert(ex, "Could not build graph from graph descriptor (unhandled exception)");
res.IsSuccess = false;
res.Message = "Could not build graph from graph descriptor (unhandled exception)";
return(res);
}
if (graph == null)
{
DebugEx.Assert("Could not build graph from graph descriptor");
res.IsSuccess = false;
res.Message = "Could not build graph from graph descriptor";
return(res);
}
//set key
graph.GraphKey = graphkey;
//do i already have it?
if (Graphs.ContainsKey(graphkey))
{
//invalidate graph
_GraphManager.InvalidateGraph(graphkey);
//remove information
Graphs.Remove(graphkey);
}
//try deploy graph
try
{
Exception exception;
var depres = graph.OnDeploy(true, null, out exception);
if (!depres)
{
res.IsSuccess = false;
res.Message = "Graph OnDeploy() failed. Message : " + (exception?.Message ?? "null");
return(res);
}
}
catch (Exception ex)
{
DebugEx.Assert(ex, "Graph OnDeploy() failed");
res.IsSuccess = false;
res.Message = "Graph OnDeploy() failed. Message : " + ex.Message;
return(res);
}
//add to lookup
var gi = new GraphInfo()
{
GraphKey = graphkey,
GraphDescriptor = graphDescriptor,
GraphDescriptorString = req.GraphDescriptor,
Graph = graph,
};
Graphs.Add(graphkey, gi);
//save!
if (IsInitialized && !SupressSave)
{
Save();
}
//associate block keys
foreach (var thingblock in graph.Blocks.OfType <BaseThings>())
{
//Add to Thing2Block set
{
//find block key set
var set = ThingKey2BlockKey.TryGetOrDefault(thingblock.ThingKey);
if (set == null)
{
set = new HashSetTS <BlockKey>();
ThingKey2BlockKey.Add(thingblock.ThingKey, set);
}
//add to loockup set
set.Add(thingblock.BlockKey);
}
//Add only for thingIn
if (thingblock.IsThingIn)
{
//find block key set
var set = ThingKey2ThingInBlockKey.TryGetOrDefault(thingblock.ThingKey);
if (set == null)
{
set = new HashSetTS <BlockKey>();
ThingKey2ThingInBlockKey.Add(thingblock.ThingKey, set);
}
//add to loockup set
set.Add(thingblock.BlockKey);
}
}
//Handle Deploy
res.IsSuccess = true;
res.Message = "Graph Deployed Successfully";
}
else
{
//Handle UnDeploy
if (Graphs.ContainsKey(graphkey) == false)
{
res.IsSuccess = true;
res.Message = "Graph Undeployed Successfully (was not deployed)";
}
else
{
//get graph
var gi = Graphs[graphkey];
var graph = gi.Graph;
//inform graph
Exception exception;
try
{
if (graph.OnUndeploy(null, out exception) == false)
{
DebugEx.Assert(exception, "Graph OnUndeploy failed");
}
}
catch (Exception ex) { DebugEx.Assert(ex, "Graph OnUndeploy failed"); }
//invalidate graph
_GraphManager.InvalidateGraph(graphkey);
//remove information
Graphs.Remove(graphkey);
//save!
if (IsInitialized && !SupressSave)
{
Save();
}
//disassociate block keys
if (graph != null)
{
foreach (var thingblock in gi.Graph.Blocks.OfType <BaseThings>())
{
//remove from thing2block
{
var set = ThingKey2BlockKey.TryGetOrDefault(thingblock.ThingKey);
if (set != null)
{
set.Remove(thingblock.BlockKey);
}
}
//remove from thignIn2block
if (thingblock.IsThingIn)
{
var set = ThingKey2ThingInBlockKey.TryGetOrDefault(thingblock.ThingKey);
if (set != null)
{
set.Remove(thingblock.BlockKey);
}
}
}
}
//done
res.IsSuccess = true;
res.Message = "Graph Undeployed Successfully";
}
}
//finish update
Node.EndActiveThingsUpdate(sets);
}
catch (Exception ex)
{
res.IsSuccess = false;
res.Message = "Unhandled exception in GraphDeploymentReq(). Message=" + ex.Message;
}
finally
{
//begin activation state snapshot
var sets = Node.BeginActiveThingsUpdate();
//update active ports/things
var activeThings = ThingKey2BlockKey.Where(kv => kv.Value.Count > 0)
.Select(kv => Node.Things.TryGetOrDefaultReadOnly(kv.Key))
.WhereNotNull().ToHashSetTS();
var activeThingKeys = activeThings.Select(t => (ThingKey)t.ThingKey).ToHashSetTS();
var activePorts = activeThings.SelectMany(t => t.Ports).ToHashSetTS();
var activePortsKeys = activePorts.Select(p => (PortKey)p.PortKey).ToHashSetTS();
//update sets
Interlocked.Exchange(ref _ActiveThings, activeThings);
Interlocked.Exchange(ref _ActivePorts, activePorts);
Interlocked.Exchange(ref _ActiveThingKeys, activeThingKeys);
Interlocked.Exchange(ref _ActivePortKeys, activePortsKeys);
//trigger node thing activation update
Node.EndActiveThingsUpdate(sets);
}
//return result msg
return(res);
}
}
/// <summary>
/// Stop broker
/// </summary>
public void Stop()
{
running = false;
this.commLayer.Stop();
this.publisherManager.Stop();
// close connection with all clients, in transit ones..
foreach (var client in this.PreClients)
client.Close();
PreClients.Clear();
PreClients = null;
//and "real" ones
foreach (var client in this.Clients.Values)
client.Close();
Clients.Clear();
Clients = null;
}
/// <summary>
/// Constructor
/// </summary>
/// <param name="commLayer">Communication layer to use (TCP)</param>
/// <param name="settings">Broker settings</param>
public MqttBroker(IMqttCommunicationLayer commLayer, MqttSettings settings)
{
// MQTT broker settings
this.settings = settings;
// MQTT communication layer
this.commLayer = commLayer;
this.commLayer.ClientConnected += commLayer_ClientConnected;
// create managers (publisher, subscriber, session and UAC)
this.subscriberManager = new MqttSubscriberManager();
this.sessionManager = new MqttSessionManager();
this.publisherManager = new MqttPublisherManager(this.subscriberManager, this.sessionManager);
this.uacManager = new MqttUacManager();
this.PreClients = new HashSetTS<MqttClient>();
this.Clients = new DictionaryTS<string, MqttClient>();
}
