WebSVN – HomeAutomation – Diff – /trunk/PcSoftware/Atom1.5/src/net/Manager.cpp



 
void Manager::SlotOnNewState(ClientId client_id, ServerId server_id, ClientState client_state)
{
    if (client_state == CLIENT_STATE_DISCONNECTED)
    {
        this->mutex_clients_.lock();
        this->clients_.erase(client_id);
        this->mutex_clients_.unlock();
    }
    else if (client_state == CLIENT_STATE_ACCEPTED)
    {
        this->mutex_clients_.lock();
        ClientList::iterator it = this->clients_.find(client_id);
       
        if (it != this->clients_.end())
        {
            TcpClient::Pointer client = TcpClient::Pointer(new TcpClient(this->io_service_, this->GetFreeClientId(), it->second->GetServerId()));

                                 Client::SignalOnNewData::slot_type(&Manager::SlotOnNewData, this, _1, _2, _3).track(Manager::instance_));
           
            client->Accept(boost::polymorphic_downcast<TcpClient*>(it->second.get())->ReleaseAcceptor());
           
            this->clients_[client->GetId()] = client;
           
            this->mutex_clients_.unlock();
           
            this->signal_on_new_state_(client_id, server_id, CLIENT_STATE_CONNECTED);
            return;
        }
        else
        {
            this->mutex_clients_.unlock();
            throw std::runtime_error("Accepted unknown client!");
        }
    }
   
    this->signal_on_new_state_(client_id, server_id, client_state);
}
 

}
 
ServerId Manager::GetFreeServerId()
{
    ServerId server_id = 1;
   
    for (ClientList::iterator it = this->clients_.begin(); it != this->clients_.end(); it++)
    {
        if (it->second->GetServerId() != server_id)
        {
            return server_id;
        }

    if (protocol != PROTOCOL_TCP)
    {
        throw std::runtime_error("Can not start server for the Serial or UDP protocol!");
        return 0;
    }
   
    this->mutex_clients_.lock();
   
    TcpClient::Pointer client = TcpClient::Pointer(new TcpClient(this->io_service_, this->GetFreeClientId(), this->GetFreeServerId()));
   
    client->ConnectSlots(Client::SignalOnNewState::slot_type(&Manager::SlotOnNewState, this, _1, _2, _3).track(Manager::instance_),
                         Client::SignalOnNewData::slot_type(&Manager::SlotOnNewData, this, _1, _2, _3).track(Manager::instance_));
   
    TcpClient::AcceptorPointer acceptor = TcpClient::AcceptorPointer(new boost::asio::ip::tcp::acceptor(this->io_service_, boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v4(), port)));    
 
    client->Accept(acceptor);
   
    this->clients_[client->GetId()] = client;
   
    this->mutex_clients_.unlock();
   
    return client->GetServerId();
}
 
ClientId Manager::Connect(Protocol protocol, std::string address, unsigned int port_or_baud)
{
    Client::Pointer client;
   
    this->mutex_clients_.lock();
   
    switch (protocol)
    {
        case PROTOCOL_TCP:
        {

            client = Client::Pointer(new SerialClient(this->io_service_, this->GetFreeClientId(), 0));
            break;
        }
        default:
        {
            this->mutex_clients_.unlock();
            throw std::runtime_error("Invalid protocol specified!");
            return 0;
        }
    }
   

    {
        client->Connect(address, port_or_baud);
    }
    catch (std::exception e)
    {
        this->mutex_clients_.unlock();
        throw e;
    }
   
    this->clients_[client->GetId()] = client;
   
    this->mutex_clients_.unlock();
   
    return client->GetId();
}
 
void Manager::SendToAll(ServerId server_id, Buffer data)
{
    this->io_service_.post(boost::bind(&Manager::SendToAllHandler, this, server_id, data));
}
 
void Manager::SendTo(ClientId client_id, Buffer data)
{
    this->io_service_.post(boost::bind(&Manager::SendToHandler, this, client_id, data));
}
 
void Manager::StopServer(ServerId server_id)
{
    this->io_service_.post(boost::bind(&Manager::StopServerHandler, this, server_id));
}
 
void Manager::Disconnect(ClientId client_id)
{
    this->io_service_.post(boost::bind(&Manager::DisconnectHandler, this, client_id));
}
 
