bigtable

import os
import time
import json
from google.cloud import bigtable
from google.cloud import happybase

os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "/Users/jlh/Desktop/bigdata-group.json"


class status(object):
    def __init__(self):
        self.client = bigtable.Client(project="heidao-market", admin=True)
        self.instance = self.client.instance("yotta-bigtable")
        self.connection = happybase.Connection(instance=self.instance)
        self.table = self.connection.table("mafia1_SnapShot_Status_result_test_solar")
        pass

    def get_scan_list(self, player_id, player_time):
        # end_time = round(time.time() * 1000)
        end_time = player_time
        start_time = end_time - 86400000
        player_id = player_id
        row_key = str(player_id)[::-1]
        row = self.table.row(row_key)
        result = {}
        if len(row) == 0:
            return result
        last_vals = {}
        last_times = {}
        for i in range(len(row)):
            col = bytes("cf1:status_" + str(i), encoding="utf8")
            if col not in row.keys():
                continue
            row_dic = json.loads(row[col])
            time_str = row_dic["time"]
            time_list = time_str.split('#')
            for key in row_dic:
                if key != "time":
                    val = row_dic[key]
                    val_list = val.split("#")
                    sums = 0.0
                    last_val = float(val_list[0])
                    last_time = start_time
                    if key in last_vals:
                        last_val = last_vals[key]
                    if key in last_times:
                        last_time = last_times[key]
                    for j in range(len(time_list)):
                        cur_time = int(time_list[j])
                        if cur_time >= last_time and cur_time <= end_time:
                            sums += (cur_time - last_time) * last_val
                            last_time = cur_time
                        if cur_time <= end_time:
                            last_val = float(val_list[j])
                    last_vals[key] = last_val
                    last_times[key] = last_time
                    if key in result:
                        result[key] = result[key] + sums
                    else:
                        result[key] = sums
                    if i == len(row) - 1:
                        result[key] = result[key] + (end_time - last_time) * last_val
                        result[key] = result[key] / 86400000
        return result

    # 根据时间寻找状态
    def get_status_by_time(self, player_id, player_time):
        row_key = str(player_id)[::-1]
        end_time = player_time
        row = self.table.row(row=row_key)
        result = {}
        if len(row) == 0:
            return result
        for i in range(len(row)):
            col = bytes("cf1:status_" + str(i), encoding="utf8")
            if col not in row.keys():
                continue
            row_dic = json.loads(row[col])
            time_str = row_dic["time"]
            time_list = time_str.split('#')
            index = -1
            for j in range(len(time_list)):
                if int(time_list[j]) >= end_time:
                    break
                index = j
            if index == -1:
                return result
            for key in row_dic:
                val = row_dic[key]
                val_list = val.split('#')
                result[key] = val_list[index]
        return result


if __name__ == "__main__":


    status = status()
    # dic = status.get_scan_list(1014223004)
    # print(dic)
    start = time.time()
    dic = status.get_scan_list(1087110896, 1583810937000)
    print(dic)

    dic = status.get_status_by_time(1087110896, 1683810937000)
    print(dic)
    stop = time.time()
    print(stop - start)