def initialize_pilot_store(self): if self.pilot_store_description!=None: self.service_url=self.pilot_store_description["service_url"] self.size = self.pilot_store_description["size"] self.pilot_store_description = self.pilot_store_description # initialize file adaptor if self.service_url.startswith("ssh:"): logger.debug("Use SSH backend") self.__filemanager = SSHFileAdaptor(self.service_url) elif self.service_url.startswith("http:"): logger.debug("Use WebHDFS backend") self.__filemanager = WebHDFSFileAdaptor(self.service_url) self.__filemanager.initialize_pilotstore() self.__filemanager.get_pilotstore_size()
class PilotStore(PilotStore): """ TROY PilotStore. """ PS_ID_PREFIX="ps-" def __init__(self, pilot_store_service=None, pilot_store_description=None, ps_url=None): """ Initialize PilotStore at given service url: ssh://<hostname> gsissh://<hostname> Currently only ssh schemes are supported. In the future all SAGA URL schemes/adaptors should be supported. """ self.id = None self.url = None self.pilot_store_description = None self.service_url=None self.size = None self.pilot_store_description = None self.pilot_data={} if ps_url==None and pilot_store_service!=None: # new ps self.id = self.PS_ID_PREFIX+str(uuid.uuid1()) self.pilot_store_description = pilot_store_description self.url = CoordinationAdaptor.add_ps(CoordinationAdaptor.get_base_url(bigdata.application_id)+"/"+pilot_store_service.id, self) elif ps_url != None: logger.warn("Reconnect to PilotStore: %s"%ps_url) dictionary = CoordinationAdaptor.get_ps(ps_url) ps_dict = dictionary["pilot_store"] for i in ps_dict: self.__setattr__(i, ps_dict[i]) self.initialize_pilot_store() def initialize_pilot_store(self): if self.pilot_store_description!=None: self.service_url=self.pilot_store_description["service_url"] self.size = self.pilot_store_description["size"] self.pilot_store_description = self.pilot_store_description # initialize file adaptor if self.service_url.startswith("ssh:"): logger.debug("Use SSH backend") self.__filemanager = SSHFileAdaptor(self.service_url) elif self.service_url.startswith("http:"): logger.debug("Use WebHDFS backend") self.__filemanager = WebHDFSFileAdaptor(self.service_url) self.__filemanager.initialize_pilotstore() self.__filemanager.get_pilotstore_size() def __get_ps_id(self, ps_url): start = ps_url.index(self.PS_ID_PREFIX) end =ps_url.index("/", start) return ps_url[start:end] def cancel(self): """ Cancel PilotStore Keyword arguments: None """ self.__filemanager.delete_pilotstore() def url_for_pd(self, pd): if self.pilot_data.has_key(pd.id): return self.service_url + "/" + str(pd.id) return None def create_pd(self, pd): self.__filemanager.create_pd(pd.id) def put_pd(self, pd): logging.debug("Put PD: %s to PS: %s"%(pd.id,self.service_url)) self.__filemanager.create_pd(pd.id) self.__filemanager.put_pd(pd) self.pilot_data[pd.id] = pd CoordinationAdaptor.update_ps(self) def remove_pd(self, pd): """ Remove pilot data from pilot store """ if self.pilot_data.has_key(pd.id): self.__filemanager.remove_pd(pd) del self.pilot_data[pd.id] CoordinationAdaptor.update_ps(self) def copy_pd(self, pd, ps_new): ps_new.create_pd(pd) self.__filemanager.copy_pd(pd, ps_new) # update meta data at ps_new ps_new.pilot_data[pd.id] = pd CoordinationAdaptor.update_ps(ps_new) def list_pilotdata(self): return self.pilot_data.values() def get_state(self): return self.__filemanager.get_state() def export_pd(self, pd, target_url): self.__filemanager.get_pd(pd, target_url) def to_dict(self): ps_dict = {} ps_dict["id"]=self.id ps_dict["url"]=self.url ps_dict["pilot_store_description"]=self.pilot_store_description logger.debug("PS Dictionary: " + str(ps_dict)) return ps_dict def __repr__(self): return self.service_url @classmethod def create_pilot_store_from_dict(cls, ps_dict): ps = PilotStore() for i in ps_dict.keys(): ps.__setattr__(i, ps_dict[i]) ps.initialize_pilot_store() logger.debug("created ps " + str(ps)) return ps