src/SALOMESDS/SALOMESDS_DataScopeServer.cxx

   1 // Copyright (C) 2007-2015  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19 // Author : Anthony GEAY (EDF R&D)
  20
  21 #include "SALOMESDS_DataScopeServer.hxx"
  22 #include "SALOMESDS_DataServerManager.hxx"
  23 #include "SALOMESDS_PickelizedPyObjRdOnlyServer.hxx"
  24 #include "SALOMESDS_PickelizedPyObjRdExtServer.hxx"
  25 #include "SALOMESDS_PickelizedPyObjRdWrServer.hxx"
  26 #include "SALOMESDS_PickelizedPyObjRdExtInitServer.hxx"
  27 #include "SALOMESDS_TrustTransaction.hxx"
  28 #include "SALOMESDS_KeyWaiter.hxx"
  29 #include "SALOMESDS_Transaction.hxx"
  30 #include "SALOME_NamingService.hxx"
  31 #include "SALOMESDS_Exception.hxx"
  32
  33 #include <sstream>
  34 #include <iterator>
  35 #include <algorithm>
  36
  37 // agy : awful, to be factorized with ContainerManager.
  38 #ifndef WIN32
  39 #include <unistd.h>
  40 #else
  41 #include <process.h>
  42 #define getpid _getpid
  43 #endif
  44
  45 using namespace SALOMESDS;
  46
  47 std::size_t DataScopeServerBase::COUNTER=0;
  48
  49 void DataScopeKiller::shutdown()
  50 {
  51   Py_Finalize();
  52   _orb->shutdown(0);
  53 }
  54
  55 DataScopeServerBase::DataScopeServerBase(CORBA::ORB_ptr orb, SALOME::DataScopeKiller_var killer, const std::string& scopeName):_globals(0),_locals(0),_pickler(0),_orb(CORBA::ORB::_duplicate(orb)),_name(scopeName),_killer(killer)
  56 {
  57 }
  58
  59 DataScopeServerBase::DataScopeServerBase(const DataScopeServerBase& other):omniServant(other),ServantBase(other),_globals(0),_locals(0),_pickler(0),_name(other._name),_vars(other._vars),_killer(other._killer)
  60 {
  61 }
  62
  63 DataScopeServerBase::~DataScopeServerBase()
  64 {
  65   // _globals is borrowed ref -> do nothing
  66   Py_XDECREF(_locals);
  67   Py_XDECREF(_pickler);
  68   for(std::list< std::pair< SALOME::BasicDataServer_var, BasicDataServer * > >::const_iterator it=_vars.begin();it!=_vars.end();it++)
  69     {
  70        BasicDataServer *obj((*it).second);
  71        if(obj)
  72          {
  73            obj->decrRef();
  74          }
  75     }
  76 }
  77
  78 /*!
  79  * Called remotely -> to protect against throw
  80  */
  81 void DataScopeServerBase::ping()
  82 {
  83 }
  84
  85 /*!
  86  * Called remotely -> to protect against throw
  87  */
  88 char *DataScopeServerBase::getScopeName()
  89 {
  90   return CORBA::string_dup(_name.c_str());
  91 }
  92
  93 /*!
  94  * Called remotely -> to protect against throw
  95  */
  96 SALOME::StringVec *DataScopeServerBase::listVars()
  97 {
  98   SALOME::StringVec *ret(new SALOME::StringVec);
  99   std::size_t sz(_vars.size());
 100   ret->length(sz);
 101   std::list< std::pair< SALOME::BasicDataServer_var, BasicDataServer * > >::iterator it(_vars.begin());
 102   for(std::size_t i=0;i<sz;it++,i++)
 103     {
 104       BasicDataServer *obj((*it).second);
 105       std::string name(obj->getVarNameCpp());
 106       (*ret)[i]=CORBA::string_dup(name.c_str());
 107     }
 108   return ret;
 109 }
 110
 111 CORBA::Boolean DataScopeServerBase::existVar(const char *varName)
 112 {
 113   std::list< std::pair< SALOME::BasicDataServer_var, BasicDataServer * > >::const_iterator it(_vars.begin());
 114   for(;it!=_vars.end();it++)
 115     if((*it).second->getVarNameCpp()==varName)
 116       return true;
 117   return false;
 118 }
 119
 120 SALOME::BasicDataServer_ptr DataScopeServerBase::retrieveVarInternal(const char *varName)
 121 {
 122   std::list< std::pair< SALOME::BasicDataServer_var, BasicDataServer * > >::const_iterator it0(retrieveVarInternal3(varName));
 123   return SALOME::BasicDataServer::_duplicate((*it0).first);
 124 }
 125
 126 BasicDataServer *DataScopeServerBase::retrieveVarInternal2(const std::string& varName)
 127 {
 128   std::list< std::pair< SALOME::BasicDataServer_var, BasicDataServer * > >::const_iterator it0(retrieveVarInternal3(varName));
 129   return (*it0).second;
 130 }
 131
 132 void DataScopeServerBase::deleteVar(const char *varName)
 133 {
 134   std::string varNameCpp(varName);
 135   std::vector<std::string> allNames(getAllVarNames());
 136   std::vector<std::string>::iterator it(std::find(allNames.begin(),allNames.end(),varNameCpp));
 137   if(it==allNames.end())
 138     {
 139       std::ostringstream oss; oss << "DataScopeServerBase::deleteVar : name \"" << varNameCpp << "\" does not exists ! Possibilities are :";
 140       std::copy(allNames.begin(),allNames.end(),std::ostream_iterator<std::string>(oss,", "));
 141       throw Exception(oss.str());
 142     }
 143   std::size_t pos(std::distance(allNames.begin(),it));
 144   std::list< std::pair< SALOME::BasicDataServer_var, BasicDataServer * > >::iterator it0(_vars.begin());
 145   for(std::size_t ii=0;ii<pos;ii++,it0++);
 146   (*it0).second->decrRef();
 147   _vars.erase(it0);
 148 }
 149
 150 CORBA::Boolean DataScopeServerBase::shutdownIfNotHostedByDSM(SALOME::DataScopeKiller_out killer)
 151 {
 152   SALOME_NamingService ns(_orb);
 153   CORBA::Object_var obj(ns.Resolve(DataServerManager::NAME_IN_NS));
 154   SALOME::DataServerManager_var dsm(SALOME::DataServerManager::_narrow(obj));
 155   if(CORBA::is_nil(dsm))
 156     throw Exception("Unable to reach in the NS the unique DataServerManager instance of the Session !");
 157   // destroy ref in the naming service
 158   std::string fullScopeName(SALOMESDS::DataServerManager::CreateAbsNameInNSFromScopeName(_name));
 159   ns.Destroy_Name(fullScopeName.c_str());
 160   // establish if dsm and this shared the same POA. If yes dsm and this are collocated !
 161   PortableServer::ServantBase *ret(0);
 162   try
 163     {
 164       ret=_poa->reference_to_servant(dsm);
 165     }
 166   catch(...) { ret=0; }
 167   //
 168   if(!ret)
 169     {
 170       enforcedRelease();
 171       killer=SALOME::DataScopeKiller::_duplicate(_killer);
 172       return true;
 173     }
 174   else
 175     {
 176       ret->_remove_ref();
 177       enforcedRelease();
 178       killer=SALOME::DataScopeKiller::_duplicate(_killer);
 179       return false;
 180     }
 181 }
 182
 183 SALOME::ByteVec *DataScopeServerBase::fetchSerializedContent(const char *varName)
 184 {
 185   BasicDataServer *var(retrieveVarInternal2(varName));
 186   PickelizedPyObjServer *varc(dynamic_cast<PickelizedPyObjServer *>(var));
 187   if(!varc)
 188     {
 189       std::ostringstream oss; oss << "DataScopeServerBase::fetchSerializedContent : var \"" << varName << "\" exists but it is not serialized !";
 190       throw Exception(oss.str());
 191     }
 192   return varc->fetchSerializedContent();
 193 }
 194
 195 SALOME::SeqOfByteVec *DataScopeServerBase::getAllKeysOfVarWithTypeDict(const char *varName)
 196 {
 197   BasicDataServer *var(retrieveVarInternal2(varName));
 198   PickelizedPyObjServer *varc(dynamic_cast<PickelizedPyObjServer *>(var));
 199   if(!varc)
 200     {
 201       std::ostringstream oss; oss << "DataScopeServerBase::getAllKeysOfVarWithTypeDict : var \"" << varName << "\" exists but it is not serialized !";
 202       throw Exception(oss.str());
 203     }
 204   if(!varc->isDict())
 205     {
 206       std::ostringstream oss; oss << "DataScopeServerBase::getAllKeysOfVarWithTypeDict : var \"" << varName << "\" exists but it is not a PyDict !";
 207       throw Exception(oss.str());
 208     }
 209   PyObject *keys(PyDict_Keys(varc->getPyObj()));
 210   if(!PyList_Check(keys))
 211     {
 212       std::ostringstream oss; oss << "DataScopeServerBase::getAllKeysOfVarWithTypeDict : var \"" << varName << "\" has keys but not of type list !";
 213       throw Exception(oss.str());
 214     }
 215   Py_ssize_t sz(PyList_Size(keys));
 216   SALOME::SeqOfByteVec *ret(new SALOME::SeqOfByteVec);
 217   ret->length(sz);
 218   for(Py_ssize_t i=0;i<sz;i++)
 219     {
 220       PyObject *item(PyList_GetItem(keys,i));
 221       Py_XINCREF(item);
 222       std::string pickel(varc->pickelize(item));//item consumed
 223       PickelizedPyObjServer::FromCppToByteSeq(pickel,(*ret)[i]);
 224     }
 225   Py_XDECREF(keys);
 226   return ret;
 227 }
 228
 229 void DataScopeServerBase::initializePython(int argc, char *argv[])
 230 {
 231   Py_Initialize();
 232   PyEval_InitThreads();
 233   PySys_SetArgv(argc,argv);
 234   PyObject *mainmod(PyImport_AddModule("__main__"));
 235   _globals=PyModule_GetDict(mainmod);
 236   if(PyDict_GetItemString(_globals, "__builtins__") == NULL)
 237     {
 238       PyObject *bimod(PyImport_ImportModule("__builtin__"));
 239       if (bimod == NULL || PyDict_SetItemString(_globals, "__builtins__", bimod) != 0)
 240         Py_FatalError("can't add __builtins__ to __main__");
 241       Py_XDECREF(bimod);
 242     }
 243   _locals=PyDict_New();
 244   PyObject *tmp(PyList_New(0));
 245   _pickler=PyImport_ImportModuleLevel(const_cast<char *>("cPickle"),_globals,_locals,tmp,-1);
 246 }
 247
 248 void DataScopeServerBase::registerToSalomePiDict() const
 249 {
 250   PyObject *mod(PyImport_ImportModule("addToKillList"));//new value
 251   if(!mod)
 252     return;
 253   PyObject *meth(PyObject_GetAttrString(mod,"addToKillList"));//new value
 254   if(!meth)
 255     { Py_XDECREF(mod); return ; }
 256   PyObject *args(PyTuple_New(2));
 257   PyTuple_SetItem(args,0,PyInt_FromLong(getpid()));
 258   PyTuple_SetItem(args,1,PyString_FromString("SALOME_DataScopeServerBase"));
 259   PyObject *res(PyObject_CallObject(meth,args));
 260   Py_XDECREF(args);
 261   Py_XDECREF(res);
 262   Py_XDECREF(meth);
 263   Py_XDECREF(mod);
 264 }
 265
 266 void DataScopeServerBase::setPOA(PortableServer::POA_var poa)
 267 {
 268   _poa=poa;
 269 }
 270
 271 void DataScopeServerBase::registerInNS(SALOME::DataScopeServerBase_ptr ptr)
 272 {
 273   std::string fullScopeName(SALOMESDS::DataServerManager::CreateAbsNameInNSFromScopeName(_name));
 274   SALOME_NamingService ns(_orb);
 275   ns.Register(ptr,fullScopeName.c_str());
 276 }
 277
 278 std::string DataScopeServerBase::BuildTmpVarNameFrom(const std::string& varName)
 279 {
 280   std::ostringstream oss;
 281   oss << varName << "@" << COUNTER++;
 282   return oss.str();
 283 }
 284
 285 std::vector< std::string > DataScopeServerBase::getAllVarNames() const
 286 {
 287   std::size_t sz(_vars.size());
 288   std::vector<std::string> ret(sz);
 289   std::list< std::pair< SALOME::BasicDataServer_var, BasicDataServer * > >::const_iterator it(_vars.begin());
 290   for(std::size_t i=0;i<sz;it++,i++)
 291     ret[i]=(*it).second->getVarNameCpp();
 292   return ret;
 293 }
 294
 295 void DataScopeServerBase::checkNotAlreadyExistingVar(const std::string& varName) const
 296 {
 297   std::vector<std::string> allNames(getAllVarNames());
 298   std::vector<std::string>::iterator it(std::find(allNames.begin(),allNames.end(),varName));
 299   if(it!=allNames.end())
 300     {
 301       std::ostringstream oss; oss << "DataScopeServerBase::checkNotAlreadyExistingVar : name \"" << varName << "\" already exists !";
 302       throw Exception(oss.str());
 303     }
 304 }
 305
 306 void DataScopeServerBase::checkExistingVar(const std::string& varName) const
 307 {
 308   std::vector<std::string> allNames(getAllVarNames());
 309   std::vector<std::string>::iterator it(std::find(allNames.begin(),allNames.end(),varName));
 310   if(it==allNames.end())
 311     {
 312       std::ostringstream oss; oss << "DataScopeServerBase::checkExistingVar : name \"" << varName << "\" does not exist !";
 313       throw Exception(oss.str());
 314     }
 315 }
 316
 317 PickelizedPyObjServer *DataScopeServerBase::checkVarExistingAndDict(const std::string& varName)
 318 {
 319   checkExistingVar(varName);
 320   BasicDataServer *var(retrieveVarInternal2(varName.c_str()));
 321   PickelizedPyObjServer *ret(dynamic_cast<PickelizedPyObjServer *>(var));
 322   if(!ret)
 323     {
 324       std::ostringstream oss; oss << "TransactionAddKeyValueHard::prepareRollBackInCaseOfFailure : var \"" << varName << "\"exists but it is not serialized !";
 325       throw Exception(oss.str());
 326     }
 327   if(!ret->isDict())
 328     {
 329       std::ostringstream oss; oss << "TransactionAddKeyValueHard::prepareRollBackInCaseOfFailure : var \"" << varName << "\"exists but it is not a Dict !";
 330       throw Exception(oss.str());
 331     }
 332   return ret;
 333 }
 334
 335 void DataScopeServerBase::moveStatusOfVarFromRdWrToRdOnly(const std::string& varName)
 336 {
 337   std::list< std::pair< SALOME::BasicDataServer_var, BasicDataServer * > >::iterator it(retrieveVarInternal4(varName));
 338   std::pair< SALOME::BasicDataServer_var, BasicDataServer * >& p(*it);
 339   PickelizedPyObjRdWrServer *varc(dynamic_cast<PickelizedPyObjRdWrServer *>(p.second));
 340   if(!varc)
 341     throw Exception("DataScopeServerBase::moveStatusOfVarFromRdWrToRdOnly : var is not a RdWr !");
 342   PyObject *pyobj(varc->getPyObj()); Py_XINCREF(pyobj);
 343   PickelizedPyObjRdOnlyServer *newVar(new PickelizedPyObjRdOnlyServer(this,varName,pyobj));
 344   CORBA::Object_var obj(newVar->activate());
 345   SALOME::BasicDataServer_var obj2(SALOME::BasicDataServer::_narrow(obj));
 346   p.first=obj2; p.second=newVar;
 347   varc->decrRef();
 348 }
 349
 350 void DataScopeServerBase::moveStatusOfVarFromRdOnlyToRdWr(const std::string& varName)
 351 {
 352   std::list< std::pair< SALOME::BasicDataServer_var, BasicDataServer * > >::iterator it(retrieveVarInternal4(varName));
 353   std::pair< SALOME::BasicDataServer_var, BasicDataServer * >& p(*it);
 354   PickelizedPyObjRdOnlyServer *varc(dynamic_cast<PickelizedPyObjRdOnlyServer *>(p.second));
 355   if(!varc)
 356     throw Exception("DataScopeServerBase::moveStatusOfVarFromRdOnlyToRdWr : var is not a RdWr !");
 357   PyObject *pyobj(varc->getPyObj()); Py_XINCREF(pyobj);
 358   PickelizedPyObjRdWrServer *newVar(new PickelizedPyObjRdWrServer(this,varName,pyobj));
 359   CORBA::Object_var obj(newVar->activate());
 360   SALOME::BasicDataServer_var obj2(SALOME::BasicDataServer::_narrow(obj));
 361   p.first=obj2; p.second=newVar;
 362   varc->decrRef();
 363 }
 364
 365 void DataScopeServerBase::moveStatusOfVarFromRdExtOrRdExtInitToRdExtInit(const std::string& varName)
 366 {
 367   std::list< std::pair< SALOME::BasicDataServer_var, BasicDataServer * > >::iterator it(retrieveVarInternal4(varName));
 368   std::pair< SALOME::BasicDataServer_var, BasicDataServer * >& p(*it);
 369   PickelizedPyObjRdExtServer *varc0(dynamic_cast<PickelizedPyObjRdExtServer *>(p.second));
 370   PickelizedPyObjRdExtInitServer *varc1(dynamic_cast<PickelizedPyObjRdExtInitServer *>(p.second));
 371   if(!varc0 && !varc1)
 372     throw Exception("DataScopeServerBase::moveStatusOfVarFromRdExtOrRdExtInitToRdExtInit : var is neither RdExt nor RdExtInit !");
 373   if(varc0)
 374     {
 375       PyObject *pyobj(varc0->getPyObj()); Py_XINCREF(pyobj);
 376       PickelizedPyObjRdExtInitServer *newVar(new PickelizedPyObjRdExtInitServer(this,varName,pyobj));
 377       CORBA::Object_var obj(newVar->activate());
 378       SALOME::BasicDataServer_var obj2(SALOME::BasicDataServer::_narrow(obj));
 379       p.first=obj2; p.second=newVar;
 380       varc0->decrRef();
 381     }
 382 }
 383
 384 void DataScopeServerBase::moveStatusOfVarFromRdExtInitToRdExt(const std::string& varName)
 385 {
 386   std::list< std::pair< SALOME::BasicDataServer_var, BasicDataServer * > >::iterator it(retrieveVarInternal4(varName));
 387   std::pair< SALOME::BasicDataServer_var, BasicDataServer * >& p(*it);
 388   PickelizedPyObjRdExtInitServer *varc(dynamic_cast<PickelizedPyObjRdExtInitServer *>(p.second));
 389   if(!varc)
 390     throw Exception("DataScopeServerBase::moveStatusOfVarFromRdExtInitToRdExt : var is not a RdExtInit !");
 391   PyObject *pyobj(varc->getPyObj()); Py_XINCREF(pyobj);
 392   PickelizedPyObjRdExtServer *newVar(new PickelizedPyObjRdExtServer(this,varName,pyobj));
 393   CORBA::Object_var obj(newVar->activate());
 394   SALOME::BasicDataServer_var obj2(SALOME::BasicDataServer::_narrow(obj));
 395   p.first=obj2; p.second=newVar;
 396   varc->decrRef();
 397 }
 398
 399 std::list< std::pair< SALOME::BasicDataServer_var, BasicDataServer * > >::const_iterator DataScopeServerBase::retrieveVarInternal3(const std::string& varName) const
 400 {
 401   std::vector<std::string> allNames(getAllVarNames());
 402   std::vector<std::string>::iterator it(std::find(allNames.begin(),allNames.end(),varName));
 403   if(it==allNames.end())
 404     {
 405       std::ostringstream oss; oss << "DataScopeServerBase::retrieveVarInternal3 : name \"" << varName << "\" does not exists ! Possibilities are :";
 406       std::copy(allNames.begin(),allNames.end(),std::ostream_iterator<std::string>(oss,", "));
 407       throw Exception(oss.str());
 408     }
 409   std::size_t pos(std::distance(allNames.begin(),it));
 410   std::list< std::pair< SALOME::BasicDataServer_var, BasicDataServer * > >::const_iterator it0(_vars.begin());
 411   for(std::size_t i=0;i<pos;i++,it0++);
 412   return it0;
 413 }
 414
 415 std::list< std::pair< SALOME::BasicDataServer_var, BasicDataServer * > >::iterator DataScopeServerBase::retrieveVarInternal4(const std::string& varName)
 416 {
 417   std::vector<std::string> allNames(getAllVarNames());
 418   std::vector<std::string>::iterator it(std::find(allNames.begin(),allNames.end(),varName));
 419   if(it==allNames.end())
 420     {
 421       std::ostringstream oss; oss << "DataScopeServerBase::retrieveVarInternal4 : name \"" << varName << "\" does not exists ! Possibilities are :";
 422       std::copy(allNames.begin(),allNames.end(),std::ostream_iterator<std::string>(oss,", "));
 423       throw Exception(oss.str());
 424     }
 425   std::size_t pos(std::distance(allNames.begin(),it));
 426   std::list< std::pair< SALOME::BasicDataServer_var, BasicDataServer * > >::iterator it0(_vars.begin());
 427   for(std::size_t i=0;i<pos;i++,it0++);
 428   return it0;
 429 }
 430
 431 ///////
 432
 433 DataScopeServer::DataScopeServer(CORBA::ORB_ptr orb, SALOME::DataScopeKiller_var killer, const std::string& scopeName):DataScopeServerBase(orb,killer,scopeName)
 434 {
 435 }
 436
 437 DataScopeServer::DataScopeServer(const DataScopeServer& other):omniServant(other),ServantBase(other),DataScopeServerBase(other)
 438 {
 439 }
 440
 441 SALOME::PickelizedPyObjRdOnlyServer_ptr DataScopeServer::createRdOnlyVar(const char *varName, const SALOME::ByteVec& constValue)
 442 {
 443   std::string varNameCpp(varName);
 444   checkNotAlreadyExistingVar(varNameCpp);
 445   PickelizedPyObjRdOnlyServer *tmp(new PickelizedPyObjRdOnlyServer(this,varNameCpp,constValue));
 446   CORBA::Object_var ret(tmp->activate());
 447   std::pair< SALOME::BasicDataServer_var, BasicDataServer * > p(SALOME::BasicDataServer::_narrow(ret),tmp);
 448   _vars.push_back(p);
 449   return SALOME::PickelizedPyObjRdOnlyServer::_narrow(ret);
 450 }
 451
 452 SALOME::PickelizedPyObjRdExtServer_ptr DataScopeServer::createRdExtVar(const char *varName, const SALOME::ByteVec& constValue)
 453 {
 454   std::string varNameCpp(varName);
 455   checkNotAlreadyExistingVar(varNameCpp);
 456   PickelizedPyObjRdExtServer *tmp(new PickelizedPyObjRdExtServer(this,varNameCpp,constValue));
 457   CORBA::Object_var ret(tmp->activate());
 458   std::pair< SALOME::BasicDataServer_var, BasicDataServer * > p(SALOME::BasicDataServer::_narrow(ret),tmp);
 459   _vars.push_back(p);
 460   return SALOME::PickelizedPyObjRdExtServer::_narrow(ret);
 461 }
 462
 463 SALOME::PickelizedPyObjRdWrServer_ptr DataScopeServer::createRdWrVar(const char *typeName, const char *varName)
 464 {
 465   std::string varNameCpp(varName),typeNameCpp(typeName);
 466   checkNotAlreadyExistingVar(varNameCpp);
 467   PickelizedPyObjRdWrServer *tmp(new PickelizedPyObjRdWrServer(this,typeNameCpp,varNameCpp));
 468   CORBA::Object_var ret(tmp->activate());
 469   std::pair< SALOME::BasicDataServer_var, BasicDataServer * > p(SALOME::BasicDataServer::_narrow(ret),tmp);
 470   _vars.push_back(p);
 471   return SALOME::PickelizedPyObjRdWrServer::_narrow(ret);
 472 }
 473
 474 DataScopeServer::~DataScopeServer()
 475 {
 476 }
 477
 478 ////////
 479
 480 DataScopeServerTransaction::DataScopeServerTransaction(CORBA::ORB_ptr orb, SALOME::DataScopeKiller_var killer, const std::string& scopeName):DataScopeServerBase(orb,killer,scopeName)
 481 {
 482   CORBA::Object_var obj(_orb->resolve_initial_references("RootPOA"));
 483   PortableServer::POA_var poa(PortableServer::POA::_narrow(obj));
 484   //
 485   PortableServer::POAManager_var mgr(poa->the_POAManager());
 486   CORBA::PolicyList policies;
 487   policies.length(1);
 488   PortableServer::ThreadPolicy_var threadPol(poa->create_thread_policy(PortableServer::ORB_CTRL_MODEL));
 489   policies[0]=PortableServer::ThreadPolicy::_duplicate(threadPol);
 490   PortableServer::POA_var poa2(poa->create_POA("POAForWaiters",mgr,policies));
 491   threadPol->destroy();
 492   //
 493   _poa_for_key_waiter=poa2;
 494 }
 495
 496 DataScopeServerTransaction::DataScopeServerTransaction(const DataScopeServerTransaction& other):omniServant(other),ServantBase(other),DataScopeServerBase(other),_poa_for_key_waiter(other.getPOA4KeyWaiter())
 497 {
 498 }
 499
 500 char *DataScopeServerTransaction::getAccessOfVar(const char *varName)
 501 {
 502   std::string varNameCpp(varName);
 503   checkExistingVar(varNameCpp);
 504   BasicDataServer *var(retrieveVarInternal2(varName));
 505   if(!var)
 506     throw Exception("DataScopeServerTransaction::getAccessOfVar : variable is NULL !");
 507   PickelizedPyObjServer *varc(dynamic_cast<PickelizedPyObjServer *>(var));
 508   if(!varc)
 509     throw Exception("DataScopeServerTransaction::getAccessOfVar : variable is not of type PickelizedPyObjServer !");
 510   std::string ret(varc->getAccessStr());
 511   return CORBA::string_dup(ret.c_str());
 512 }
 513
 514 void DataScopeServerTransaction::createRdOnlyVarInternal(const std::string& varName, const SALOME::ByteVec& constValue)
 515 {
 516   checkNotAlreadyExistingVar(varName);
 517   PickelizedPyObjRdOnlyServer *tmp(new PickelizedPyObjRdOnlyServer(this,varName,constValue));
 518   CORBA::Object_var ret(tmp->activate());
 519   std::pair< SALOME::BasicDataServer_var, BasicDataServer * > p(SALOME::BasicDataServer::_narrow(ret),tmp);
 520   _vars.push_back(p);
 521 }
 522
 523 void DataScopeServerTransaction::createRdExtVarInternal(const std::string& varName, const SALOME::ByteVec& constValue)
 524 {
 525   checkNotAlreadyExistingVar(varName);
 526   PickelizedPyObjRdExtServer *tmp(new PickelizedPyObjRdExtServer(this,varName,constValue));
 527   CORBA::Object_var ret(tmp->activate());
 528   std::pair< SALOME::BasicDataServer_var, BasicDataServer * > p(SALOME::BasicDataServer::_narrow(ret),tmp);
 529   _vars.push_back(p);
 530 }
 531
 532 void DataScopeServerTransaction::createRdExtInitVarInternal(const std::string& varName, const SALOME::ByteVec& constValue)
 533 {
 534   checkNotAlreadyExistingVar(varName);
 535   PickelizedPyObjRdExtInitServer *tmp(new PickelizedPyObjRdExtInitServer(this,varName,constValue));
 536   CORBA::Object_var ret(tmp->activate());
 537   std::pair< SALOME::BasicDataServer_var, BasicDataServer * > p(SALOME::BasicDataServer::_narrow(ret),tmp);
 538   _vars.push_back(p);
 539 }
 540
 541 void DataScopeServerTransaction::createRdWrVarInternal(const std::string& varName, const SALOME::ByteVec& constValue)
 542 {
 543   checkNotAlreadyExistingVar(varName);
 544   PickelizedPyObjRdWrServer *tmp(new PickelizedPyObjRdWrServer(this,varName,constValue));
 545   CORBA::Object_var ret(tmp->activate());
 546   std::pair< SALOME::BasicDataServer_var, BasicDataServer * > p(SALOME::BasicDataServer::_narrow(ret),tmp);
 547   _vars.push_back(p);
 548 }
 549
 550 SALOME::Transaction_ptr DataScopeServerTransaction::createRdOnlyVarTransac(const char *varName, const SALOME::ByteVec& constValue)
 551 {
 552   checkNotAlreadyExistingVar(varName);
 553   TransactionRdOnlyVarCreate *ret(new TransactionRdOnlyVarCreate(this,varName,constValue));
 554   CORBA::Object_var obj(ret->activate());
 555   return SALOME::Transaction::_narrow(obj);
 556 }
 557
 558 SALOME::Transaction_ptr DataScopeServerTransaction::createRdExtVarTransac(const char *varName, const SALOME::ByteVec& constValue)
 559 {
 560   checkNotAlreadyExistingVar(varName);
 561   TransactionRdExtVarCreate *ret(new TransactionRdExtVarCreate(this,varName,constValue));
 562   CORBA::Object_var obj(ret->activate());
 563   return SALOME::Transaction::_narrow(obj);
 564 }
 565
 566 SALOME::Transaction_ptr DataScopeServerTransaction::createRdExtInitVarTransac(const char *varName, const SALOME::ByteVec& constValue)
 567 {
 568   checkNotAlreadyExistingVar(varName);
 569   TransactionRdExtInitVarCreate *ret(new TransactionRdExtInitVarCreate(this,varName,constValue));
 570   CORBA::Object_var obj(ret->activate());
 571   return SALOME::Transaction::_narrow(obj);
 572 }
 573
 574 SALOME::Transaction_ptr DataScopeServerTransaction::createRdWrVarTransac(const char *varName, const SALOME::ByteVec& constValue)
 575 {
 576   checkNotAlreadyExistingVar(varName);
 577   TransactionRdWrVarCreate *ret(new TransactionRdWrVarCreate(this,varName,constValue));
 578   CORBA::Object_var obj(ret->activate());
 579   return SALOME::Transaction::_narrow(obj);
 580 }
 581
 582 void DataScopeServerTransaction::addWaitKey(KeyWaiter *kw)
 583 {
 584   if(!kw)
 585     throw Exception("DataScopeServerTransaction::addWaitKey : NULL input object !");
 586   _waiting_keys.push_back(kw);
 587 }
 588
 589 void DataScopeServerTransaction::pingKey(PyObject *keyObj)
 590 {
 591   PyObject *cmpObj(getPyCmpFunc());
 592   if(!keyObj)
 593     throw Exception("ataScopeServerTransaction::pingKey : Key Object is NULL !");
 594   PyObject *args(PyTuple_New(2));
 595   PyTuple_SetItem(args,0,keyObj); Py_XINCREF(keyObj);
 596   std::size_t ii(0);
 597   // this part does nothing except to be sure that in notify key all will be OK.
 598   for(std::list< KeyWaiter *>::iterator it=_waiting_keys.begin();it!=_waiting_keys.end();it++,ii++)
 599     {
 600       PyObject *waitKey((*it)->getKeyPyObj());
 601       PyTuple_SetItem(args,1,waitKey); Py_XINCREF(waitKey);
 602       PyObject *res(PyObject_CallObject(cmpObj,args));
 603       if(res==NULL)
 604         {
 605           std::ostringstream oss; oss << "DataScopeServerTransaction::pingKey : for object id #" << ii << " error during cmp(k,wk[i]) !";
 606           throw Exception(oss.str());
 607         }
 608       PyInt_AsLong(res);
 609       if(PyErr_Occurred())
 610         {
 611           std::ostringstream oss; oss << "DataScopeServerTransaction::pingKey : for object id #" << ii << " error during interpretation of cmp(k,wk[i]) !";
 612           throw Exception(oss.str());
 613         }
 614       Py_XDECREF(res);
 615     }
 616 }
 617
 618 void DataScopeServerTransaction::notifyKey(const std::string& varName, PyObject *keyObj, PyObject *valueObj)
 619 {
 620   PyObject *cmpObj(getPyCmpFunc());
 621   if(!keyObj)
 622     throw Exception("DataScopeServerTransaction::notifyKey : MAIN INTERNAL ERROR ! Key Object is NULL !");
 623   PyObject *args(PyTuple_New(2));
 624   PyTuple_SetItem(args,0,keyObj); Py_XINCREF(keyObj);
 625   std::size_t ii(0);
 626   std::list< KeyWaiter *> newList,listOfEltToWakeUp;
 627   for(std::list< KeyWaiter *>::iterator it=_waiting_keys.begin();it!=_waiting_keys.end();it++,ii++)
 628     {
 629       if((*it)->getVarName()!=varName)
 630         {
 631           newList.push_back(*it);
 632           continue;
 633         }
 634       PyObject *waitKey((*it)->getKeyPyObj());
 635       PyTuple_SetItem(args,1,waitKey); Py_XINCREF(waitKey);
 636       PyObject *res(PyObject_CallObject(cmpObj,args));
 637       if(res==NULL)
 638         {
 639           std::ostringstream oss; oss << "DataScopeServerTransaction::notifyKey : MAIN INTERNAL ERROR ! for object id #" << ii << " error during cmp(k,wk[i]) !";
 640           throw Exception(oss.str());
 641         }
 642       long resCpp(PyInt_AsLong(res));
 643       if(PyErr_Occurred())
 644         {
 645           std::ostringstream oss; oss << "DataScopeServerTransaction::notifyKey : MAIN INTERNAL ERROR ! for object id #" << ii << " error during interpretation of cmp(k,wk[i]) !";
 646           throw Exception(oss.str());
 647         }
 648       Py_XDECREF(res);
 649       if(resCpp==0)
 650         listOfEltToWakeUp.push_back(*it);
 651       else
 652         newList.push_back(*it);
 653     }
 654   for(std::list< KeyWaiter *>::iterator it=listOfEltToWakeUp.begin();it!=listOfEltToWakeUp.end();it++)
 655     (*it)->valueJustCome(valueObj);
 656   for(std::list< KeyWaiter *>::iterator it=listOfEltToWakeUp.begin();it!=listOfEltToWakeUp.end();it++)
 657     (*it)->go();
 658   _waiting_keys=newList;
 659 }
 660
 661 SALOME::Transaction_ptr DataScopeServerTransaction::addKeyValueInVarHard(const char *varName, const SALOME::ByteVec& key, const SALOME::ByteVec& value)
 662 {
 663   checkVarExistingAndDict(varName);
 664   TransactionAddKeyValueHard *ret(new TransactionAddKeyValueHard(this,varName,key,value));
 665   CORBA::Object_var obj(ret->activate());
 666   return SALOME::Transaction::_narrow(obj);
 667 }
 668
 669 SALOME::Transaction_ptr DataScopeServerTransaction::addKeyValueInVarErrorIfAlreadyExisting(const char *varName, const SALOME::ByteVec& key, const SALOME::ByteVec& value)
 670 {
 671   checkVarExistingAndDict(varName);
 672   TransactionAddKeyValueErrorIfAlreadyExisting *ret(new TransactionAddKeyValueErrorIfAlreadyExisting(this,varName,key,value));
 673   CORBA::Object_var obj(ret->activate());
 674   return SALOME::Transaction::_narrow(obj);
 675 }
 676
 677 SALOME::TransactionMultiKeyAddSession_ptr DataScopeServerTransaction::addMultiKeyValueSession(const char *varName)
 678 {
 679   checkVarExistingAndDict(varName);
 680   TransactionMultiKeyAddSession *ret(new TransactionMultiKeyAddSession(this,varName));
 681   CORBA::Object_var obj(ret->activate());
 682   return SALOME::TransactionMultiKeyAddSession::_narrow(obj);
 683 }
 684
 685 SALOME::Transaction_ptr DataScopeServerTransaction::removeKeyInVarErrorIfNotAlreadyExisting(const char *varName, const SALOME::ByteVec& key)
 686 {
 687   checkVarExistingAndDict(varName);
 688   TransactionRemoveKeyInVarErrorIfNotAlreadyExisting *ret(new TransactionRemoveKeyInVarErrorIfNotAlreadyExisting(this,varName,key));
 689   CORBA::Object_var obj(ret->activate());
 690   return SALOME::Transaction::_narrow(obj);
 691 }
 692
 693 SALOME::TransactionRdWrAccess_ptr DataScopeServerTransaction::createWorkingVarTransac(const char *varName, const SALOME::ByteVec& constValue)
 694 {
 695   std::string varNameCpp(varName);
 696   checkNotAlreadyExistingVar(varName);
 697   PickelizedPyObjRdWrServer *tmp(new PickelizedPyObjRdWrServer(this,varNameCpp,constValue));
 698   CORBA::Object_var obj(tmp->activate());
 699   std::pair< SALOME::BasicDataServer_var, BasicDataServer * > p(SALOME::BasicDataServer::_narrow(obj),tmp);
 700   _vars.push_back(p);
 701   //
 702   TransactionMorphRdWrIntoRdOnly *ret(new TransactionMorphRdWrIntoRdOnly(this,varName));
 703   CORBA::Object_var obj2(ret->activate());
 704   return SALOME::TransactionRdWrAccess::_narrow(obj2);
 705 }
 706
 707 SALOME::KeyWaiter_ptr DataScopeServerTransaction::waitForKeyInVar(const char *varName, const SALOME::ByteVec& keyVal)
 708 {
 709   PickelizedPyObjServer *pickelObj(checkVarExistingAndDict(varName));
 710   KeyWaiter *ret(new KeyWaiter(pickelObj,keyVal));
 711   CORBA::Object_var obj(ret->activate());//KeyWaiter instance activated inside a multithread POA contrary to all of objects in SALOMESDS in single thread !
 712   return SALOME::KeyWaiter::_narrow(obj);
 713 }
 714
 715 SALOME::KeyWaiter_ptr DataScopeServerTransaction::waitForKeyInVarAndKillIt(const char *varName, const SALOME::ByteVec& keyVal, SALOME::Transaction_out transac)
 716 {
 717   PickelizedPyObjServer *pickelObj(checkVarExistingAndDict(varName));
 718   KeyWaiter *ret0(new KeyWaiter(pickelObj,keyVal));
 719   CORBA::Object_var obj(ret0->activate());//KeyWaiter instance activated inside a multithread POA contrary to all of objects in SALOMESDS in single thread !
 720   //
 721   TransactionRemoveKeyInVarErrorIfNotAlreadyExisting *ret1(new TransactionRemoveKeyInVarErrorIfNotAlreadyExisting(this,varName,keyVal));
 722   CORBA::Object_var obj2(ret1->activate());
 723   transac=SALOME::Transaction::_narrow(obj2);
 724   //
 725   return SALOME::KeyWaiter::_narrow(obj);
 726 }
 727
 728 SALOME::ByteVec *DataScopeServerTransaction::waitForMonoThrRev(SALOME::KeyWaiter_ptr kw)
 729 {
 730   PortableServer::ServantBase *ret(0);
 731   try
 732     {
 733       ret=_poa_for_key_waiter->reference_to_servant(kw);
 734     }
 735   catch(...) { ret=0; }
 736   KeyWaiter *retc(dynamic_cast<KeyWaiter *>(ret));
 737   if(!retc)
 738     throw Exception("DataScopeServerTransaction::invokeMonoThr : internal error 1 !");
 739   retc->_remove_ref();
 740   return retc->waitForMonoThr();
 741 }
 742
 743 void DataScopeServerTransaction::atomicApply(const SALOME::ListOfTransaction& transactions)
 744 {
 745   std::size_t sz(transactions.length());
 746   if(sz==0)
 747     return ;
 748   std::vector< AutoServantPtr<Transaction> > transactionsCpp(sz);
 749   for(std::size_t i=0;i<sz;i++)
 750     {
 751       PortableServer::ServantBase *eltBase(0);
 752       Transaction *elt(0);
 753       try
 754         {
 755           eltBase=_poa->reference_to_servant(transactions[i]);
 756           elt=dynamic_cast<Transaction *>(eltBase);
 757         }
 758       catch(...)
 759         {
 760           std::ostringstream oss; oss << "DataScopeServerTransaction::atomicApply : the elt #" << i << " is invalid !";
 761           throw Exception(oss.str());
 762         }
 763       if(!elt)
 764         {
 765           std::ostringstream oss; oss << "DataScopeServerTransaction::atomicApply : the elt #" << i << " is null ?";
 766           throw Exception(oss.str());
 767         }
 768       elt->_remove_ref();
 769       transactionsCpp[i]=elt;
 770     }
 771   {// important do not merge loops !
 772     std::vector<TrustTransaction> transactions2(sz);
 773     bool mustRollback(true);
 774     for(std::size_t i=0;i<sz;i++)
 775       transactions2[i].setTransaction(transactionsCpp[i],&mustRollback);
 776     for(std::size_t i=0;i<sz;i++)
 777       transactions2[i].operate();
 778     mustRollback=false;
 779   }
 780   for(std::size_t i=0;i<sz;i++)
 781     transactionsCpp[i]->notify();
 782 }
 783
 784 /*!
 785  * Returns borrowed reference.
 786  */
 787 PyObject *DataScopeServerTransaction::getPyCmpFunc()
 788 {
 789   PyObject *builtins(PyDict_GetItemString(_globals,"__builtins__"));//borrowed
 790   if(builtins==NULL)
 791     throw Exception("Fail to find reference to builtins !");
 792   PyObject *builtins2(PyModule_GetDict(builtins));//borrowed
 793   if(builtins2==NULL)
 794     throw Exception("Fail to invoke __dict__ on builtins !");
 795   PyObject *cmpObj(PyDict_GetItemString(builtins2,"cmp"));
 796   if(cmpObj==NULL)
 797     throw Exception("Fail to find cmp in __builtins__ !");
 798   return cmpObj;
 799 }
 800
 801 DataScopeServerTransaction::~DataScopeServerTransaction()
 802 {
 803 }
 804