Merge pull request #4158 from taosdata/xiaoping/add_test_case2

insert data for ningsi

tests/pytest/insert/ningsiInsert.py

+###################################################################
+#           Copyright (c) 2016 by TAOS Technologies, Inc.
+#                     All rights reserved.
+#
+#  This file is proprietary and confidential to TAOS Technologies.
+#  No part of this file may be reproduced, stored, transmitted,
+#  disclosed or used in any form or by any means other than as
+#  expressly provided by the written permission from Jianhui Tao
+#
+###################################################################
+
+# -*- coding: utf-8 -*-
+
+import sys
+import requests, json
+import threading
+import string
+import random
+import time
+
+class RestfulInsert:
+    def init(self):
+        self.header = {'Authorization': 'Basic cm9vdDp0YW9zZGF0YQ=='}
+        self.url = "http://ningsi60:6041/rest/sql"
+        self.ts = 1104508800000
+        self.numOfThreads = 10
+        self.numOfTables = 3000
+        self.dbName = 'netmonitortaos'
+        self.stbName = 'devinfomt'
+        self.prefix = 'dev'
+
+    def get_random_string(self, length):
+        letters = string.ascii_lowercase
+        result_str = ''.join(random.choice(letters) for i in range(length))
+        return result_str
+    
+    def createTables(self, threadID):
+        print("create table: thread %d started" % threadID)
+        tablesPerThread = int (self.numOfTables / self.numOfThreads)
+        for i in range(tablesPerThread):
+            data = "create table '%s'.dev_%d using '%s'.'%s' tags('%s', '%s')" % (self.dbName, i + threadID * tablesPerThread, self.dbName, self.stbName, self.get_random_string(25), self.get_random_string(25))
+            response = requests.post(self.url, data, headers = self.header)
+            if response.status_code != 200:
+                print(response.content)
+
+    def insertData(self, threadID):
+        print("insert data: thread %d started" % threadID)
+        tablesPerThread = int (self.numOfTables / self.numOfThreads)
+        base_ts = self.ts
+        while True:
+            i = 0
+            for i in range(tablesPerThread):        
+                data = "insert into %s.dev_%d values(%d, '%s', '%s', %d, %d, %d)" % (self.dbName, i + threadID * tablesPerThread, base_ts, self.get_random_string(25), self.get_random_string(30), random.randint(1, 10000), random.randint(1, 10000), random.randint(1, 10000))                
+                response = requests.post(self.url, data, headers = self.header)
+                if response.status_code != 200:
+                    print(response.content)
+
+            time.sleep(30)
+            base_ts = base_ts + 1
+
+    def run(self):    
+        data = "create database if not exists %s keep 7300" % self.dbName
+        requests.post(self.url, data, headers = self.header)
+
+        data = "create table '%s'.'%s' (timeid  timestamp, devdesc binary(50), devname binary(50), cpu bigint, temp bigint, ram bigint) tags(devid binary(50), modelid binary(50))" % (self.dbName, self.stbName)
+        requests.post(self.url, data, headers = self.header)
+
+        threads = []        
+        for i in range(self.numOfThreads):
+            thread = threading.Thread(target=self.createTables, args=(i,))
+            thread.start()
+            threads.append(thread)            
+        
+        for i in range(self.numOfThreads):
+            threads[i].join()
+
+        threads = []        
+        for i in range(self.numOfThreads):
+            thread = threading.Thread(target=self.insertData, args=(i,))
+            thread.start()
+            threads.append(thread)            
+        
+        for i in range(self.numOfThreads):
+            threads[i].join()
+
+ri = RestfulInsert()
+ri.init()
+ri.run()