00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
00044
00045
00046
00047
00048
00049
00050
00051
00052
00053
00054 #include "PeriodicBroadcast.h"
00055
00056 namespace ariba {
00057 namespace utility {
00058
00059 use_logging_cpp(PeriodicBroadcast);
00060 const long PeriodicBroadcast::timerinterval = 1;
00061 const long PeriodicBroadcast::servicetimeout = 3;
00062 const unsigned int PeriodicBroadcast::serverport_v4 = 5634;
00063 const unsigned int PeriodicBroadcast::serverport_v6 = 5636;
00064
00065 PeriodicBroadcast::PeriodicBroadcast(BootstrapInformationCallback* _callback, string info)
00066 : BootstrapModule(_callback),
00067 server(io_service, &newRemoteServices, &newRemoteServicesMutex) {
00068 }
00069
00070 PeriodicBroadcast::~PeriodicBroadcast(){
00071 }
00072
00073 void PeriodicBroadcast::threadFunc(PeriodicBroadcast* obj){
00074 obj->io_service.run();
00075 }
00076
00077 string PeriodicBroadcast::getName(){
00078 return "PeriodicBroadcast";
00079 }
00080
00081 string PeriodicBroadcast::getInformation(){
00082 return "periodic broadcasting of service information";
00083 }
00084
00085 bool PeriodicBroadcast::isFunctional(){
00086 return true;
00087 }
00088
00089 void PeriodicBroadcast::start(){
00090 io_service_thread = new boost::thread(
00091 boost::bind(&PeriodicBroadcast::threadFunc, this) );
00092
00093 Timer::setInterval( timerinterval*1000 );
00094 Timer::start();
00095 }
00096
00097 void PeriodicBroadcast::stop(){
00098 io_service.stop();
00099 io_service_thread->join();
00100 delete io_service_thread;
00101 io_service_thread = NULL;
00102
00103 Timer::stop();
00104 }
00105
00106 void PeriodicBroadcast::publishService(string name, string info1, string info2, string info3){
00107 Service service(name, info1, info2, info3) ;
00108
00109 boost::mutex::scoped_lock lock( localServicesMutex );
00110 if(name.empty()) return;
00111
00112 localServices.insert( std::make_pair(name, service) );
00113 }
00114
00115 void PeriodicBroadcast::revokeService(string name){
00116 boost::mutex::scoped_lock lock( localServicesMutex );
00117 if(name.empty()) return;
00118
00119 ServiceList::iterator i = localServices.find( name );
00120 if( i != localServices.end() ) localServices.erase( name );
00121 }
00122
00123 void PeriodicBroadcast::eventFunction(){
00124 sendLocalServices();
00125 updateRemoteServices();
00126 }
00127
00128 void PeriodicBroadcast::sendLocalServices(){
00129 boost::mutex::scoped_lock lock( localServicesMutex );
00130
00131 ServiceList::iterator i = localServices.begin();
00132 ServiceList::iterator iend = localServices.end();
00133
00134 for( ; i != iend; i++)
00135 server.sendservice( i->second );
00136 }
00137
00138 void PeriodicBroadcast::updateRemoteServices(){
00139
00140
00141
00142 {
00143 boost::mutex::scoped_lock lock( remoteServicesMutex );
00144 bool deleted;
00145
00146 do {
00147 deleted = false;
00148
00149 ServiceList::iterator i = remoteServices.begin();
00150 ServiceList::iterator iend = remoteServices.end();
00151
00152 for( ; i != iend; i++ ){
00153
00154 if( time(NULL) > (i->second.getLastseen() + servicetimeout) ){
00155 remoteServices.erase( i );
00156 deleted = true;
00157 break;
00158 }
00159 }
00160
00161 } while(deleted);
00162 }
00163
00164
00165
00166 {
00167 boost::mutex::scoped_lock lock( newRemoteServicesMutex );
00168 typedef std::pair<string,Service> mapitem;
00169
00170 BOOST_FOREACH( mapitem item, newRemoteServices ){
00171
00172 string name = item.first;
00173 Service service = item.second;
00174
00175 ServiceList::iterator i = remoteServices.find( name );
00176 if( i != remoteServices.end() ) {
00177
00178 i->second.setLastseen( service.getLastseen() );
00179 continue;
00180 }
00181
00182 {
00183
00184
00185 boost::mutex::scoped_lock lock2( remoteServicesMutex );
00186 remoteServices.insert( std::make_pair(name, service) );
00187 }
00188
00189 callback->onBootstrapServiceFound(name,
00190 service.getInfo1(), service.getInfo2(), service.getInfo3());
00191 }
00192
00193
00194 newRemoteServices.clear();
00195 }
00196 }
00197
00198 }}