from cassandra.cluster import Cluster from cassandra.query import BatchStatement import time import hashlib from cassandra.auth import PlainTextAuthProvider import base64 from multiprocessing import Process from multiprocessing import Pool def sha256(data): sha256 = hashlib.sha256() sha256.update(data.encode('utf-8')) return sha256.hexdigest() def execute(): # 定义用户名和密码 username = 'tsp_eu_stress' password = 'UaiKeq6cKsm2S4y' # 创建身份验证提供程序 auth_provider = PlainTextAuthProvider(username=username, password=password) # 连接到 Cassandra 集群 cluster = Cluster(['10.95.0.81'], auth_provider=auth_provider) session = cluster.connect() # # 创建Cassandra集群连接 # cluster = Cluster(['172.25.116.188']) session = cluster.connect('tsp_eu') # 创建批量操作语句 # batch = BatchStatement() query2 = "insert into tsp_eu.realtime_his(v,ct,data,st,type) values(%s, %s,%s, %s,%s)" query3 = "insert into tsp_eu.realtime_hf(v,ct,data,st,type) values(%s, %s,%s, %s,%s)" #j为每个vin插入条数；i为插入的vin数 for j in range(1): for i in range(1,1000001): # time.sleep(1) # 创建插入语句 # tbox登陆登出插入语句 # query1 = "INSERT INTO tbox_login_logout (v,st,ct,reason,result,tbox_sn,type) VALUES (%s, %s,%s, %s,%s, %s,%s)" # batch = BatchStatement() # batch1 = BatchStatement() # batch.add(query1, ( # sha256('TESTVIN' + str(i).zfill(10)), int(time.time() * 1000), int(time.time() * 1000), "1", "1", "1111", # "2")) # print('TESTVIN' + str(i).zfill(10)) # session.execute(batch) # # # # IHU车机登陆登出插入语句 # query2 = "insert into tsp_eu.hu_login_logout(v,st,ct,hu_sn,status,status_msg,type) values(%s, %s,%s, %s,%s, %s,%s)" # batch1.add(query2, (sha256('TESTVIN' + str(i).zfill(10)), int(time.time() * 1000), int(time.time() * 1000), # 'TESTVIN' + str(i).zfill(10), '1', 'ddd', '1')) # session.execute(batch1) # 低频数据插入语句 # batch2 = BatchStatement() # print('TESTVIN' + str(i).zfill(10)) # batch2.add(query2, ( # sha256('TESTVIN' + str(i).zfill(10)), int(time.time() * 1000), "{\"202F\":255,\"2030\":255,\"2031\":255,\"2032\":255,\"2035\":255,\"2036\":255,\"2039\":255,\"203A\":255,\"203B\":255,\"203C\":255}",int(time.time() * 1000), '0')) # session.execute(batch2) # # # 高频数据插入语句 # batch3 = BatchStatement() # query3 = "insert into tsp_eu.realtime_hf(v,ct,data,st,type) values(%s, %s,%s, %s,%s)" # batch2.add(query3, ( # sha256('TESTVIN' + str(i).zfill(10)), int(time.time() * 1000), "{\"0001\":46720,\"2005\":0,\"2076\":0,\"2077\":38912,\"2082\":2958360575,\"2084\":111428784,\"2085\":46720,\"2086\":0,\"2087\":2908749824,\"208A\":176,\"208C\":68}", # int(time.time() * 1000), '0')) # session.execute(batch2) # 实时数据插入语句-每个vin只能有一条数据 batch4 = BatchStatement() query4 = "insert into tsp_eu.realtime_new(v,ct,data,st) values(%s, %s,%s, %s)" print('HESTVIN' + str(i).zfill(10)) batch4.add(query4, ( sha256('HESTVIN' + str(i).zfill(10)), int(time.time() * 1000), {'2001': '350', '2002': '1', '2003': '2000', '2004': '600', '2005': '60', '2006': '83', '2007': '350', '2008': '0', '2009': '10000', '200A': '600', '200B': '1000', '200C': '3000', '200D': '0', '200E': '1', '200F': '3', '2010': '1', '2011': '0', '2012': '0', '2013': '2', '2014': '1', '2015': '0', '2016': '0', '2017': '3', '2018': '0', '2019': '0', '201A': '0', '201B': '0', '201C': '0', '201D': '0', '201E': '0', '201F': '0', '2020': '0', '2021': '0', '2022': '0', '2023': '0', '2024': '0', '2025': '0', '2026': '230', '2027': '230', '2028': '230', '2029': '0', '202A': '6', '202B': '6', '202C': '1', '202D': '0', '202E': '0', '202F': '0', '2030': '1', '2031': '2', '2032': '0', '2033': '1', '2034': '1', '2035': '1', '2036': '1', '2037': '3', '2038': '1', '2039': '1', '203A': '0', '203B': '0', '203C': '0', '203D': '0', '203E': '150', '203F': '0', '2040': '0', '2041': '3', '2042': '17000', '2043': '1000', '2044': '85', '2045': '0', '2046': '23', '2047': '23', '2048': '23', '2049': '23', '204A': '70', '204B': '70', '204C': '70', '204D': '70', '204E': '0', '204F': '0', '2050': '0', '2051': '0', '2052': '0', '2053': '0', '2054': '0', '2055': '0', '2056': '0', '2057': '0', '2058': '0', '2059': '0', '205A': '0', '205B': '0', '205C': '0', '205D': '0', '205E': '1687590135166', '205F': '1687590135166', '2060': '5000', '2061': '180', '2062': '3000', '2063': '0', '2064': '0', '2065': '0', '2066': '0', '2067': '0', '2068': '0', '2069': '0', '206A': '0', '206B': '0', '206C': '0', '206D': '5', '206E': '0', '206F': '5', '2070': '0', '2071': '0', '2072': '0', '2073': '0', '2076': '3', '2077': '350', '2078': '0', '2080': '255', '2081': '255', '2082': '4294967295', '2083': '255', '2084': '4294967295', '2085': '4294967295', '2086': '4294967295', '2087': '4294967295', '2088': '0', '2089': '255', '208A': '255', '208B': '255', '208C': '1800', '208D': '4294967295', '208E': '4294967295', '208F': '4294967295', '2090': '4294967295', '2091': '255', '2092': '255', '2093': '255', '2094': '255', '2095': '65535'}, int(time.time() * 1000))) session.execute(batch4) # # 最新诊断 # batch5 = BatchStatement() # query5 = "insert into tsp_eu.diagnosis_new(v,ct,data,out_vehicle_type,st) values(%s,%s, %s,%s, %s)" # batch5.add(query5, ( # base64.urlsafe_b64encode(('TESTVIN' + str(i).zfill(10)).encode())[0:23].decode(), int(time.time() * 1000), # {'10': 'null', '11': 'null', '12': 'null', '13': 'null', '15': 'null', '19': 'null', # '2': '[9717033,9717267]', '20': 'null', '21': 'null', '22': 'null', '24': 'null', '32': 'null', # '35': 'null', '36': 'null', '5': 'null', '8': 'null'}, 'T19CEV_OUTSIDE_001_TEST_KB51', # int(time.time() * 1000))) # session.execute(batch5) # # 历史诊断 # batch6 = BatchStatement() # query6 = "insert into tsp_eu.diagnosis_his(v,ct,data,st) values(%s, %s,%s, %s)" # batch6.add(query6, ( # sha256('TESTVIN' + str(i).zfill(10)), int(time.time() * 1000), # "{\"5010\":0,\"5011\":1,\"5001\":4,\"5A01\":[]}", # int(time.time() * 1000))) # session.execute(batch6) # 执行批量操作 # session.execute(batch) # 关闭连接 session.shutdown() cluster.shutdown() if __name__ == '__main__': # poo1=Pool(30) # for i in range(1000): # poo1.apply_async(execute) # poo1.close() # poo1.join() # for i in range(10): # p=Process(target=execute) # p.start() execute()