def stringReceived(self, rawRequest):
request = self.unpickler.loads(rawRequest)
if request['type'] == 'cache-query':
metric = request['metric']
datapoints = MetricCache.get(metric, [])
result = dict(datapoints=datapoints)
if settings.LOG_CACHE_HITS:
log.query('[%s] cache query for \"%s\" returned %d values' % (self.peerAddr, metric, len(datapoints)))
instrumentation.increment('cacheQueries')
elif request['type'] == 'cache-query-bulk':
datapointsByMetric = {}
metrics = request['metrics']
for metric in metrics:
datapointsByMetric[metric] = MetricCache.get(metric, [])
result = dict(datapointsByMetric=datapointsByMetric)
if settings.LOG_CACHE_HITS:
log.query('[%s] cache query bulk for \"%d\" metrics returned %d values' %
(self.peerAddr, len(metrics), sum([len(datapoints) for datapoints in datapointsByMetric.values()])))
instrumentation.increment('cacheBulkQueries')
instrumentation.append('cacheBulkQuerySize', len(metrics))
elif request['type'] == 'get-metadata':
result = management.getMetadata(request['metric'], request['key'])
elif request['type'] == 'set-metadata':
result = management.setMetadata(request['metric'], request['key'], request['value'])
else:
result = dict(error="Invalid request type \"%s\"" % request['type'])
response = pickle.dumps(result, protocol=-1)
self.sendString(response)
def stringReceived(self, metric):
values = MetricCache.get(metric, [])
log.query('cache query for %s returned %d values' %
(metric, len(values)))
response = pickle.dumps(values, protocol=-1)
self.sendString(response)
increment('cacheQueries')
def stringReceived(self, rawRequest):
request = self.unpickler.loads(rawRequest)
if request['type'] == 'cache-query':
metric = request['metric']
datapoints = MetricCache.get(metric, {}).items()
result = dict(datapoints=datapoints)
if settings.LOG_CACHE_HITS:
log.query('[%s] cache query for \"%s\" returned %d values' % (self.peerAddr, metric, len(datapoints)))
instrumentation.increment('cacheQueries')
elif request['type'] == 'cache-query-bulk':
datapointsByMetric = {}
metrics = request['metrics']
for metric in metrics:
datapointsByMetric[metric] = MetricCache.get(metric, {}).items()
result = dict(datapointsByMetric=datapointsByMetric)
if settings.LOG_CACHE_HITS:
log.query('[%s] cache query bulk for \"%d\" metrics returned %d values' %
(self.peerAddr, len(metrics), sum([len(datapoints) for datapoints in datapointsByMetric.values()])))
instrumentation.increment('cacheBulkQueries')
instrumentation.append('cacheBulkQuerySize', len(metrics))
elif request['type'] == 'get-metadata':
result = management.getMetadata(request['metric'], request['key'])
elif request['type'] == 'set-metadata':
result = management.setMetadata(request['metric'], request['key'], request['value'])
else:
result = dict(error="Invalid request type \"%s\"" % request['type'])
response = pickle.dumps(result, protocol=-1)
self.sendString(response)
def stringReceived(self, rawRequest):
request = pickle.loads(rawRequest)
if request['type'] == 'cache-query':
metric = request['metric']
datapoints = MetricCache.get(metric, [])
result = dict(datapoints=datapoints)
log.query('[%s] cache query for \"%s\" returned %d values' % (self.peerAddr, metric, len(datapoints)))
instrumentation.increment('cacheQueries')
elif request['type'] == 'get-metadata':
result = management.getMetadata(request['metric'], request['key'])
elif request['type'] == 'set-metadata':
result = management.setMetadata(request['metric'], request['key'], request['value'])
else:
result = dict(error="Invalid request type \"%s\"" % request['type'])
response = pickle.dumps(result, protocol=-1)
self.sendString(response)
def stringReceived(self, rawRequest):
request = self.unpickler.loads(rawRequest)
if request['type'] == 'cache-query':
metric = request['metric']
datapoints = MetricCache.get(metric, [])
result = dict(datapoints=datapoints)
log.query('[%s] cache query for \"%s\" returned %d values' % (self.peerAddr, metric, len(datapoints)))
instrumentation.increment('cacheQueries')
elif request['type'] == 'get-metadata':
result = management.getMetadata(request['metric'], request['key'])
elif request['type'] == 'set-metadata':
result = management.setMetadata(request['metric'], request['key'], request['value'])
else:
result = dict(error="Invalid request type \"%s\"" % request['type'])
response = pickle.dumps(result, protocol=-1)
self.sendString(response)
def stringReceived(self, rawRequest):
request = self.unpickler.loads(rawRequest)
if request["type"] == "cache-query":
metric = request["metric"]
datapoints = MetricCache.get(metric, [])
result = dict(datapoints=datapoints)
if settings.LOG_CACHE_HITS:
log.query('[%s] cache query for "%s" returned %d values' % (self.peerAddr, metric, len(datapoints)))
instrumentation.increment("cacheQueries")
elif request["type"] == "cache-query-bulk":
datapointsByMetric = {}
metrics = request["metrics"]
for metric in metrics:
datapointsByMetric[metric] = MetricCache.get(metric, [])
result = dict(datapointsByMetric=datapointsByMetric)
if settings.LOG_CACHE_HITS:
log.query(
'[%s] cache query bulk for "%d" metrics returned %d values'
% (
self.peerAddr,
len(metrics),
sum([len(datapoints) for datapoints in datapointsByMetric.values()]),
)
)
instrumentation.increment("cacheBulkQueries")
instrumentation.append("cacheBulkQuerySize", len(metrics))
elif request["type"] == "get-metadata":
result = management.getMetadata(request["metric"], request["key"])
elif request["type"] == "set-metadata":
result = management.setMetadata(request["metric"], request["key"], request["value"])
else:
result = dict(error='Invalid request type "%s"' % request["type"])
response = pickle.dumps(result, protocol=-1)
self.sendString(response)
def lineReceived(self, rawRequest):
log.query('[%s] query received: %s' % (self.peerAddr, rawRequest))
request_type, request = rawRequest.split('|', 1)
if request_type == 'cache-query':
datapoints = MetricCache.get(request, [])
result = dict(datapoints=datapoints)
if settings.LOG_CACHE_HITS:
log.query('[%s] cache query for \"%s\" returned %d values' % (self.peerAddr, request, len(datapoints)))
instrumentation.increment('cacheQueries')
elif request_type == 'cache-query-bulk':
datapointsByMetric = {}
metrics = request.split('|')
for metric in metrics:
datapointsByMetric[metric] = MetricCache.get(metric, [])
result = dict(datapointsByMetric=datapointsByMetric)
if settings.LOG_CACHE_HITS:
log.query('[%s] cache query bulk for \"%d\" metrics returned %d values' %
(self.peerAddr, len(metrics), sum([len(datapoints) for datapoints in datapointsByMetric.values()])))
instrumentation.increment('cacheBulkQueries')
instrumentation.append('cacheBulkQuerySize', len(metrics))
elif request_type == 'get-metadata':
metric, key = request.split('|')
result = management.getMetadata(metric, key)
else:
result = dict(error="Invalid request type \"%s\"" % request_type)
self.sendLine(json.dumps(result))
def connectionLost(self, reason):
if reason.check(ConnectionDone):
log.query("%s disconnected" % self.peerAddr)
else:
log.query("%s connection lost: %s" % (self.peerAddr, reason.value))
def connectionMade(self):
peer = self.transport.getPeer()
self.peerAddr = "%s:%d" % (peer.host, peer.port)
log.query("%s connected" % self.peerAddr)
self.unpickler = get_unpickler(
insecure=settings.USE_INSECURE_UNPICKLER)
def connectionLost(self, reason):
if reason.check(ConnectionDone):
log.query("%s disconnected" % self.peerAddr)
else:
log.query("%s connection lost: %s" % (self.peerAddr, reason.value))
def connectionMade(self):
peer = self.transport.getPeer()
self.peerAddr = "%s:%d" % (peer.host, peer.port)
log.query("%s connected" % self.peerAddr)
self.unpickler = get_unpickler(insecure=settings.USE_INSECURE_UNPICKLER)
def stringReceived(self, metric):
values = MetricCache.get(metric, [])
log.query('cache query for %s returned %d values' % (metric, len(values)))
response = pickle.dumps(values, protocol=-1)
self.sendString(response)
increment('cacheQueries')
def connectionMade(self):
peer = self.transport.getPeer()
self.peerAddr = "%s:%d" % (peer.host, peer.port)
log.query("%s connected" % self.peerAddr)
