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)
