98 lines
1.9 KiB
Go
98 lines
1.9 KiB
Go
|
package db
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"strings"
|
|||
|
|
"testing"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func testCreateClickHouse(_ *testing.T) *ClickHouseRepo {
|
|||
|
|
host := "192.168.2.224"
|
|||
|
|
port := "9000"
|
|||
|
|
// port = 8123
|
|||
|
|
username := "default"
|
|||
|
|
password := "123456"
|
|||
|
|
database := "samba"
|
|||
|
|
return NewClickHouseRepo(host, port, username, password, database)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func testSelect(t *testing.T) {
|
|||
|
|
ch := testCreateClickHouse(t)
|
|||
|
|
err := ch.Open()
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
defer ch.Close()
|
|||
|
|
rows, err := ch.Select("select user_id, coins from coins_flow limit 1")
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
defer func() { _ = rows.Close() }()
|
|||
|
|
cols, err := rows.Columns()
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
fmt.Println(strings.Join(cols, ","))
|
|||
|
|
// 一行数据，使用any，避开数据类型问题
|
|||
|
|
var vRows = make([]any, len(cols))
|
|||
|
|
// 存实际的值，byte数组，长度以列的数量为准
|
|||
|
|
var values = make([][]byte, len(cols))
|
|||
|
|
for i := 0; i < len(cols); i++ {
|
|||
|
|
vRows[i] = &values[i]
|
|||
|
|
}
|
|||
|
|
for rows.Next() {
|
|||
|
|
err = rows.Scan(vRows...)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println(err.Error())
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
var vString []string
|
|||
|
|
for _, v := range values {
|
|||
|
|
vString = append(vString, string(v))
|
|||
|
|
}
|
|||
|
|
fmt.Println(vString, ",")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func testInsert(t *testing.T) {
|
|||
|
|
ch := testCreateClickHouse(t)
|
|||
|
|
err := ch.Open()
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
defer ch.Close()
|
|||
|
|
_, err = ch.Exec("INSERT INTO coins_flow(user_id, coins) VALUES (11,11)", 11, 11)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
// last, err := result.LastInsertId()
|
|||
|
|
// if err != nil {
|
|||
|
|
// t.Fatal(err)
|
|||
|
|
// }
|
|||
|
|
// fmt.Println(last)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func testDelete(t *testing.T) {
|
|||
|
|
ch := testCreateClickHouse(t)
|
|||
|
|
err := ch.Open()
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
defer ch.Close()
|
|||
|
|
_, err = ch.Exec("DELETE FROM coins_flow WHERE user_id = 11 AND coins = 11")
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
// last, err := result.LastInsertId()
|
|||
|
|
// if err != nil {
|
|||
|
|
// t.Fatal(err)
|
|||
|
|
// }
|
|||
|
|
// fmt.Println(last)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestClickHouse(t *testing.T) {
|
|||
|
|
testSelect(t)
|
|||
|
|
// testInsert(t)
|
|||
|
|
testDelete(t)
|
|||
|
|
}
|