public async override Task ReceiveAsync(WebSocket socket, WebSocketReceiveResult result, byte[] buffer)
{
var reader = new StreamReader(new MemoryStream(buffer), Encoding.Default);
var msg = new JsonSerializer().Deserialize <GunMessage>(new JsonTextReader(reader));
if (_duplicateManager.Check(msg.Key))
{
return;
}
_duplicateManager.Track(msg.Key);
if (msg is PutMessage)
{
var change = _graph.Mix(((PutMessage)msg).PutChanges);
}
if (msg is GetMessage)
{
var ack = _graph.Get(((GetMessage)msg).Get);
if (ack != null)
{
var response = new PutMessage()
{
Key = _duplicateManager.Track(DuplicateManager.Random()),
At = msg.Key,
PutChanges = ack
};
await this.SendMessageToAllAsync(JsonConvert.SerializeObject(response));
}
}
await this.SendMessageToAllAsync(JsonConvert.SerializeObject(msg));
}
public void AddFile(PutMessage file)
{
if (_db.Files.Any(a => a.TargetName == file.TargetName))
{
throw new FileExistsException(file.TargetName);
}
_db.Files.Add(file);
_db.SaveChanges();
}
public void Start()
{
var returnedData = NodeCommunicationClient.Send(
_configuration.ConfigurationToolAddress,
_configuration.User,
_configuration.Password,
new SingleValueMessage {
Value = CONFIG
},
Command.REQ,
nameof(ConfigurationTool));
var name = MessagePackSerializer.Deserialize <SingleValueMessage>(returnedData.Content).Value;
name = name.Substring(0, name.Length - 4);
while (true)
{
Console.Write($"{name}> ");
var command = Console.ReadLine();
var msgCommand = Command.UNKNOWN;
IMessage content = null;
//Local commands
if (clear.IsMatch(command))
{
Console.Clear();
continue;
}
//Add user
if (addUser.IsMatch(command))
{
var groups = addUser.Match(command).Groups.Select(a => a.Value).ToList();
msgCommand = Command.CONF_ADD_USER;
var addUserMsg = new AddUserMessage
{
Level = int.Parse(groups[3]),
Username = groups[1],
Password = groups[2]
};
content = addUserMsg;
}
//Set user level
else if (setUserLevel.IsMatch(command))
{
var groups = setUserLevel.Match(command).Groups.Select(a => a.Value).ToList();
msgCommand = Command.CONF_SET_USER_LEVEL;
var setUserLevelMsg = new SetUserLevelMessage
{
User = groups[1],
NewLevel = int.Parse(groups[2])
};
content = setUserLevelMsg;
}
//Remove user
else if (removeUser.IsMatch(command))
{
var userName = removeUser.Match(command).Groups[1].Value;
msgCommand = Command.CONF_REMOVE_USER;
var removeUserMsg = new SingleValueMessage {
Value = userName
};
content = removeUserMsg;
}
//Upload file
else if (putProgram.IsMatch(command) || putPipeline.IsMatch(command))
{
var isProgram = putProgram.IsMatch(command);
var groups = isProgram
? putProgram.Match(command).Groups.Select(a => a.Value).ToList()
: putPipeline.Match(command).Groups.Select(a => a.Value).ToList();
msgCommand = isProgram ? Command.CONF_PUT_PROGRAM : Command.CONF_PUT_PIPELINE;
var putMsg = new PutMessage
{
TargetName = groups[2],
TargetType = isProgram ? TargetTypes.PROGRAM : TargetTypes.PIPELINE,
Content = File.ReadAllBytes(groups[1]),
Runtime = groups[3]
};
content = putMsg;
}
//Distribute program/pipeline to child
else if (distribute.IsMatch(command))
{
var groups = distribute.Match(command).Groups.Select(a => a.Value).ToList();
msgCommand = Command.CONF_DISTRIBUTE;
var distributeMsg = new DistributeMessage
{
File = groups[1],
Target = groups[2]
};
content = distributeMsg;
}
//Get status
else if (getStatus.IsMatch(command))
{
var groups = getStatus.Match(command).Groups.Select(a => a.Value).ToList();
msgCommand = Command.CONF_GET_STATUS;
var statusTarget = StatusTarget.All;
switch (groups[1].First().ToString().ToUpper())
{
case "A":
statusTarget = StatusTarget.All;
break;
case "N":
statusTarget = StatusTarget.Nodes;
break;
case "S":
statusTarget = StatusTarget.Self;
break;
}
var getStatusMsg = new GetStatusMessage
{
Target = statusTarget,
StatusInfo = new List <Status>()
};
content = getStatusMsg;
}
//Ok?
else if (ok.IsMatch(command))
{
msgCommand = Command.OK;
var okMsg = new EmptyMessage();
content = okMsg;
}
//Fail!
else if (fail.IsMatch(command))
{
msgCommand = Command.FAIL;
var failMsg = new EmptyMessage();
content = failMsg;
}
if (msgCommand != Command.UNKNOWN)
{
var response = NodeCommunicationClient.Send(
_configuration.ConfigurationToolAddress,
_configuration.User,
_configuration.Password,
content,
msgCommand,
CONFIG);
if (response.Message.Command == Command.OK)
{
switch (msgCommand)
{
case Command.CONF_GET_STATUS:
var statusMessage =
MessagePackSerializer.Deserialize <GetStatusMessage>(response.Content);
Console.WriteLine();
statusMessage.StatusInfo.ForEach(stat =>
{
Console.WriteLine($"{stat.Name}");
Console.WriteLine($"{stat.Name} - " + (stat.Up ? "UP" : "DOWN") + "\n",
stat.Up ? Log.Green : Log.Red);
});
break;
case Command.OK:
Console.WriteLine("Ok!", Log.Green);
break;
case Command.FAIL:
Console.WriteLine(":(", Log.Red);
break;
}
}
else
{
Console.WriteLine(
$"Execution at node failed!:\n{MessagePackSerializer.Deserialize<SingleValueMessage>(response.Content).Value}",
Log.Red);
}
}
else
{
Console.WriteLine($"Unknown command: {command}!", Log.Red);
}
}
}
