[13538] | 1 | using System;
|
---|
[13524] | 2 | using System.Collections.Concurrent;
|
---|
| 3 | using System.Collections.Generic;
|
---|
[13541] | 4 | using System.Linq;
|
---|
[13524] | 5 | using System.Net;
|
---|
[13538] | 6 | using DistributedGA.Core.Domain;
|
---|
| 7 | using DistributedGA.Core.Interface;
|
---|
[13937] | 8 | using DistributedGA.Core.Util;
|
---|
[13524] | 9 |
|
---|
[13924] | 10 | namespace DistributedGA.Core.Implementation {
|
---|
| 11 | public class PeerNetworkMessageHandler : IMessageHandler {
|
---|
| 12 | //own peer-instance Information
|
---|
| 13 | private PeerInfo ownInstance = null;
|
---|
[13524] | 14 |
|
---|
[13924] | 15 | //uses peer-list from IPeerListManager to decide which peers to contact
|
---|
| 16 | private IPeerListManager peerListManager;
|
---|
[13524] | 17 |
|
---|
[13924] | 18 | //uses IMessageSender to send populations to peers
|
---|
| 19 | private IMessageSender sender = null;
|
---|
[13524] | 20 |
|
---|
[13924] | 21 | //provides current population for the higher layer IMigrationOperator
|
---|
| 22 | //to queues are used to gather and and provide population more efficiently
|
---|
| 23 | private object activeQueueLocker = new object();
|
---|
[13937] | 24 | private SizedConcurrentQueue<byte[]> writeQueue;
|
---|
| 25 | private SizedConcurrentQueue<byte[]> readQueue;
|
---|
[13524] | 26 |
|
---|
[13924] | 27 | //uses IMessageService for recieving population from one peer at once
|
---|
| 28 | private IMessageService host = null;
|
---|
[13524] | 29 |
|
---|
[13541] | 30 |
|
---|
| 31 |
|
---|
[13924] | 32 | public void Init(string lanIpPrefix, string contactServerUrl) {
|
---|
| 33 | try {
|
---|
| 34 | ownInstance = new PeerInfo() {
|
---|
| 35 | IpAddress = GetInternalIpAddress(lanIpPrefix),
|
---|
| 36 | Port = 0,
|
---|
| 37 | ProblemInstance = "TestProblem"
|
---|
| 38 | }; // TODO: get own peerinfo
|
---|
[13524] | 39 |
|
---|
[13937] | 40 | writeQueue = new SizedConcurrentQueue<byte[]>();
|
---|
| 41 | readQueue = new SizedConcurrentQueue<byte[]>();
|
---|
| 42 | writeQueue.Limit = 1000;
|
---|
| 43 | readQueue.Limit = writeQueue.Limit;
|
---|
[13524] | 44 |
|
---|
[13924] | 45 | host = new WcfMessageService();
|
---|
| 46 | ownInstance.Port = host.Init(ownInstance.IpAddress); //getting port, on which service is hostet
|
---|
[13937] | 47 | host.OnDataRecieved += new EventHandler<MessageRecieveEventArgs>(OnDataRecieved);
|
---|
[13524] | 48 |
|
---|
[13924] | 49 | peerListManager = new WcfPeerListManager();
|
---|
| 50 | peerListManager.Init(ownInstance, contactServerUrl);
|
---|
[13524] | 51 |
|
---|
[13924] | 52 | sender = new WcfMessageSender();
|
---|
| 53 | sender.Init(ownInstance);
|
---|
[13524] | 54 |
|
---|
[13924] | 55 | }
|
---|
| 56 | catch (Exception ex) {
|
---|
| 57 | AddError("PeerNetworkMessageHandler.Init", ex);
|
---|
| 58 | }
|
---|
| 59 | }
|
---|
[13524] | 60 |
|
---|
[13924] | 61 | public void Dispose() {
|
---|
| 62 | try {
|
---|
| 63 | host.Dispose();
|
---|
| 64 | sender.Dispose();
|
---|
| 65 | peerListManager.Dispose();
|
---|
| 66 | }
|
---|
| 67 | catch (Exception ex) {
|
---|
| 68 | AddError("PeerNetworkMessageHandler.Dispose", ex);
|
---|
| 69 | }
|
---|
| 70 | }
|
---|
[13547] | 71 |
|
---|
[13924] | 72 | public void PublishDataToNetwork(byte[][] data) {
|
---|
| 73 | try {
|
---|
| 74 | foreach (PeerInfo peer in peerListManager.GetPeerList()) {
|
---|
| 75 | //maybe something will go wrong during network communication...
|
---|
| 76 | try {
|
---|
| 77 | sender.SendData(peer, data);
|
---|
| 78 | }
|
---|
| 79 | catch (Exception ex) {
|
---|
| 80 | AddError("PeerNetworkMessageHandler.PublishDataToNetwork(during sending to one peer!)", ex);
|
---|
| 81 | }
|
---|
[13524] | 82 | }
|
---|
[13924] | 83 | }
|
---|
| 84 | catch (Exception ex) {
|
---|
| 85 | AddError("PeerNetworkMessageHandler.PublishDataToNetwork", ex);
|
---|
| 86 | }
|
---|
| 87 | }
|
---|
[13524] | 88 |
|
---|
[13924] | 89 | public byte[][] GetDataFromNetwork() {
|
---|
| 90 | try {
|
---|
| 91 | List<byte[]> res = new List<byte[]>();
|
---|
| 92 | byte[] item = null;
|
---|
| 93 | lock (activeQueueLocker) {
|
---|
| 94 | var tmp = readQueue;
|
---|
| 95 | readQueue = writeQueue;
|
---|
| 96 | writeQueue = tmp;
|
---|
[13524] | 97 | }
|
---|
[13541] | 98 |
|
---|
[13924] | 99 | //creating resultset
|
---|
| 100 | while (!readQueue.IsEmpty) {
|
---|
| 101 | if (readQueue.TryDequeue(out item)) {
|
---|
| 102 | res.Add(item);
|
---|
| 103 | }
|
---|
[13524] | 104 | }
|
---|
[13924] | 105 | return res.ToArray();
|
---|
| 106 | }
|
---|
| 107 | catch (Exception ex) {
|
---|
| 108 | AddError("PeerNetworkMessageHandler.GetDataFromNetwork", ex);
|
---|
| 109 | return null;
|
---|
| 110 | }
|
---|
| 111 | }
|
---|
[13524] | 112 |
|
---|
[13924] | 113 | public PeerInfo GetPeerInfo() {
|
---|
| 114 | return ownInstance;
|
---|
| 115 | }
|
---|
[13524] | 116 |
|
---|
[13924] | 117 | public List<PeerInfo> GetCurrentNetwork() {
|
---|
| 118 | return peerListManager.GetPeerList();
|
---|
| 119 | }
|
---|
[13524] | 120 |
|
---|
[13924] | 121 | private string GetInternalIpAddress(string ipPrefix) {
|
---|
| 122 | try {
|
---|
| 123 | var strHostName = Dns.GetHostName();
|
---|
| 124 | // Then using host name, get the IP address list..
|
---|
| 125 | IPHostEntry ipEntry = Dns.GetHostEntry(strHostName);
|
---|
| 126 | IPAddress[] addr = ipEntry.AddressList;
|
---|
[13541] | 127 |
|
---|
[13924] | 128 | return addr
|
---|
| 129 | .Select(ip => ip.ToString())
|
---|
| 130 | .First(str => str.StartsWith(ipPrefix));
|
---|
| 131 | }
|
---|
| 132 | catch { return null; }
|
---|
| 133 | }
|
---|
| 134 |
|
---|
[13937] | 135 | private void OnDataRecieved(object sender, MessageRecieveEventArgs e) {
|
---|
[13924] | 136 | if (e != null) {
|
---|
| 137 | lock (activeQueueLocker) {
|
---|
| 138 | foreach (byte[] item in e.data) {
|
---|
| 139 | writeQueue.Enqueue(item);
|
---|
| 140 | }
|
---|
[13887] | 141 | }
|
---|
[13924] | 142 | }
|
---|
| 143 | }
|
---|
[13538] | 144 |
|
---|
[13924] | 145 | private void AddError(string source, Exception ex) {
|
---|
| 146 | if (peerListManager != null) {
|
---|
| 147 | try {
|
---|
| 148 | peerListManager.SendLogToServer(string.Concat("Source: ", source, ", Exception: ", ex.Message));
|
---|
[13524] | 149 | }
|
---|
[13924] | 150 | catch { }
|
---|
| 151 | }
|
---|
| 152 | }
|
---|
[13524] | 153 |
|
---|
[13924] | 154 | }
|
---|
[13524] | 155 | }
|
---|