-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathmain.go
111 lines (98 loc) · 2.46 KB
/
main.go
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
package main
import (
"context"
"fmt"
"log"
"time"
"github.com/ClickHouse/clickhouse-go/v2"
)
func main() {
dsn := "clickhouse://admin:[email protected]:19000/my_database?dial_timeout=200ms&max_execution_time=60"
options, err := clickhouse.ParseDSN(dsn)
if err != nil {
log.Fatal(err)
}
options.MaxIdleConns = 0
options.MaxOpenConns = 0
options.ConnMaxLifetime = 0
ctx := context.Background()
// 配置连接参数
db, err := clickhouse.Open(options)
if err != nil {
log.Fatal(err)
}
if err := db.Ping(ctx); err != nil {
if exception, ok := err.(*clickhouse.Exception); ok {
fmt.Printf("[%d] %s \n%s\n", exception.Code, exception.Message, exception.StackTrace)
} else {
fmt.Printf("err : %v \n", err)
}
return
}
if err = db.Exec(ctx, "DROP TABLE IF EXISTS example"); err != nil {
log.Fatal(err)
}
// 创建表: Memory 引擎以未压缩的形式将数据存储在 RAM 中
err = db.Exec(ctx, `
CREATE TABLE IF NOT EXISTS example (
country FixedString(2),
os_id UInt8,
browser_id UInt8,
categories Array(Int16),
action_day Date,
action_time DateTime
) engine=Memory
`)
if err != nil {
log.Fatal(err)
}
// 插入数据
stmt, err := db.PrepareBatch(ctx, "INSERT INTO example (country, os_id, browser_id, categories, action_day, action_time) VALUES (?, ?, ?, ?, ?, ?)")
if err != nil {
log.Fatal(err)
}
for i := 0; i < 100; i++ {
if err = stmt.Append(
"RU",
uint8(10+i),
uint8(100+i),
[]int16{1, 2, 3},
time.Now(),
time.Now(),
); err != nil {
log.Fatal(err)
}
}
if err = stmt.Send(); err != nil {
log.Fatal(err)
}
// 查询数据
rows, err := db.Query(ctx, "SELECT country, os_id, browser_id, categories, action_day, action_time FROM example")
if err != nil {
log.Fatal(err)
}
defer rows.Close()
for rows.Next() {
var e = Example{}
if err = rows.ScanStruct(&e); err != nil {
log.Fatal(err)
}
log.Printf("country: %s, os: %d, browser: %d, categories: %v, action_day: %s, action_time: %s",
e.Country, e.Os, e.Browser, e.Categories, e.ActionDay, e.ActionTime)
}
if err = rows.Err(); err != nil {
log.Fatal(err)
}
// 删除表
if err = db.Exec(ctx, "DROP TABLE example"); err != nil {
log.Fatal(err)
}
}
type Example struct {
Country string `ch:"country"`
Os uint8 `ch:"os_id"`
Browser uint8 `ch:"browser_id"`
Categories []int16 `ch:"categories"`
ActionDay time.Time `ch:"action_day"`
ActionTime time.Time `ch:"action_time"`
}