代码拉取完成,页面将自动刷新
package clickhouse_test
import (
"database/sql/driver"
"net"
"reflect"
"testing"
"time"
"github.com/ClickHouse/clickhouse-go"
"github.com/ClickHouse/clickhouse-go/lib/data"
"github.com/stretchr/testify/assert"
)
func Test_ColumnarInsert(t *testing.T) {
const (
ddl = `
CREATE TABLE clickhouse_test_columnar_insert (
uint8 UInt8,
uint16 UInt16,
uint32 UInt32,
uint64 UInt64,
float32 Float32,
float64 Float64,
string String,
fString FixedString(2),
date Date,
datetime DateTime,
enum8 Enum8 ('a' = 1, 'b' = 2),
enum16 Enum16('c' = 1, 'd' = 2),
array Array(String),
arrayArray Array(Array(String)),
arrayWithValue Array(UInt64),
arrayWithValueFast Array(UInt64),
ipv4 IPv4,
ipv6 IPv6
) Engine=Memory
`
dml = `
INSERT INTO clickhouse_test_columnar_insert (
uint8,
uint16,
uint32,
uint64,
float32,
float64,
string,
fString,
date,
datetime,
enum8,
enum16,
array,
arrayArray,
arrayWithValue,
arrayWithValueFast,
ipv4,
ipv6
) VALUES (
?,
?,
?,
?,
?,
?,
?,
?,
?,
?,
?,
?,
?,
?,
?,
?
)
`
)
if connect, err := clickhouse.OpenDirect("tcp://127.0.0.1:9000?debug=true"); assert.NoError(t, err) {
{
connect.Begin()
stmt, _ := connect.Prepare("DROP TABLE IF EXISTS clickhouse_test_columnar_insert")
stmt.Exec([]driver.Value{})
connect.Commit()
}
{
if _, err := connect.Begin(); assert.NoError(t, err) {
if stmt, err := connect.Prepare(ddl); assert.NoError(t, err) {
if _, err := stmt.Exec([]driver.Value{}); assert.NoError(t, err) {
assert.NoError(t, connect.Commit())
}
}
}
}
{
if _, err := connect.Begin(); assert.NoError(t, err) {
if _, err := connect.Prepare(dml); assert.NoError(t, err) {
block, err := connect.Block()
assert.NoError(t, err)
block.Reserve()
block.NumRows = 100
for i := 0; i < 100; i++ {
block.WriteUInt8(0, uint8(i))
block.WriteUInt16(1, uint16(i))
block.WriteUInt32(2, uint32(i))
block.WriteUInt64(3, uint64(i))
block.WriteFloat32(4, float32(i))
block.WriteFloat64(5, float64(i))
block.WriteString(6, "string")
block.WriteFixedString(7, []byte("CH"))
block.WriteDate(8, time.Now())
block.WriteDateTime(9, time.Now())
block.WriteUInt8(10, 1)
block.WriteUInt16(11, 2)
block.WriteArray(12, []string{"A", "B", "C"})
block.WriteArray(13, [][]string{[]string{"A", "B"}, []string{"CC", "DD", "EE"}})
block.WriteArrayWithValue(14, newUint64SliceValueSimple([]uint64{1, 2, 3}))
block.WriteArrayWithValue(15, newUint64SliceValueFast([]uint64{10, 20, 30}))
block.WriteIP(16, net.ParseIP("213.180.204.62"))
block.WriteIP(17, net.ParseIP("2606:4700:5c::a29f:2e07"))
if !assert.NoError(t, err) {
return
}
}
assert.NoError(t, connect.Commit())
}
}
}
}
}
type uint64Value struct {
value uint64
}
func (v *uint64Value) Kind() reflect.Kind {
return reflect.String
}
func (v *uint64Value) Len() int {
panic("uint64 has no length")
}
func (v *uint64Value) Index(i int) data.Value {
panic("uint64 has no index")
}
func (v *uint64Value) Interface() interface{} {
return v.value
}
type uint64SliceValueSimple struct {
uint64Slice []uint64
}
func newUint64SliceValueSimple(v []uint64) *uint64SliceValueSimple {
return &uint64SliceValueSimple{uint64Slice: v}
}
func (v *uint64SliceValueSimple) Kind() reflect.Kind {
return reflect.Slice
}
func (v *uint64SliceValueSimple) Len() int {
return len(v.uint64Slice)
}
func (v *uint64SliceValueSimple) Index(i int) data.Value {
return &uint64Value{value: v.uint64Slice[i]}
}
func (v *uint64SliceValueSimple) Interface() interface{} {
return v.uint64Slice
}
type uint64SliceValueFast struct {
uint64Slice []uint64
uint64Value *uint64Value
value data.Value
}
func newUint64SliceValueFast(v []uint64) *uint64SliceValueFast {
var uint64Value uint64Value
return &uint64SliceValueFast{
uint64Slice: v,
uint64Value: &uint64Value,
value: &uint64Value,
}
}
func (v *uint64SliceValueFast) Kind() reflect.Kind {
return reflect.Slice
}
func (v *uint64SliceValueFast) Len() int {
return len(v.uint64Slice)
}
func (v *uint64SliceValueFast) Index(i int) data.Value {
v.uint64Value.value = v.uint64Slice[i]
// NB: This avoids the CPU cost of converting *uint64Value to data.Value.
return v.value
}
func (v *uint64SliceValueFast) Interface() interface{} {
return v.uint64Slice
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。