def _Cursor__send_message(self, *args, **kwargs):
# print self.__dict__
job = get_current_job()
if job:
subtype = "find"
collection = self._Cursor__collection.name # pylint: disable=no-member
if collection == "$cmd":
items = self._Cursor__spec.items() # pylint: disable=no-member
if len(items) > 0:
subtype, collection = items[0]
job.set_current_io({
"type": "mongodb.%s" % subtype,
"data": {
"collection":
"%s.%s" %
(self._Cursor__collection.database.name, collection) # pylint: disable=no-member
}
})
ret = Cursor._Cursor__send_message(self, *args, **kwargs) # pylint: disable=no-member
if job:
job.set_current_io(None)
return ret
def _Cursor__send_message(self, *args, **kwargs):
# print self.__dict__
job = get_current_job()
if job:
subtype = "cursor"
collection = self._Cursor__collection.name # pylint: disable=no-member
if collection == "$cmd":
items = self._Cursor__spec.items() # pylint: disable=no-member
if len(items) > 0:
subtype, collection = items[0]
job.set_current_io({
"type": "mongodb.%s" % subtype,
"data": {
"collection": "%s.%s" % (self._Cursor__collection.database.name, collection) # pylint: disable=no-member
}
})
ret = Cursor._Cursor__send_message(self, *args, **kwargs) # pylint: disable=no-member
if job:
job.set_current_io(None)
return ret
def _Cursor__send_message(self, *args, **kwargs):
# print self.__dict__
job = get_current_job()
if job:
subtype = "cursor"
collection = self._Cursor__collection.name # pylint: disable=no-member
if collection == "$cmd":
items = self._Cursor__spec.items() # pylint: disable=no-member
if len(items) > 0:
subtype, collection = items[0]
full_name = "%s.%s" % (self._Cursor__collection.database.name, collection) # pylint: disable=no-member
job.set_current_io({
"type": "mongodb.%s" % subtype,
"data": {
"collection": full_name
}
})
if config.get("mongodb_pre_hook"):
config.get("mongodb_pre_hook")({
"collection": full_name,
"method": subtype,
"args": (getattr(args[0], "spec", None), ),
"kwargs": kwargs,
"client": self._Cursor__collection.database.client,
"job": job
})
start_time = time.time()
ret = False
try:
ret = Cursor._Cursor__send_message(self, *args, **kwargs) # pylint: disable=no-member
finally:
stop_time = time.time()
if job:
job.set_current_io(None)
if config.get("mongodb_post_hook"):
config.get("mongodb_post_hook")({
"collection": full_name,
"method": subtype,
"args": (getattr(args[0], "spec", None), ),
"kwargs": kwargs,
"client": self._Cursor__collection.database.client,
"job": job,
"result": ret,
"time": stop_time - start_time
})
return ret
def _Cursor__send_message(self, *args, **kwargs):
# print self.__dict__
job = get_current_job()
if job:
subtype = "cursor"
collection = self._Cursor__collection.name # pylint: disable=no-member
if collection == "$cmd":
items = list(self._Cursor__spec.items()) # pylint: disable=no-member
if len(items) > 0:
subtype, collection = items[0]
full_name = "%s.%s" % (
self._Cursor__collection.database.name, collection) # pylint: disable=no-member
job.set_current_io({
"type": "mongodb.%s" % subtype,
"data": {
"collection": full_name
}
})
if config.get("mongodb_pre_hook"):
config.get("mongodb_pre_hook")({
"collection":
full_name,
"method":
subtype,
"args": (getattr(args[0], "spec", None), ),
"kwargs":
kwargs,
"client":
self._Cursor__collection.database.client,
"job":
job
})
start_time = time.time()
ret = False
try:
ret = Cursor._Cursor__send_message(self, *args, **kwargs) # pylint: disable=no-member
finally:
stop_time = time.time()
if job:
job.set_current_io(None)
if config.get("mongodb_post_hook"):
config.get("mongodb_post_hook")({
"collection":
full_name,
"method":
subtype,
"args": (getattr(args[0], "spec", None), ),
"kwargs":
kwargs,
"client":
self._Cursor__collection.database.client,
"job":
job,
"result":
ret,
"time":
stop_time - start_time
})
return ret