void Manager::SendToAllHandler(ServerId server_id, Buffer data)
{
    this->mutex_clients_.lock();
   
    for (ClientList::iterator it = this->clients_.begin(); it != this->clients_.end(); it++)
    {
        if (it->second->GetServerId() == server_id)
        {
            it->second->Send(data);
        }
    }
   
    this->mutex_clients_.unlock();
}
 
void Manager::SendToHandler(ClientId client_id, Buffer data)
{
    this->mutex_clients_.lock();
   
    ClientList::iterator it = this->clients_.find(client_id);
   
    if (it != this->clients_.end())
    {
        it->second->Send(data);
    }
   
    this->mutex_clients_.unlock();
}
 
void Manager::StopServerHandler(ServerId server_id)
{
    this->mutex_clients_.lock();
   
    for (ClientList::iterator it = this->clients_.begin(); it != this->clients_.end(); it++)
    {
        if (it->second->GetServerId() == server_id)
        {
            this->mutex_clients_.unlock();
            it->second->Disconnect();
            this->mutex_clients_.lock();
        }
    }
   
    this->mutex_clients_.unlock();
}
 
void Manager::DisconnectHandler(ClientId client_id)
{
    this->mutex_clients_.lock();
   
    ClientList::iterator it = this->clients_.find(client_id);
   
    this->mutex_clients_.unlock();
   
    if (it != this->clients_.end())
    {
        it->second->Disconnect();
    }
   
   
}
 
}; // namespace net
}; // namespace atom

