def query_range(self, field, from_, to): # query is performed by views try: return self.bucket.view("_design/%s/%s" % (self.view_doc, field), start_key=from_, end_key=to) except Exception, e: self.log("Query can't be run, missing view ?....") raise interface.FunctionNotImplemented() def reset(self): """ Just delete all documents that has been created """ for id in range(self._first_id, self._next_id): self.bucket.delete(str(id)) self._first_id = self._next_id = 1 def disk_usage(self): raise interface.FunctionNotImplemented() def memory_usage(self): raise interface.FunctionNotImplemented() interface.implements("CouchBaseDb", CouchBaseDb)
self.collection.documents.update(id, fields) def get(self, id): self.log("get method is not implemented, going to all documents") for doc in self.collection.documents(): if doc.id == id: return doc def query(self, field, value): raise interface.FunctionNotImplemented() def query_range(self, field, from_, to): raise interface.FunctionNotImplemented() def reset(self): """ Reset on ArangoDb means delete and create collection """ self.collection.delete() self.collection.create(waitForSync=True) def disk_usage(self): raise interface.FunctionNotImplemented() def memory_usage(self): raise interface.FunctionNotImplemented() interface.implements("ArangoDb", ArangoDb)
def query_range(self, field, from_, to): docs = self.collection.find({field: { "$gte" : from_, "$lte" : to}}) l_docs = [ x for x in docs] return l_docs def reset(self): """ Reset on MongoDB means delete and create collection """ self.connection[self.database_name].drop_collection(self.collection_name) self.connection[self.database_name].create_collection(self.collection_name) self.collection = self.connection[self.database_name][self.collection_name] def disk_usage(self): data = self.connection[self.database_name].command("dbstats") storage_size = int(data['storageSize']) / 1024 / 1024 return storage_size def memory_usage(self): data = self.connection.admin.command(son.SON([('serverStatus', 1)])) if self.mem_type_for_profile == "resident": return int(data['mem']['resident']) else: return int(data['mem']['virtual']) interface.implements("MongoDb", MongoDb)