clickhouse-go/examples/std/geo.go at main · ClickHouse/clickhouse-go · GitHub

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
112
113
114
115
116
117
118
119
120
121
package std

import (
	"context"
	"fmt"

	"github.com/ClickHouse/clickhouse-go/v2"
	"github.com/paulmach/orb"
)

func GeoInsertRead() error {
	conn, err := GetStdOpenDBConnection(clickhouse.Native, clickhouse.Settings{
		"allow_experimental_geo_types": 1,
	}, nil, nil)
	if err != nil {
		return err
	}
	ctx := context.Background()
	conn.ExecContext(ctx, "DROP TABLE IF EXISTS example")

	_, err = conn.ExecContext(ctx, `
		CREATE TABLE example (
				point Point,
				ring Ring,
				lineString LineString,
				polygon Polygon,
				mPolygon MultiPolygon,
				mLineString MultiLineString
			)
			Engine Memory
		`)
	if err != nil {
		return err
	}

	scope, err := conn.Begin()
	if err != nil {
		return err
	}
	batch, err := scope.Prepare("INSERT INTO example")
	if err != nil {
		return err
	}

	_, err = batch.Exec(
		orb.Point{11, 22},
		orb.Ring{
			orb.Point{1, 2},
			orb.Point{1, 2},
		},
		orb.LineString{
			orb.Point{1, 2},
			orb.Point{3, 4},
			orb.Point{5, 6},
		},
		orb.Polygon{
			orb.Ring{
				orb.Point{1, 2},
				orb.Point{12, 2},
			},
			orb.Ring{
				orb.Point{11, 2},
				orb.Point{1, 12},
			},
		},
		orb.MultiPolygon{
			orb.Polygon{
				orb.Ring{
					orb.Point{1, 2},
					orb.Point{12, 2},
				},
				orb.Ring{
					orb.Point{11, 2},
					orb.Point{1, 12},
				},
			},
			orb.Polygon{
				orb.Ring{
					orb.Point{1, 2},
					orb.Point{12, 2},
				},
				orb.Ring{
					orb.Point{11, 2},
					orb.Point{1, 12},
				},
			},
		},
		orb.MultiLineString{
			orb.LineString{
				orb.Point{1, 2},
				orb.Point{3, 4},
			},
			orb.LineString{
				orb.Point{5, 6},
				orb.Point{7, 8},
			},
		},
	)
	if err != nil {
		return err
	}

	if err = scope.Commit(); err != nil {
		return err
	}

	var (
		point       orb.Point
		ring        orb.Ring
		lineString  orb.LineString
		polygon     orb.Polygon
		mPolygon    orb.MultiPolygon
		mLineString orb.MultiLineString
	)

	if err = conn.QueryRow("SELECT * FROM example").Scan(&point, &ring, &lineString, &polygon, &mPolygon, &mLineString); err != nil {
		return err
	}
	fmt.Printf("point=%v, ring=%v, lineString=%v, polygon=%v, mPolygon=%v, mLineString=%v\n", point, ring, lineString, polygon, mPolygon, mLineString)
	return nil
}