133 lines
3.1 KiB
Go
133 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/taosdata/driver-go/v3/af/tmq"
|
|
tmqcommon "github.com/taosdata/driver-go/v3/common/tmq"
|
|
_ "github.com/taosdata/driver-go/v3/taosSql"
|
|
)
|
|
|
|
var done = make(chan struct{})
|
|
|
|
func main() {
|
|
// init env
|
|
conn, err := sql.Open("taosSql", "root:taosdata@tcp(127.0.0.1:6030)/")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer func() {
|
|
conn.Close()
|
|
}()
|
|
initEnv(conn)
|
|
// ANCHOR: create_consumer
|
|
// create consumer
|
|
consumer, err := tmq.NewConsumer(&tmqcommon.ConfigMap{
|
|
"td.connect.user": "root",
|
|
"td.connect.pass": "taosdata",
|
|
"auto.offset.reset": "latest",
|
|
"msg.with.table.name": "true",
|
|
"enable.auto.commit": "true",
|
|
"auto.commit.interval.ms": "1000",
|
|
"group.id": "group2",
|
|
"client.id": "1",
|
|
})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
// ANCHOR_END: create_consumer
|
|
// ANCHOR: subscribe
|
|
err = consumer.Subscribe("topic_meters", nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
for i := 0; i < 50; i++ {
|
|
ev := consumer.Poll(100)
|
|
if ev != nil {
|
|
switch e := ev.(type) {
|
|
case *tmqcommon.DataMessage:
|
|
// process your data here
|
|
fmt.Printf("get message:%v\n", e)
|
|
// ANCHOR: commit_offset
|
|
// commit offset
|
|
topicPartition, err := consumer.CommitOffsets([]tmqcommon.TopicPartition{e.TopicPartition})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
fmt.Println(topicPartition)
|
|
// ANCHOR_END: commit_offset
|
|
case tmqcommon.Error:
|
|
fmt.Printf("%% Error: %v: %v\n", e.Code(), e)
|
|
panic(e)
|
|
}
|
|
// commit all offsets
|
|
topicPartition, err := consumer.Commit()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
fmt.Println(topicPartition)
|
|
|
|
}
|
|
}
|
|
// ANCHOR_END: subscribe
|
|
// ANCHOR: seek
|
|
// get assignment
|
|
partitions, err := consumer.Assignment()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
for i := 0; i < len(partitions); i++ {
|
|
fmt.Println(partitions[i])
|
|
// seek to the beginning
|
|
err = consumer.Seek(tmqcommon.TopicPartition{
|
|
Topic: partitions[i].Topic,
|
|
Partition: partitions[i].Partition,
|
|
Offset: 0,
|
|
}, 0)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
// ANCHOR_END: seek
|
|
// ANCHOR: close
|
|
// unsubscribe
|
|
err = consumer.Unsubscribe()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
// close consumer
|
|
err = consumer.Close()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
// ANCHOR_END: close
|
|
<-done
|
|
}
|
|
|
|
func initEnv(conn *sql.DB) {
|
|
_, err := conn.Exec("CREATE DATABASE IF NOT EXISTS power")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
_, err = conn.Exec("CREATE STABLE IF NOT EXISTS power.meters (ts TIMESTAMP, current FLOAT, voltage INT, phase FLOAT) TAGS (groupId INT, location BINARY(24))")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
_, err = conn.Exec("CREATE TOPIC IF NOT EXISTS topic_meters AS SELECT ts, current, voltage, phase, groupid, location FROM power.meters")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
go func() {
|
|
for i := 0; i < 10; i++ {
|
|
time.Sleep(time.Second)
|
|
_, err = conn.Exec("INSERT INTO power.d1001 USING power.meters TAGS (2, 'California.SanFrancisco') VALUES (NOW , 10.2, 219, 0.32)")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
done <- struct{}{}
|
|
}()
|
|
}
|