class SpatialWorker(QueryWorker):
def __init__(self, workload_settings, target_settings, shutdown_event):
super(QueryWorker, self).__init__(workload_settings, target_settings,
shutdown_event)
self.total_workers = self.ws.spatial.workers
self.throughput = self.ws.spatial.throughput
self.name = 'spatial-worker'
self.new_queries = SpatialQueryFromFile(
workload_settings.spatial.queries,
workload_settings.spatial.dimensionality,
workload_settings.spatial.view_names,
workload_settings.spatial.params)
host, port = self.ts.node.split(':')
params = {'bucket': self.ts.bucket, 'host': host, 'port': port,
'username': self.ts.bucket, 'password': self.ts.password}
self.cb = SpatialGen(**params)
@with_sleep
def do_batch(self):
for i in xrange(self.BATCH_SIZE):
offset = self.curr_queries.value - self.BATCH_SIZE + i
ddoc_name, view_name, query = self.new_queries.next(offset)
self.cb.query(ddoc_name, view_name, query=query)
def __init__(self, workload_settings, target_settings, shutdown_event):
super(QueryWorker, self).__init__(workload_settings, target_settings,
shutdown_event)
self.total_workers = self.ws.spatial.workers
self.throughput = self.ws.spatial.throughput
self.name = 'spatial-worker'
self.new_queries = SpatialQueryFromFile(
workload_settings.spatial.queries,
workload_settings.spatial.dimensionality,
workload_settings.spatial.view_names,
workload_settings.spatial.params)
host, port = self.ts.node.split(':')
params = {'bucket': self.ts.bucket, 'host': host, 'port': port,
'username': self.ts.bucket, 'password': self.ts.password}
self.cb = SpatialGen(**params)
def __init__(self, settings, workload, spatial_settings, prefix=None):
super(SpringSpatialQueryLatency,
self).__init__(settings, workload, prefix)
self.offset = 0
view_names = self._parse_views(spatial_settings.indexes)
self.clients = []
for bucket in self.get_buckets():
client = SpatialGen(bucket=bucket,
host=settings.master_node,
username=bucket,
password=settings.bucket_password)
self.clients.append((bucket, client))
self.new_queries = SpatialQueryFromFile(
spatial_settings.queries, spatial_settings.dimensionality,
view_names, spatial_settings.params)
