from datetime import datetime import calendar import csv client = RiakClient(host=sys.argv[1], pb_port=8087) table_name = sys.argv[2] # Create table print "Creating table..." print "Schema: CREATE TABLE " + table_name + " (status varchar not null, extid varchar not null, ts timestamp not null, avgMeasuredTime sint64 not null, avgSpeed sint64 not null, medianMeasuredTime sint64 not null, vehicleCount sint64 not null, PRIMARY KEY ((status, extid, quantum(ts, 30, 'd')), status, extid, ts))" time1 = int(round(time.time() * 1000)) try: print "Create table return: ", str( client.ts_query( table_name, "CREATE TABLE {table} (status varchar not null, extid varchar not null, ts timestamp not null, avgMeasuredTime sint64 not null, avgSpeed sint64 not null, medianMeasuredTime sint64 not null, vehicleCount sint64 not null, PRIMARY KEY ((status, extid, quantum(ts, 30, 'd')), status, extid, ts))" )) except RiakError as e: print "Table " + table_name + " already exists." time2 = int(round(time.time() * 1000)) print "Time elapsed (ms): ", str(time2 - time1) print "\n" table = client.table(table_name) def changetime(stime): dt = datetime.strptime(stime, '%Y-%m-%dT%H:%M:%S') #print dt return calendar.timegm(datetime.timetuple(dt)) * 1000
from riak.client import RiakClient from riak.riak_error import RiakError import sys import time client = RiakClient(host=sys.argv[1], pb_port=8087) table_name = sys.argv[2] table = client.table(table_name) query = "SELECT COUNT(*) FROM {table} where ts > 1392249600000 and ts < 1397347199000 and status='OK' and extid='668'" print "Query: ", query time1 = int(round(time.time() * 1000)) ts_obj = client.ts_query(table_name, query) time2 = int(round(time.time() * 1000)) print ts_obj.rows print "Time elapsed (ms): ", str(time2 - time1) print "\n" query = "SELECT * FROM {table} where ts > 1392249600000 and ts < 1397347199000 and status='OK' and extid='668'" print "Query: ", query time1 = int(round(time.time() * 1000)) ts_obj = client.ts_query(table_name, query) time2 = int(round(time.time() * 1000)) print ts_obj.rows print "Time elapsed (ms): ", str(time2 - time1) print "\n" query = "SELECT MAX(avgSpeed) FROM {table} where ts > 1392249600000 and ts < 1397347199000 and status='OK' and extid='668'" print "Query: ", query time1 = int(round(time.time() * 1000)) ts_obj = client.ts_query(table_name, query)
import os import time from datetime import datetime import calendar import csv client = RiakClient(host=sys.argv[1], pb_port=8087) table_name = sys.argv[2] # Create table print "Creating table..." print "Schema: CREATE TABLE " + table_name + " (status varchar not null, extid varchar not null, ts timestamp not null, avgMeasuredTime sint64 not null, avgSpeed sint64 not null, medianMeasuredTime sint64 not null, vehicleCount sint64 not null, PRIMARY KEY ((status, extid, quantum(ts, 30, 'd')), status, extid, ts))" time1 = int(round(time.time() * 1000)) try: print "Create table return: ", str(client.ts_query(table_name, "CREATE TABLE {table} (status varchar not null, extid varchar not null, ts timestamp not null, avgMeasuredTime sint64 not null, avgSpeed sint64 not null, medianMeasuredTime sint64 not null, vehicleCount sint64 not null, PRIMARY KEY ((status, extid, quantum(ts, 30, 'd')), status, extid, ts))")) except RiakError as e: print "Table " + table_name + " already exists." time2 = int(round(time.time() * 1000)) print "Time elapsed (ms): ", str(time2 - time1) print "\n" table = client.table(table_name) def changetime(stime): dt=datetime.strptime(stime,'%Y-%m-%dT%H:%M:%S') #print dt return calendar.timegm(datetime.timetuple(dt))*1000 totalcount=0 batchcount=0