-
Notifications
You must be signed in to change notification settings - Fork 0
/
write_into_influxdb.py
86 lines (75 loc) · 2.23 KB
/
write_into_influxdb.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
#!/usr/bin/env python3
from influxdb import InfluxDBClient
import uuid
import random
import time
client = InfluxDBClient(host='localhost', port=8086)
client.create_database('writetest')
measurement_name = 'm1'
number_of_points = 250000
data_end_time = int(time.time() * 1000) #milliseconds
location_tags = [ "reservoir",
"orchard",
"vineyard",
"quarry",
"hospital",
"bakery",
"warehouse",
"outhouse",
"restaurant",
"cafeteria",
"delicatessen",
"office"]
fruit_tags = [ "apple",
"banana",
"cantaloupe",
"cherry",
"coconut",
"durian",
"fig",
"gooseberry",
"grape",
"grapefruit",
"guava",
"lemon",
"lime",
"lychee",
"mango",
"papaya",
"passionfruit",
"peach",
"pineapple",
"plum",
"strawberry",
"tangerine",
"tomato",
"watermelon"]
id_tags = []
for i in range(100):
id_tags.append(str(uuid.uuid4()))
data = []
data.append("{measurement},location={location},fruit={fruit},id={id} x={x},y={y},z={z}i {timestamp}"
.format(measurement=measurement_name,
location=random.choice(location_tags),
fruit=random.choice(fruit_tags),
id=random.choice(id_tags),
x=round(random.random(),4),
y=round(random.random(),4),
z=random.randint(0,50),
timestamp=data_end_time))
current_point_time = data_end_time
for i in range(number_of_points-1):
current_point_time = current_point_time - random.randint(1,100)
data.append("{measurement},location={location},fruit={fruit},id={id} x={x},y={y},z={z}i {timestamp}"
.format(measurement=measurement_name,
location=random.choice(location_tags),
fruit=random.choice(fruit_tags),
id=random.choice(id_tags),
x=round(random.random(),4),
y=round(random.random(),4),
z=random.randint(0,50),
timestamp=current_point_time))
client_write_start_time = time.perf_counter()
client.write_points(data, database='writetest', time_precision='ms', batch_size=10000, protocol='line')
client_write_end_time = time.perf_counter()
print("Client Library Write: {time}s".format(time=client_write_end_time - client_write_start_time))