rename planetstackanalytics
diff --git a/planetstack/hpc_wizard/xos_analytics.py b/planetstack/hpc_wizard/xos_analytics.py
new file mode 100644
index 0000000..a95d3ae
--- /dev/null
+++ b/planetstack/hpc_wizard/xos_analytics.py
@@ -0,0 +1,475 @@
+from bigquery_analytics import BigQueryAnalytics, BIGQUERY_AVAILABLE
+import datetime
+import re
+import os
+import sys
+import time
+import json
+import traceback
+import urllib2
+
+# XXX hardcoded path
+sys.path.append("/opt/xos")
+
+os.environ.setdefault("DJANGO_SETTINGS_MODULE", "xos.settings")
+from django.conf import settings
+from django import db
+from django.db import connection
+from core.models import Slice, Sliver, ServiceClass, Reservation, Tag, Network, User, Node, Image, Deployment, Site, NetworkTemplate, NetworkSlice, Service
+
+BLUE_LOAD=5000000
+RED_LOAD=15000000
+
+glo_cached_queries = {}
+
+class XOSAnalytics(BigQueryAnalytics):
+ def __init__(self, tableName=None):
+ if not tableName:
+ tableName = settings.BIGQUERY_TABLE
+
+ BigQueryAnalytics.__init__(self, tableName)
+
+ def service_to_sliceNames(self, serviceName):
+ service=Service.objects.get(name=serviceName)
+ try:
+ slices = service.slices.all()
+ except:
+ # BUG in data model -- Slice.service has related name 'service' and
+ # it should be 'slices'
+ slices = service.service.all()
+
+ return [slice.name for slice in slices]
+
+ def compose_query(self, filter={}, timeBucket="60", avg=[], sum=[], count=[], computed=[], val=[], groupBy=["Time"], orderBy=["Time"], tableName=None, latest=False, maxAge=60*60):
+ if tableName is None:
+ tableName = self.tableName
+
+ maxAge = maxAge * 1000
+ tablePart = "[%s.%s@-%d--1]" % ("vicci", tableName, maxAge)
+
+ fields = []
+ fieldNames = []
+ srcFieldNames = ["time"]
+
+ fields.append("SEC_TO_TIMESTAMP(INTEGER(TIMESTAMP_TO_SEC(time)/%s)*%s) as Time" % (str(timeBucket),str(timeBucket)))
+ #fields.append("INTEGER(TIMESTAMP_TO_SEC(time)/%s)*%s as Time" % (str(timeBucket),str(timeBucket)))
+
+ for fieldName in avg:
+ fields.append("AVG(%s) as avg_%s" % (fieldName, fieldName.replace("%","")))
+ fieldNames.append("avg_%s" % fieldName.replace("%",""))
+ srcFieldNames.append(fieldName)
+
+ for fieldName in sum:
+ fields.append("SUM(%s) as sum_%s" % (fieldName, fieldName.replace("%","")))
+ fieldNames.append("sum_%s" % fieldName.replace("%",""))
+ srcFieldNames.append(fieldName)
+
+ for fieldName in count:
+ fields.append("COUNT(distinct %s) as count_%s" % (fieldName, fieldName.replace("%","")))
+ fieldNames.append("count_%s" % fieldName.replace("%",""))
+ srcFieldNames.append(fieldName)
+
+ for fieldName in val:
+ fields.append(fieldName)
+ fieldNames.append(fieldName)
+ srcFieldNames.append(fieldName)
+
+ for fieldName in computed:
+ operator = "/"
+ parts = fieldName.split("/")
+ computedFieldName = "computed_" + parts[0].replace("%","")+"_div_"+parts[1].replace("%","")
+ if len(parts)==1:
+ operator = "*"
+ parts = computed.split("*")
+ computedFieldName = "computed_" + parts[0].replace("%","")+"_mult_"+parts[1].replace("%","")
+ fields.append("SUM(%s)%sSUM(%s) as %s" % (parts[0], operator, parts[1], computedFieldName))
+ fieldNames.append(computedFieldName)
+ srcFieldNames.append(parts[0])
+ srcFieldNames.append(parts[1])
+
+ for fieldName in groupBy:
+ if (fieldName not in ["Time"]):
+ fields.append(fieldName)
+ fieldNames.append(fieldName)
+ srcFieldNames.append(fieldName)
+
+ fields = ", ".join(fields)
+
+ where = []
+
+ if filter.get("slice",None):
+ where.append("%%slice='%s'" % filter["slice"])
+ if filter.get("site",None):
+ where.append("%%site='%s'" % filter["site"])
+ if filter.get("node",None):
+ where.append("%%hostname='%s'" % filter["node"])
+ if filter.get("event",None):
+ where.append("event='%s'" % filter["event"])
+ if filter.get("service",None):
+ sliceNames = self.service_to_sliceNames(filter["service"])
+ if sliceNames:
+ where.append("(" + " OR ".join(["%%slice='%s'" % sliceName for sliceName in sliceNames]) +")")
+
+ if where:
+ where = " WHERE " + " AND ".join(where)
+ else:
+ where =""
+
+ if groupBy:
+ groupBySub = " GROUP BY " + ",".join(groupBy + ["%hostname"])
+ groupBy = " GROUP BY " + ",".join(groupBy)
+ else:
+ groupBySub = " GROUP BY %hostname"
+ groupBy = ""
+
+ if orderBy:
+ orderBy = " ORDER BY " + ",".join(orderBy)
+ else:
+ orderBy = ""
+
+ if latest:
+ latestFields = ["table1.%s as %s" % (x,x) for x in srcFieldNames]
+ latestFields = ", ".join(latestFields)
+ tablePart = """(SELECT %s FROM %s AS table1
+ JOIN
+ (SELECT %%hostname, event, max(time) as maxtime from %s GROUP BY %%hostname, event) AS latest
+ ON
+ table1.%%hostname = latest.%%hostname AND table1.event = latest.event AND table1.time = latest.maxtime)""" % (latestFields, tablePart, tablePart)
+
+ if computed:
+ subQuery = "SELECT %%hostname, %s FROM %s" % (fields, tablePart)
+ if where:
+ subQuery = subQuery + where
+ subQuery = subQuery + groupBySub
+
+ sumFields = []
+ for fieldName in fieldNames:
+ if fieldName.startswith("avg"):
+ sumFields.append("AVG(%s) as avg_%s"%(fieldName,fieldName))
+ sumFields.append("MAX(%s) as max_%s"%(fieldName,fieldName))
+ elif (fieldName.startswith("count")) or (fieldName.startswith("sum")) or (fieldName.startswith("computed")):
+ sumFields.append("SUM(%s) as sum_%s"%(fieldName,fieldName))
+ else:
+ sumFields.append(fieldName)
+
+ sumFields = ",".join(sumFields)
+
+ query = "SELECT %s, %s FROM (%s)" % ("Time", sumFields, subQuery)
+ if groupBy:
+ query = query + groupBy
+ if orderBy:
+ query = query + orderBy
+ else:
+ query = "SELECT %s FROM %s" % (fields, tablePart)
+ if where:
+ query = query + " " + where
+ if groupBy:
+ query = query + groupBy
+ if orderBy:
+ query = query + orderBy
+
+ return query
+
+ def get_list_from_req(self, req, name, default=[]):
+ value = req.GET.get(name, None)
+ if not value:
+ return default
+ value=value.replace("@","%")
+ return value.split(",")
+
+ def format_result(self, format, result, query, dataSourceUrl):
+ if not BIGQUERY_AVAILABLE:
+ msg = "BigQuery Statistics Unavaiable"
+ else:
+ msg = None
+
+ if (format == "json_dicts"):
+ result = {"query": query, "rows": result, "dataSourceUrl": dataSourceUrl, "msg": msg}
+ return ("application/javascript", json.dumps(result))
+
+ elif (format == "json_arrays"):
+ new_result = []
+ for row in result:
+ new_row = []
+ for key in sorted(row.keys()):
+ new_row.append(row[key])
+ new_result.append(new_row)
+ new_result = {"query": query, "rows": new_result, "msg": msg}
+ return ("application/javascript", json.dumps(new_result))
+
+ elif (format == "html_table"):
+ new_rows = []
+ for row in result:
+ new_row = []
+ for key in sorted(row.keys()):
+ new_row.append("<TD>%s</TD>" % str(row[key]))
+ new_rows.append("<TR>%s</TR>" % "".join(new_row))
+
+ new_result = "<TABLE>%s</TABLE>" % "\n".join(new_rows)
+
+ return ("text/html", new_result)
+
+ def merge_datamodel_sites(self, rows, slice=None):
+ """ For a query that included "site" in its groupby, merge in the
+ opencloud site information.
+ """
+
+ if slice:
+ try:
+ slice = Slice.objects.get(name=slice)
+ except:
+ slice = None
+
+ for row in rows:
+ sitename = row["site"]
+ try:
+ model_site = Site.objects.get(name=sitename)
+ except:
+ # we didn't find it in the data model
+ continue
+
+ allocated_slivers = 0
+ if model_site and slice:
+ for sliver in slice.slivers.all():
+ if sliver.node.site == model_site:
+ allocated_slivers = allocated_slivers + 1
+
+ row["lat"] = float(model_site.location.latitude)
+ row["long"] = float(model_site.location.longitude)
+ row["url"] = model_site.site_url
+ row["numNodes"] = model_site.nodes.count()
+ row["allocated_slivers"] = allocated_slivers
+
+ max_cpu = row.get("max_avg_cpu", row.get("max_cpu",0))
+ cpu=float(max_cpu)/100.0
+ row["hotness"] = max(0.0, ((cpu*RED_LOAD) - BLUE_LOAD)/(RED_LOAD-BLUE_LOAD))
+
+ def compose_cached_query(self, querySpec='default'):
+ """ Compose a query that returns the 'most recent' row for each (hostname, event)
+ pair.
+
+ Note that groupByFields cannot contain any values that are 'Null' or those
+ rows will be excluded. For example, if groupByFields includes cp, then
+ there will be no libvirt_event rows, since libvirt_event does not have
+ cp.
+
+ This means we can't really have 'one query to rule them'. Settle on
+ having a couple of different queries, and have the caller specify
+ which one he wants.
+ """
+
+ fieldNames = ["%hostname", "%bytes_sent", "%bytes_hit", "%healthy", "time", "event", "%site", "%elapsed", "%cpu"]
+
+ if querySpec=="default":
+ groupByFields = ["%hostname", "event"]
+ elif (querySpec=="hpc"):
+ fieldNames.append("%cp")
+ groupByFields = ["%hostname", "event", "%cp"]
+ else:
+ raise ValueError("Unknown queryspec %s" % querySpec)
+
+ fields = ["table1.%s AS %s" % (x,x) for x in fieldNames]
+ fields = ", ".join(fields)
+
+ tableDesc = "%s.%s" % (self.projectName, self.tableName)
+
+ groupByOn = ["table1.time = latest.maxtime"]
+ for field in groupByFields:
+ groupByOn.append("table1.%s = latest.%s" % (field, field))
+
+ groupByOn = " AND ".join(groupByOn)
+ groupByFields = ", ".join(groupByFields)
+
+ base_query = "SELECT %s FROM [%s@-3600000--1] AS table1 JOIN (SELECT %s, max(time) as maxtime from [%s@-3600000--1] GROUP BY %s) AS latest ON %s" % \
+ (fields, tableDesc, groupByFields, tableDesc, groupByFields, groupByOn)
+
+ return base_query
+
+ def get_cached_query_results(self, q, wait=True):
+ global glo_cached_queries
+
+ if q in glo_cached_queries:
+ if (time.time() - glo_cached_queries[q]["time"]) <= 60:
+ print "using cached query"
+ return glo_cached_queries[q]["rows"]
+
+ if not wait:
+ return None
+
+ print "refreshing cached query"
+ result = self.run_query(q)
+ glo_cached_queries[q] = {"time": time.time(), "rows": result}
+
+ return result
+
+ def process_request(self, req):
+ print req.GET
+
+ tqx = req.GET.get("tqx", None)
+
+ slice = req.GET.get("slice", None)
+ site = req.GET.get("site", None)
+ node = req.GET.get("node", None)
+ service = req.GET.get("service", None)
+ event = req.GET.get("event", "libvirt_heartbeat")
+ cp = req.GET.get("cp", None)
+
+ format = req.GET.get("format", "json_dicts")
+
+ timeBucket = int(req.GET.get("timeBucket", 60))
+ avg = self.get_list_from_req(req, "avg")
+ sum = self.get_list_from_req(req, "sum")
+ count = self.get_list_from_req(req, "count")
+ computed = self.get_list_from_req(req, "computed")
+ groupBy = self.get_list_from_req(req, "groupBy", ["Time"])
+ orderBy = self.get_list_from_req(req, "orderBy", ["Time"])
+
+ maxRows = req.GET.get("maxRows", None)
+ mergeDataModelSites = req.GET.get("mergeDataModelSites", None)
+
+ maxAge = int(req.GET.get("maxAge", 60*60))
+
+ cached = req.GET.get("cached", None)
+ cachedGroupBy = self.get_list_from_req(req, "cachedGroupBy", ["doesnotexist"])
+
+ filter={}
+ if slice:
+ filter["slice"] = slice
+ if site:
+ filter["site"] = site
+ if node:
+ filter["hostname"] = node
+ if event:
+ filter["event"] = event
+ if cp:
+ filter["cp"] = cp
+
+ q = self.compose_query(filter, timeBucket, avg, sum, count, computed, [], groupBy, orderBy, maxAge=maxAge)
+
+ print q
+
+ dataSourceUrl = "http://" + req.META["SERVER_NAME"] + ":" + req.META["SERVER_PORT"] + req.META["PATH_INFO"] + "?" + req.META["QUERY_STRING"].replace("format=","origFormat=").replace("%","%25") + "&format=charts";
+
+ if (format=="dataSourceUrl"):
+ result = {"dataSourceUrl": dataSourceUrl}
+ return ("application/javascript", result)
+
+ elif (format=="raw"):
+ result = self.run_query_raw(q)
+ result["dataSourceUrl"] = dataSourceUrl
+
+ result = json.dumps(result);
+
+ return ("application/javascript", result)
+
+ elif (format=="nodata"):
+ result = {"dataSourceUrl": dataSourceUrl, "query": q}
+ result = json.dumps(result);
+ return {"application/javascript", result}
+
+ elif (format=="charts"):
+ bq_result = self.run_query_raw(q)
+
+ # cloudscrutiny code is probably better!
+ table = {}
+ table["cols"] = self.schema_to_cols(bq_result["schema"])
+ rows = []
+ if "rows" in bq_result:
+ for row in bq_result["rows"]:
+ rowcols = []
+ for (colnum,col) in enumerate(row["f"]):
+ if (colnum==0):
+ dt = datetime.datetime.fromtimestamp(float(col["v"]))
+ rowcols.append({"v": 'new Date("%s")' % dt.isoformat()})
+ else:
+ try:
+ rowcols.append({"v": float(col["v"])})
+ except:
+ rowcols.append({"v": col["v"]})
+ rows.append({"c": rowcols})
+ table["rows"] = rows
+
+ if tqx:
+ reqId = tqx.strip("reqId:")
+ else:
+ reqId = "0"
+
+ result = {"status": "okColumnChart", "reqId": reqId, "table": table, "version": "0.6"}
+
+ result = "google.visualization.Query.setResponse(" + json.dumps(result) + ");"
+
+ def unquote_it(x): return x.group()[1:-1].replace('\\"', '"')
+
+ p = re.compile(r'"new Date\(\\"[^"]*\\"\)"')
+ result=p.sub(unquote_it, result)
+
+ return ("application/javascript", result)
+
+ else:
+ if cached:
+ results = self.get_cached_query_results(self.compose_cached_query(cached))
+
+ result = self.postprocess_results(results, filter=filter, sum=sum, count=count, avg=avg, computed=computed, maxDeltaTime=120, groupBy=cachedGroupBy)
+ else:
+ result = self.run_query(q)
+
+ if maxRows:
+ result = result[-int(maxRows):]
+
+ if mergeDataModelSites:
+ self.merge_datamodel_sites(result)
+
+ return self.format_result(format, result, q, dataSourceUrl)
+
+def DoXOSAnalytics(request):
+ bq = XOSAnalytics()
+ result = bq.process_request(request)
+
+ return result
+
+def main():
+ bq = XOSAnalytics(tableName="demoevents")
+
+ q = bq.compose_cached_query()
+ results = bq.run_query(q)
+
+ #results = bq.postprocess_results(results,
+ # filter={"slice": "HyperCache"},
+ # groupBy=["site"],
+ # computed=["bytes_sent/elapsed"],
+ # sum=["bytes_sent", "computed_bytes_sent_div_elapsed"], avg=["cpu"],
+ # maxDeltaTime=60)
+
+ #results = bq.postprocess_results(results, filter={"slice": "HyperCache"}, maxi=["cpu"], count=["hostname"], computed=["bytes_sent/elapsed"], groupBy=["Time", "site"], maxDeltaTime=80)
+
+ results = bq.postprocess_results(results,filter={"event": "libvirt_heartbeat"}, avg=["cpu"], count=["hostname"], groupBy=["doesnotexist"])
+
+ bq.dump_table(results)
+
+ sys.exit(0)
+
+ q=bq.compose_query(sum=["%bytes_sent"], avg=["%cpu"], latest=True, groupBy=["Time", "%site"])
+ print q
+ bq.dump_table(bq.run_query(q))
+
+ q=bq.compose_query(avg=["%cpu","%bandwidth"], count=["%hostname"], slice="HyperCache")
+ print q
+ bq.dump_table(bq.run_query(q))
+
+ q=bq.compose_query(computed=["%bytes_sent/%elapsed"])
+ print
+ print q
+ bq.dump_table(bq.run_query(q))
+
+ q=bq.compose_query(timeBucket=60*60, avg=["%cpu"], count=["%hostname"], computed=["%bytes_sent/%elapsed"])
+ print
+ print q
+ bq.dump_table(bq.run_query(q))
+
+if __name__ == "__main__":
+ main()
+
+
+
+
+