Rev 1594	Rev 1595
Line 61...	Line 61...
61		61
62	void Manager::SlotOnNewState(ClientId client_id, ServerId server_id, ClientState client_state)	62	void Manager::SlotOnNewState(ClientId client_id, ServerId server_id, ClientState client_state)
63	{	63	{
64	if (client_state == CLIENT_STATE_DISCONNECTED)	64	if (client_state == CLIENT_STATE_DISCONNECTED)
65	{	65	{
-		66	this->mutex_clients_.lock();
66	this->clients_.erase(client_id);	67	this->clients_.erase(client_id);
-		68	this->mutex_clients_.unlock();
67	}	69	}
68	else if (client_state == CLIENT_STATE_ACCEPTED)	70	else if (client_state == CLIENT_STATE_ACCEPTED)
69	{	71	{
-		72	this->mutex_clients_.lock();
70	ClientList::iterator it = this->clients_.find(client_id);	73	ClientList::iterator it = this->clients_.find(client_id);
71		74
72	if (it != this->clients_.end())	75	if (it != this->clients_.end())
73	{	76	{
74	TcpClient::Pointer client = TcpClient::Pointer(new TcpClient(this->io_service_, this->GetFreeClientId(), it->second->GetServerId()));	77	TcpClient::Pointer client = TcpClient::Pointer(new TcpClient(this->io_service_, this->GetFreeClientId(), it->second->GetServerId()));
Line 77...	Line 80...
77	Client::SignalOnNewData::slot_type(&Manager::SlotOnNewData, this, _1, _2, _3).track(Manager::instance_));	80	Client::SignalOnNewData::slot_type(&Manager::SlotOnNewData, this, _1, _2, _3).track(Manager::instance_));
78		81
79	client->Accept(boost::polymorphic_downcast<TcpClient*>(it->second.get())->ReleaseAcceptor());	82	client->Accept(boost::polymorphic_downcast<TcpClient*>(it->second.get())->ReleaseAcceptor());
80		83
81	this->clients_[client->GetId()] = client;	84	this->clients_[client->GetId()] = client;
-		85
-		86	this->mutex_clients_.unlock();
82		87
83	this->signal_on_new_state_(client_id, server_id, CLIENT_STATE_CONNECTED);	88	this->signal_on_new_state_(client_id, server_id, CLIENT_STATE_CONNECTED);
84	return;	89	return;
85	}	90	}
86	else	91	else
87	{	92	{
-		93	this->mutex_clients_.unlock();
88	throw std::runtime_error("Accepted unknown client!");	94	throw std::runtime_error("Accepted unknown client!");
89	}	95	}
90	}	96	}
91		97
92	this->signal_on_new_state_(client_id, server_id, client_state);	98	this->signal_on_new_state_(client_id, server_id, client_state);
93	}	99	}
94		100
Line 98...	Line 104...
98	}	104	}
99		105
100	ServerId Manager::GetFreeServerId()	106	ServerId Manager::GetFreeServerId()
101	{	107	{
102	ServerId server_id = 1;	108	ServerId server_id = 1;
103		109
104	for (ClientList::iterator it = this->clients_.begin(); it != this->clients_.end(); it++)	110	for (ClientList::iterator it = this->clients_.begin(); it != this->clients_.end(); it++)
105	{	111	{
106	if (it->second->GetServerId() != server_id)	112	if (it->second->GetServerId() != server_id)
107	{	113	{
108	return server_id;	114	return server_id;
109	}	115	}
Line 131...	Line 137...
131	if (protocol != PROTOCOL_TCP)	137	if (protocol != PROTOCOL_TCP)
132	{	138	{
133	throw std::runtime_error("Can not start server for the Serial or UDP protocol!");	139	throw std::runtime_error("Can not start server for the Serial or UDP protocol!");
134	return 0;	140	return 0;
135	}	141	}
-		142
-		143	this->mutex_clients_.lock();
136		144
137	TcpClient::Pointer client = TcpClient::Pointer(new TcpClient(this->io_service_, this->GetFreeClientId(), this->GetFreeServerId()));	145	TcpClient::Pointer client = TcpClient::Pointer(new TcpClient(this->io_service_, this->GetFreeClientId(), this->GetFreeServerId()));
138		146
139	client->ConnectSlots(Client::SignalOnNewState::slot_type(&Manager::SlotOnNewState, this, _1, _2, _3).track(Manager::instance_),	147	client->ConnectSlots(Client::SignalOnNewState::slot_type(&Manager::SlotOnNewState, this, _1, _2, _3).track(Manager::instance_),
140	Client::SignalOnNewData::slot_type(&Manager::SlotOnNewData, this, _1, _2, _3).track(Manager::instance_));	148	Client::SignalOnNewData::slot_type(&Manager::SlotOnNewData, this, _1, _2, _3).track(Manager::instance_));
141		149
142	TcpClient::AcceptorPointer acceptor = TcpClient::AcceptorPointer(new boost::asio::ip::tcp::acceptor(this->io_service_, boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v4(), port)));	150	TcpClient::AcceptorPointer acceptor = TcpClient::AcceptorPointer(new boost::asio::ip::tcp::acceptor(this->io_service_, boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v4(), port)));
143		151
144	client->Accept(acceptor);	152	client->Accept(acceptor);
145		153
146	this->clients_[client->GetId()] = client;	154	this->clients_[client->GetId()] = client;
-		155
-		156	this->mutex_clients_.unlock();
147		157
148	return client->GetServerId();	158	return client->GetServerId();
149	}	159	}
150		160
151	ClientId Manager::Connect(Protocol protocol, std::string address, unsigned int port_or_baud)	161	ClientId Manager::Connect(Protocol protocol, std::string address, unsigned int port_or_baud)
152	{	162	{
153	Client::Pointer client;	163	Client::Pointer client;
-		164
-		165	this->mutex_clients_.lock();
154		166
155	switch (protocol)	167	switch (protocol)
156	{	168	{
157	case PROTOCOL_TCP:	169	case PROTOCOL_TCP:
158	{	170	{
Line 169...	Line 181...
169	client = Client::Pointer(new SerialClient(this->io_service_, this->GetFreeClientId(), 0));	181	client = Client::Pointer(new SerialClient(this->io_service_, this->GetFreeClientId(), 0));
170	break;	182	break;
171	}	183	}
172	default:	184	default:
173	{	185	{
-		186	this->mutex_clients_.unlock();
174	throw std::runtime_error("Invalid protocol specified!");	187	throw std::runtime_error("Invalid protocol specified!");
175	return 0;	188	return 0;
176	}	189	}
177	}	190	}
178		191
Line 183...	Line 196...
183	{	196	{
184	client->Connect(address, port_or_baud);	197	client->Connect(address, port_or_baud);
185	}	198	}
186	catch (std::exception e)	199	catch (std::exception e)
187	{	200	{
-		201	this->mutex_clients_.unlock();
188	throw e;	202	throw e;
189	}	203	}
190		204
191	this->clients_[client->GetId()] = client;	205	this->clients_[client->GetId()] = client;
-		206
-		207	this->mutex_clients_.unlock();
192		208
193	return client->GetId();	209	return client->GetId();
194	}	210	}
195		211
196	void Manager::SendToAll(ServerId server_id, Buffer data)	212	void Manager::SendToAll(ServerId server_id, Buffer data)
197	{	213	{
198	this->io_service_.post(boost::bind(&Manager::SendToAllHandler, this, server_id, data));	214	this->io_service_.post(boost::bind(&Manager::SendToAllHandler, this, server_id, data));
199	}	215	}
200		216
201	void Manager::SendTo(ClientId client_id, Buffer data)	217	void Manager::SendTo(ClientId client_id, Buffer data)
202	{	218	{
203	this->io_service_.post(boost::bind(&Manager::SendToHandler, this, client_id, data));	219	this->io_service_.post(boost::bind(&Manager::SendToHandler, this, client_id, data));
204	}	220	}
205		221
206	void Manager::StopServer(ServerId server_id)	222	void Manager::StopServer(ServerId server_id)
207	{	223	{
208	this->io_service_.post(boost::bind(&Manager::StopServerHandler, this, server_id));	224	this->io_service_.post(boost::bind(&Manager::StopServerHandler, this, server_id));
209	}	225	}
210		226
211	void Manager::Disconnect(ClientId client_id)	227	void Manager::Disconnect(ClientId client_id)
212	{	228	{
213	this->io_service_.post(boost::bind(&Manager::DisconnectHandler, this, client_id));	229	this->io_service_.post(boost::bind(&Manager::DisconnectHandler, this, client_id));
214	}	230	}
215		231
216	void Manager::SendToAllHandler(ServerId server_id, Buffer data)	232	void Manager::SendToAllHandler(ServerId server_id, Buffer data)
217	{	233	{
218	~~for (ClientList::iterator it = this->clients_.begin(); it !=~~ this->clients_.~~end~~(); it++)	234	this->mutex_clients_.lock();
219	{	235
220	if (it->second->~~GetServerId~~() ~~== server_id~~)	236	for (ClientList::iterator it = this->clients_.begin(); it != this->clients_.end(); it++)
221	{	237	{
-		238	if (it->second->GetServerId() == server_id)
-		239	{
222	it->second->Send(data);	240	it->second->Send(data);
223	}	241	}
224	}	242	}
-		243
-		244	this->mutex_clients_.unlock();
225	}	245	}
226		246
227	void Manager::SendToHandler(ClientId client_id, Buffer data)	247	void Manager::SendToHandler(ClientId client_id, Buffer data)
228	{	248	{
-		249	this->mutex_clients_.lock();
-		250
229	ClientList::iterator it = this->clients_.find(client_id);	251	ClientList::iterator it = this->clients_.find(client_id);
230		252
231	if (it != this->clients_.end())	253	if (it != this->clients_.end())
232	{	254	{
233	it->second->Send(data);	255	it->second->Send(data);
234	}	256	}
-		257
-		258	this->mutex_clients_.unlock();
235	}	259	}
236		260
237	void Manager::StopServerHandler(ServerId server_id)	261	void Manager::StopServerHandler(ServerId server_id)
238	{	262	{
-		263	this->mutex_clients_.lock();
-		264
239	for (ClientList::iterator it = this->clients_.begin(); it != this->clients_.end(); it++)	265	for (ClientList::iterator it = this->clients_.begin(); it != this->clients_.end(); it++)
240	{	266	{
241	if (it->second->GetServerId() == server_id)	267	if (it->second->GetServerId() == server_id)
242	{	268	{
-		269	this->mutex_clients_.unlock();
243	it->second->Disconnect();	270	it->second->Disconnect();
-		271	this->mutex_clients_.lock();
244	}	272	}
245	}	273	}
-		274
-		275	this->mutex_clients_.unlock();
246	}	276	}
247		277
248	void Manager::DisconnectHandler(ClientId client_id)	278	void Manager::DisconnectHandler(ClientId client_id)
249	{	279	{
-		280	this->mutex_clients_.lock();
-		281
250	ClientList::iterator it = this->clients_.find(client_id);	282	ClientList::iterator it = this->clients_.find(client_id);
-		283
-		284	this->mutex_clients_.unlock();
251		285
252	if (it != this->clients_.end())	286	if (it != this->clients_.end())
253	{	287	{
254	it->second->Disconnect();	288	it->second->Disconnect();
255	}	289	}
-		290
-		291
256	}	292	}
257		293
258	}; // namespace net	294	}; // namespace net
259	}; // namespace atom	295	}; // namespace atom

Subversion Repositories HomeAutomation

(root)/trunk/PcSoftware/Atom1.5/src/net/Manager.cpp – Rev 1594 → 1595