feat: elasticsearch, influxdb.

2025-06-25 19:50:51 +08:00
parent 735ca567bf
commit d31ab9cdf3
16 changed files with 3217 additions and 64 deletions
--- a/database/influxdb/README.md
+++ b/database/influxdb/README.md
@@ -0,0 +1,56 @@
+# InfluxDB
+
+### Docker部署
+
+pull image
+
+```bash
+docker pull bitnami/influxdb:latest
+```
+
+#### 2.x
+
+```bash
+docker run -itd \
+    --name influxdb2-server \
+    -p 8086:8086 \
+    -e INFLUXDB_HTTP_AUTH_ENABLED=true \
+    -e INFLUXDB_ADMIN_USER=admin \
+    -e INFLUXDB_ADMIN_USER_PASSWORD=123456789 \
+    -e INFLUXDB_ADMIN_USER_TOKEN=admintoken123 \
+    -e INFLUXDB_DB=my_database \
+    bitnami/influxdb:2.7.11
+```
+
+create admin user sql script:
+
+```sql
+create user "admin" with password '123456789' with all privileges
+```
+
+管理后台: <http://localhost:8086/>
+
+#### 3.x
+
+```bash
+docker run -itd \
+    --name influxdb3-server \
+    -p 8181:8181 \
+    -e INFLUXDB_NODE_ID=0 \
+    -e INFLUXDB_HTTP_PORT_NUMBER=8181 \
+    -e INFLUXDB_HTTP_AUTH_ENABLED=true \
+    -e INFLUXDB_CREATE_ADMIN_TOKEN=yes \
+    -e INFLUXDB_DB=my_database \
+    bitnami/influxdb:latest
+
+docker run -itd \
+  --name influxdb3-explorer \
+  -p 8888:80 \
+  -p 8889:8888 \
+  quay.io/influxdb/influxdb3-explorer:latest \
+  --mode=admin
+```
+
+这个版本分离出来一个管理后台 InfluxDB Explorer：<http://localhost:8888/>
+
+在管理后台填写：`http://host.docker.internal:8181`
--- a/database/influxdb/client.go
+++ b/database/influxdb/client.go
@@ -1,29 +1,124 @@
 package influxdb

 import (
-	"github.com/InfluxCommunity/influxdb3-go/influxdb3"
+	"context"

+	"github.com/go-kratos/kratos/v2/encoding"
+	_ "github.com/go-kratos/kratos/v2/encoding/json"
 	"github.com/go-kratos/kratos/v2/log"

+	"github.com/InfluxCommunity/influxdb3-go/v2/influxdb3"
+
 	conf "github.com/tx7do/kratos-bootstrap/api/gen/go/conf/v1"
 )

-func NewInfluxClient(cfg *conf.Bootstrap, l *log.Helper) *influxdb3.Client {
+type Client struct {
+	cli *influxdb3.Client
+
+	log   *log.Helper
+	codec encoding.Codec
+}
+
+func NewClient(logger log.Logger, cfg *conf.Bootstrap) (*Client, error) {
+	c := &Client{
+		log:   log.NewHelper(log.With(logger, "module", "influxdb-client")),
+		codec: encoding.GetCodec("json"),
+	}
+
+	if err := c.createInfluxdbClient(cfg); err != nil {
+		return nil, err
+	}
+
+	return c, nil
+}
+
+// createInfluxdbClient 创建InfluxDB客户端
+func (c *Client) createInfluxdbClient(cfg *conf.Bootstrap) error {
 	if cfg.Data == nil || cfg.Data.Influxdb == nil {
-		l.Warn("influxdb config is nil")
 		return nil
 	}

 	client, err := influxdb3.New(influxdb3.ClientConfig{
-		Host:         cfg.Data.Influxdb.Address,
-		Token:        cfg.Data.Influxdb.Token,
-		Database:     cfg.Data.Influxdb.Bucket,
-		Organization: cfg.Data.Influxdb.Organization,
+		Host:         cfg.Data.Influxdb.GetHost(),
+		Token:        cfg.Data.Influxdb.GetToken(),
+		Database:     cfg.Data.Influxdb.GetDatabase(),
+		Organization: cfg.Data.Influxdb.GetOrganization(),
 	})
 	if err != nil {
-		l.Fatalf("failed opening connection to influxdb: %v", err)
-		return nil
+		c.log.Errorf("failed to create influxdb client: %v", err)
+		return err
 	}

-	return client
+	c.cli = client
+
+	return nil
+}
+
+// Close 关闭InfluxDB客户端
+func (c *Client) Close() {
+	if c.cli == nil {
+		c.log.Warn("influxdb client is nil, nothing to close")
+		return
+	}
+
+	if err := c.cli.Close(); err != nil {
+		c.log.Errorf("failed to close influxdb client: %v", err)
+	} else {
+		c.log.Info("influxdb client closed successfully")
+	}
+}
+
+// Query 查询数据
+func (c *Client) Query(ctx context.Context, query string) (*influxdb3.QueryIterator, error) {
+	if c.cli == nil {
+		return nil, ErrInfluxDBClientNotInitialized
+	}
+
+	result, err := c.cli.Query(
+		ctx,
+		query,
+		influxdb3.WithQueryType(influxdb3.InfluxQL),
+	)
+	if err != nil {
+		c.log.Errorf("failed to query data: %v", err)
+		return nil, ErrInfluxDBQueryFailed
+	}
+
+	return result, nil
+}
+
+// Insert 插入数据
+func (c *Client) Insert(ctx context.Context, point *influxdb3.Point) error {
+	if c.cli == nil {
+		return ErrInfluxDBClientNotInitialized
+	}
+	if point == nil {
+		return ErrInvalidPoint
+	}
+
+	points := []*influxdb3.Point{point}
+	if err := c.cli.WritePoints(ctx, points); err != nil {
+		c.log.Errorf("failed to insert data: %v", err)
+		return ErrInsertFailed
+	}
+
+	return nil
+}
+
+// BatchInsert 批量插入数据
+func (c *Client) BatchInsert(ctx context.Context, points []*influxdb3.Point) error {
+	if c.cli == nil {
+		return ErrInfluxDBClientNotInitialized
+	}
+
+	if len(points) == 0 {
+		return ErrNoPointsToInsert
+	}
+
+	if err := c.cli.WritePoints(ctx, points); err != nil {
+		c.log.Errorf("failed to batch insert data: %v", err)
+		return ErrBatchInsertFailed
+	}
+
+	return nil
 }
--- a/database/influxdb/client_test.go
+++ b/database/influxdb/client_test.go
@@ -0,0 +1,211 @@
+package influxdb
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/InfluxCommunity/influxdb3-go/v2/influxdb3"
+	"github.com/go-kratos/kratos/v2/log"
+	"github.com/stretchr/testify/assert"
+	"github.com/tx7do/go-utils/trans"
+	conf "github.com/tx7do/kratos-bootstrap/api/gen/go/conf/v1"
+	"google.golang.org/protobuf/types/known/timestamppb"
+)
+
+type Candle struct {
+	Symbol    *string
+	Open      *float64
+	High      *float64
+	Low       *float64
+	Close     *float64
+	Volume    *float64
+	StartTime *timestamppb.Timestamp
+}
+
+func (c *Candle) GetSymbol() string {
+	if c.Symbol != nil {
+		return *c.Symbol
+	}
+	return ""
+}
+
+func (c *Candle) GetOpen() float64 {
+	if c.Open != nil {
+		return *c.Open
+	}
+	return 0.0
+}
+
+func (c *Candle) GetHigh() float64 {
+	if c.High != nil {
+		return *c.High
+	}
+	return 0.0
+}
+
+func (c *Candle) GetLow() float64 {
+	if c.Low != nil {
+		return *c.Low
+	}
+	return 0.0
+}
+
+func (c *Candle) GetClose() float64 {
+	if c.Close != nil {
+		return *c.Close
+	}
+	return 0.0
+}
+
+func (c *Candle) GetVolume() float64 {
+	if c.Volume != nil {
+		return *c.Volume
+	}
+	return 0.0
+}
+
+func (c *Candle) GetStartTime() *timestamppb.Timestamp {
+	if c.StartTime != nil {
+		return c.StartTime
+	}
+	return timestamppb.Now()
+}
+
+type CandleMapper struct{}
+
+var candleMapper CandleMapper
+
+func (m *CandleMapper) ToPoint(data *Candle) *influxdb3.Point {
+	p := influxdb3.NewPoint(
+		"candles",
+		map[string]string{"s": data.GetSymbol()},
+		nil,
+		data.StartTime.AsTime(),
+	)
+
+	p.
+		SetDoubleField("o", data.GetOpen()).
+		SetDoubleField("h", data.GetHigh()).
+		SetDoubleField("l", data.GetLow()).
+		SetDoubleField("c", data.GetClose()).
+		SetDoubleField("v", data.GetVolume())
+
+	return p
+}
+
+func (m *CandleMapper) ToData(point *influxdb3.Point) *Candle {
+	symbol, _ := point.GetTag("s")
+
+	return &Candle{
+		Symbol:    &symbol,
+		Open:      point.GetDoubleField("o"),
+		High:      point.GetDoubleField("h"),
+		Low:       point.GetDoubleField("l"),
+		Close:     point.GetDoubleField("c"),
+		Volume:    point.GetDoubleField("v"),
+		StartTime: timestamppb.New(point.Values.Timestamp),
+	}
+}
+
+func createTestClient() *Client {
+	cli, _ := NewClient(
+		log.DefaultLogger,
+		&conf.Bootstrap{
+			Data: &conf.Data{
+				Influxdb: &conf.Data_InfluxDB{
+					Host:         "http://localhost:8181",
+					Token:        "apiv3_yYde4mJo0BYC7Ipi_00ZEex-A8if4swdqTBXiO-lCUDKhsIavHlRCQfo3p_DzI7S34ADHOC7Qxf600VVgW6LQQ",
+					Database:     "finances",
+					Organization: "primary",
+				},
+			},
+		},
+	)
+	return cli
+}
+
+func TestNewClient(t *testing.T) {
+	client := createTestClient()
+	assert.NotNil(t, client)
+}
+
+func TestClient_Insert(t *testing.T) {
+	client := createTestClient()
+	assert.NotNil(t, client)
+
+	item := &Candle{
+		StartTime: timestamppb.New(time.Now()),
+		Symbol:    trans.Ptr("AAPL"),
+		Open:      trans.Ptr(1.0),
+		High:      trans.Ptr(2.0),
+		Low:       trans.Ptr(3.0),
+		Close:     trans.Ptr(4.0),
+		Volume:    trans.Ptr(1000.0),
+	}
+
+	point := candleMapper.ToPoint(item)
+
+	err := client.Insert(context.Background(), point)
+	assert.NoError(t, err)
+}
+
+func TestClient_BatchInsert(t *testing.T) {
+	client := createTestClient()
+	assert.NotNil(t, client)
+
+	items := []*Candle{
+		{
+			StartTime: timestamppb.New(time.Now()),
+			Symbol:    trans.Ptr("AAPL"),
+			Open:      trans.Ptr(1.0),
+			High:      trans.Ptr(2.0),
+			Low:       trans.Ptr(3.0),
+			Close:     trans.Ptr(4.0),
+			Volume:    trans.Ptr(1000.0),
+		},
+	}
+
+	var points []*influxdb3.Point
+	for _, item := range items {
+		point := candleMapper.ToPoint(item)
+		points = append(points, point)
+	}
+
+	err := client.BatchInsert(
+		context.Background(),
+		points,
+	)
+	assert.NoError(t, err)
+}
+
+func TestClient_Query(t *testing.T) {
+	client := createTestClient()
+	assert.NotNil(t, client)
+
+	ctx := context.Background()
+
+	sql := `SELECT * FROM candles`
+
+	iterator, err := client.Query(ctx, sql)
+	assert.NoError(t, err)
+
+	for iterator.Next() {
+		point, _ := iterator.AsPoints().AsPoint()
+		candle := candleMapper.ToData(point)
+		t.Logf("[%v] Candle: %s, Open: %f, High: %f, Low: %f, Close: %f, Volume: %f\n",
+			candle.GetStartTime().AsTime().String(),
+			candle.GetSymbol(),
+			candle.GetOpen(), candle.GetHigh(), candle.GetLow(), candle.GetClose(), candle.GetVolume(),
+		)
+	}
+
+	candles, err := Query(ctx, client, sql, &candleMapper)
+	assert.NoError(t, err)
+	for _, candle := range candles {
+		t.Logf("Candle: %s, Open: %f, High: %f, Low: %f, Close: %f, Volume: %f\n",
+			candle.GetSymbol(),
+			candle.GetOpen(), candle.GetHigh(), candle.GetLow(), candle.GetClose(), candle.GetVolume(),
+		)
+	}
+}
--- a/database/influxdb/errors.go
+++ b/database/influxdb/errors.go
@@ -0,0 +1,25 @@
+package influxdb
+
+import "github.com/go-kratos/kratos/v2/errors"
+
+var (
+	ErrInfluxDBClientNotInitialized = errors.InternalServer("INFLUXDB_CLIENT_NOT_INITIALIZED", "client not initialized")
+
+	ErrInfluxDBConnectFailed = errors.InternalServer("INFLUXDB_CONNECT_FAILED", "connect failed")
+
+	ErrInfluxDBCreateDatabaseFailed = errors.InternalServer("INFLUXDB_CREATE_DATABASE_FAILED", "database create failed")
+
+	ErrInfluxDBQueryFailed = errors.InternalServer("INFLUXDB_QUERY_FAILED", "query failed")
+
+	ErrClientNotConnected = errors.InternalServer("INFLUXDB_CLIENT_NOT_CONNECTED", "client not connected")
+
+	ErrInvalidPoint = errors.InternalServer("INFLUXDB_INVALID_POINT", "invalid point")
+
+	ErrNoPointsToInsert = errors.InternalServer("INFLUXDB_NO_POINTS_TO_INSERT", "no points to insert")
+
+	ErrEmptyData = errors.InternalServer("INFLUXDB_EMPTY_DATA", "empty data")
+
+	ErrBatchInsertFailed = errors.InternalServer("INFLUXDB_BATCH_INSERT_FAILED", "batch insert failed")
+
+	ErrInsertFailed = errors.InternalServer("INFLUXDB_INSERT_FAILED", "insert failed")
+)
--- a/database/influxdb/go.mod
+++ b/database/influxdb/go.mod
@@ -1,36 +1,42 @@
 module github.com/tx7do/kratos-bootstrap/database/influxdb

-go 1.23.0
+go 1.23.10

-toolchain go1.23.3
+toolchain go1.24.4

 replace github.com/tx7do/kratos-bootstrap/api => ../../api

 require (
-	github.com/InfluxCommunity/influxdb3-go v0.14.0
+	github.com/InfluxCommunity/influxdb3-go/v2 v2.8.0
 	github.com/go-kratos/kratos/v2 v2.8.4
-	github.com/tx7do/kratos-bootstrap/api v0.0.21
+	github.com/stretchr/testify v1.10.0
+	github.com/tx7do/go-utils v1.1.29
+	github.com/tx7do/kratos-bootstrap/api v0.0.25
+	google.golang.org/protobuf v1.36.6
 )

 require (
-	github.com/apache/arrow/go/v15 v15.0.2 // indirect
+	github.com/apache/arrow-go/v18 v18.3.1 // indirect
+	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/frankban/quicktest v1.14.0 // indirect
 	github.com/goccy/go-json v0.10.5 // indirect
 	github.com/google/flatbuffers v25.2.10+incompatible // indirect
+	github.com/google/uuid v1.6.0 // indirect
 	github.com/influxdata/line-protocol/v2 v2.2.1 // indirect
 	github.com/klauspost/compress v1.18.0 // indirect
-	github.com/klauspost/cpuid/v2 v2.2.10 // indirect
+	github.com/klauspost/cpuid/v2 v2.2.11 // indirect
 	github.com/pierrec/lz4/v4 v4.1.22 // indirect
+	github.com/pmezard/go-difflib v1.0.0 // indirect
 	github.com/zeebo/xxh3 v1.0.2 // indirect
-	golang.org/x/exp v0.0.0-20250531010427-b6e5de432a8b // indirect
-	golang.org/x/mod v0.24.0 // indirect
-	golang.org/x/net v0.40.0 // indirect
-	golang.org/x/sync v0.14.0 // indirect
+	golang.org/x/exp v0.0.0-20250620022241-b7579e27df2b // indirect
+	golang.org/x/mod v0.25.0 // indirect
+	golang.org/x/net v0.41.0 // indirect
+	golang.org/x/sync v0.15.0 // indirect
 	golang.org/x/sys v0.33.0 // indirect
-	golang.org/x/text v0.25.0 // indirect
-	golang.org/x/tools v0.33.0 // indirect
+	golang.org/x/text v0.26.0 // indirect
+	golang.org/x/tools v0.34.0 // indirect
 	golang.org/x/xerrors v0.0.0-20240903120638-7835f813f4da // indirect
-	google.golang.org/genproto/googleapis/rpc v0.0.0-20250528174236-200df99c418a // indirect
-	google.golang.org/grpc v1.72.2 // indirect
-	google.golang.org/protobuf v1.36.6 // indirect
+	google.golang.org/genproto/googleapis/rpc v0.0.0-20250603155806-513f23925822 // indirect
+	google.golang.org/grpc v1.73.0 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
 )
--- a/database/influxdb/go.sum
+++ b/database/influxdb/go.sum
@@ -1,7 +1,11 @@
-github.com/InfluxCommunity/influxdb3-go v0.14.0 h1:lFEJRZM+hQzuCz36k6YKeE6CE9oKBfKkIO2bYLghar0=
-github.com/InfluxCommunity/influxdb3-go v0.14.0/go.mod h1:+0XXsEt0XMP0o7WBvDXLCJ8MxmxfgjG3mLrYPRRWsLs=
-github.com/apache/arrow/go/v15 v15.0.2 h1:60IliRbiyTWCWjERBCkO1W4Qun9svcYoZrSLcyOsMLE=
-github.com/apache/arrow/go/v15 v15.0.2/go.mod h1:DGXsR3ajT524njufqf95822i+KTh+yea1jass9YXgjA=
+github.com/InfluxCommunity/influxdb3-go/v2 v2.8.0 h1:auHy7TmHQJVRs+r59k+UIlN9yuY4eFq7d6xrsGSo0E8=
+github.com/InfluxCommunity/influxdb3-go/v2 v2.8.0/go.mod h1:wccnTQV9OQ9XvW7ttXINSccyzSmaADzYFheoCHW2sCs=
+github.com/andybalholm/brotli v1.1.1 h1:PR2pgnyFznKEugtsUo0xLdDop5SKXd5Qf5ysW+7XdTA=
+github.com/andybalholm/brotli v1.1.1/go.mod h1:05ib4cKhjx3OQYUY22hTVd34Bc8upXjOLL2rKwwZBoA=
+github.com/apache/arrow-go/v18 v18.3.1 h1:oYZT8FqONiK74JhlH3WKVv+2NKYoyZ7C2ioD4Dj3ixk=
+github.com/apache/arrow-go/v18 v18.3.1/go.mod h1:12QBya5JZT6PnBihi5NJTzbACrDGXYkrgjujz3MRQXU=
+github.com/apache/thrift v0.21.0 h1:tdPmh/ptjE1IJnhbhrcl2++TauVjy242rkV/UzJChnE=
+github.com/apache/thrift v0.21.0/go.mod h1:W1H8aR/QRtYNvrPeFXBtobyRkd0/YVhTc6i07XIAgDw=
 github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
 github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
@@ -20,13 +24,15 @@ github.com/goccy/go-json v0.10.5 h1:Fq85nIqj+gXn/S5ahsiTlK3TmC85qgirsdTP/+DeaC4=
 github.com/goccy/go-json v0.10.5/go.mod h1:oq7eo15ShAhp70Anwd5lgX2pLfOS3QCiwU/PULtXL6M=
 github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=
 github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=
+github.com/golang/snappy v1.0.0 h1:Oy607GVXHs7RtbggtPBnr2RmDArIsAefDwvrdWvRhGs=
+github.com/golang/snappy v1.0.0/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
 github.com/google/flatbuffers v25.2.10+incompatible h1:F3vclr7C3HpB1k9mxCGRMXq6FdUalZ6H/pNX4FP1v0Q=
 github.com/google/flatbuffers v25.2.10+incompatible/go.mod h1:1AeVuKshWv4vARoZatz6mlQ0JxURH0Kv5+zNeJKJCa8=
 github.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
 github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
 github.com/google/go-cmp v0.5.6/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
-github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=
-github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
+github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
+github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU=
 github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
 github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/influxdata/line-protocol-corpus v0.0.0-20210519164801-ca6fa5da0184/go.mod h1:03nmhxzZ7Xk2pdG+lmMd7mHDfeVOYFyhOgwO61qWU98=
@@ -36,10 +42,12 @@ github.com/influxdata/line-protocol/v2 v2.0.0-20210312151457-c52fdecb625a/go.mod
 github.com/influxdata/line-protocol/v2 v2.1.0/go.mod h1:QKw43hdUBg3GTk2iC3iyCxksNj7PX9aUSeYOYE/ceHY=
 github.com/influxdata/line-protocol/v2 v2.2.1 h1:EAPkqJ9Km4uAxtMRgUubJyqAr6zgWM0dznKMLRauQRE=
 github.com/influxdata/line-protocol/v2 v2.2.1/go.mod h1:DmB3Cnh+3oxmG6LOBIxce4oaL4CPj3OmMPgvauXh+tM=
+github.com/klauspost/asmfmt v1.3.2 h1:4Ri7ox3EwapiOjCki+hw14RyKk201CN4rzyCJRFLpK4=
+github.com/klauspost/asmfmt v1.3.2/go.mod h1:AG8TuvYojzulgDAMCnYn50l/5QV3Bs/tp6j0HLHbNSE=
 github.com/klauspost/compress v1.18.0 h1:c/Cqfb0r+Yi+JtIEq73FWXVkRonBlf0CRNYc8Zttxdo=
 github.com/klauspost/compress v1.18.0/go.mod h1:2Pp+KzxcywXVXMr50+X0Q/Lsb43OQHYWRCY2AiWywWQ=
-github.com/klauspost/cpuid/v2 v2.2.10 h1:tBs3QSyvjDyFTq3uoc/9xFpCuOsJQFNPiAhYdw2skhE=
-github.com/klauspost/cpuid/v2 v2.2.10/go.mod h1:hqwkgyIinND0mEev00jJYCxPNVRVXFQeu1XKlok6oO0=
+github.com/klauspost/cpuid/v2 v2.2.11 h1:0OwqZRYI2rFrjS4kvkDnqJkKHdHaRnCm68/DY4OxRzU=
+github.com/klauspost/cpuid/v2 v2.2.11/go.mod h1:hqwkgyIinND0mEev00jJYCxPNVRVXFQeu1XKlok6oO0=
 github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
 github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=
 github.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk=
@@ -49,6 +57,10 @@ github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
 github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
 github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
 github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
+github.com/minio/asm2plan9s v0.0.0-20200509001527-cdd76441f9d8 h1:AMFGa4R4MiIpspGNG7Z948v4n35fFGB3RR3G/ry4FWs=
+github.com/minio/asm2plan9s v0.0.0-20200509001527-cdd76441f9d8/go.mod h1:mC1jAcsrzbxHt8iiaC+zU4b1ylILSosueou12R++wfY=
+github.com/minio/c2goasm v0.0.0-20190812172519-36a3d3bbc4f3 h1:+n/aFZefKZp7spd8DFdX7uMikMLXX4oubIzJF4kv/wI=
+github.com/minio/c2goasm v0.0.0-20190812172519-36a3d3bbc4f3/go.mod h1:RagcQ7I8IeTMnF8JTXieKnO4Z6JCsikNEzj0DwauVzE=
 github.com/niemeyer/pretty v0.0.0-20200227124842-a10e7caefd8e/go.mod h1:zD1mROLANZcx1PVRCS0qkT7pwLkGfwJo4zjcN/Tysno=
 github.com/pierrec/lz4/v4 v4.1.22 h1:cKFw6uJDK+/gfw5BcDL0JL5aBsAFdsIT18eRtLj7VIU=
 github.com/pierrec/lz4/v4 v4.1.22/go.mod h1:gZWDp/Ze/IJXGXf23ltt2EXimqmTUXEy0GFuRQyBid4=
@@ -57,52 +69,56 @@ github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZN
 github.com/rogpeppe/go-internal v1.6.1/go.mod h1:xXDCJY+GAPziupqXw64V24skbSoqbTEfhy4qGm1nDQc=
 github.com/rogpeppe/go-internal v1.11.0 h1:cWPaGQEPrBb5/AsnsZesgZZ9yb1OQ+GOISoDNXVBh4M=
 github.com/rogpeppe/go-internal v1.11.0/go.mod h1:ddIwULY96R17DhadqLgMfk9H9tvdUzkipdSkR5nkCZA=
-github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
-github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
+github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
+github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
+github.com/tx7do/go-utils v1.1.29 h1:kO1JDMVX++ZY4+aXGk3pOtDz5WBPDA3LxhIWkzXkvH8=
+github.com/tx7do/go-utils v1.1.29/go.mod h1:bmt7c85QmHURtd7h6QOu7k0QKOJTwjJ+cFP29nljdSw=
 github.com/zeebo/assert v1.3.0 h1:g7C04CbJuIDKNPFHmsk4hwZDO5O+kntRxzaUoNXj+IQ=
 github.com/zeebo/assert v1.3.0/go.mod h1:Pq9JiuJQpG8JLJdtkwrJESF0Foym2/D9XMU5ciN/wJ0=
 github.com/zeebo/xxh3 v1.0.2 h1:xZmwmqxHZA8AI603jOQ0tMqmBr9lPeFwGg6d+xy9DC0=
 github.com/zeebo/xxh3 v1.0.2/go.mod h1:5NWz9Sef7zIDm2JHfFlcQvNekmcEl9ekUZQQKCYaDcA=
 go.opentelemetry.io/auto/sdk v1.1.0 h1:cH53jehLUN6UFLY71z+NDOiNJqDdPRaXzTel0sJySYA=
 go.opentelemetry.io/auto/sdk v1.1.0/go.mod h1:3wSPjt5PWp2RhlCcmmOial7AvC4DQqZb7a7wCow3W8A=
-go.opentelemetry.io/otel v1.34.0 h1:zRLXxLCgL1WyKsPVrgbSdMN4c0FMkDAskSTQP+0hdUY=
-go.opentelemetry.io/otel v1.34.0/go.mod h1:OWFPOQ+h4G8xpyjgqo4SxJYdDQ/qmRH+wivy7zzx9oI=
-go.opentelemetry.io/otel/metric v1.34.0 h1:+eTR3U0MyfWjRDhmFMxe2SsW64QrZ84AOhvqS7Y+PoQ=
-go.opentelemetry.io/otel/metric v1.34.0/go.mod h1:CEDrp0fy2D0MvkXE+dPV7cMi8tWZwX3dmaIhwPOaqHE=
-go.opentelemetry.io/otel/sdk v1.34.0 h1:95zS4k/2GOy069d321O8jWgYsW3MzVV+KuSPKp7Wr1A=
-go.opentelemetry.io/otel/sdk v1.34.0/go.mod h1:0e/pNiaMAqaykJGKbi+tSjWfNNHMTxoC9qANsCzbyxU=
-go.opentelemetry.io/otel/sdk/metric v1.34.0 h1:5CeK9ujjbFVL5c1PhLuStg1wxA7vQv7ce1EK0Gyvahk=
-go.opentelemetry.io/otel/sdk/metric v1.34.0/go.mod h1:jQ/r8Ze28zRKoNRdkjCZxfs6YvBTG1+YIqyFVFYec5w=
-go.opentelemetry.io/otel/trace v1.34.0 h1:+ouXS2V8Rd4hp4580a8q23bg0azF2nI8cqLYnC8mh/k=
-go.opentelemetry.io/otel/trace v1.34.0/go.mod h1:Svm7lSjQD7kG7KJ/MUHPVXSDGz2OX4h0M2jHBhmSfRE=
-golang.org/x/exp v0.0.0-20250531010427-b6e5de432a8b h1:QoALfVG9rhQ/M7vYDScfPdWjGL9dlsVVM5VGh7aKoAA=
-golang.org/x/exp v0.0.0-20250531010427-b6e5de432a8b/go.mod h1:U6Lno4MTRCDY+Ba7aCcauB9T60gsv5s4ralQzP72ZoQ=
-golang.org/x/mod v0.24.0 h1:ZfthKaKaT4NrhGVZHO1/WDTwGES4De8KtWO0SIbNJMU=
-golang.org/x/mod v0.24.0/go.mod h1:IXM97Txy2VM4PJ3gI61r1YEk/gAj6zAHN3AdZt6S9Ww=
-golang.org/x/net v0.40.0 h1:79Xs7wF06Gbdcg4kdCCIQArK11Z1hr5POQ6+fIYHNuY=
-golang.org/x/net v0.40.0/go.mod h1:y0hY0exeL2Pku80/zKK7tpntoX23cqL3Oa6njdgRtds=
-golang.org/x/sync v0.14.0 h1:woo0S4Yywslg6hp4eUFjTVOyKt0RookbpAHG4c1HmhQ=
-golang.org/x/sync v0.14.0/go.mod h1:1dzgHSNfp02xaA81J2MS99Qcpr2w7fw1gpm99rleRqA=
+go.opentelemetry.io/otel v1.35.0 h1:xKWKPxrxB6OtMCbmMY021CqC45J+3Onta9MqjhnusiQ=
+go.opentelemetry.io/otel v1.35.0/go.mod h1:UEqy8Zp11hpkUrL73gSlELM0DupHoiq72dR+Zqel/+Y=
+go.opentelemetry.io/otel/metric v1.35.0 h1:0znxYu2SNyuMSQT4Y9WDWej0VpcsxkuklLa4/siN90M=
+go.opentelemetry.io/otel/metric v1.35.0/go.mod h1:nKVFgxBZ2fReX6IlyW28MgZojkoAkJGaE8CpgeAU3oE=
+go.opentelemetry.io/otel/sdk v1.35.0 h1:iPctf8iprVySXSKJffSS79eOjl9pvxV9ZqOWT0QejKY=
+go.opentelemetry.io/otel/sdk v1.35.0/go.mod h1:+ga1bZliga3DxJ3CQGg3updiaAJoNECOgJREo9KHGQg=
+go.opentelemetry.io/otel/sdk/metric v1.35.0 h1:1RriWBmCKgkeHEhM7a2uMjMUfP7MsOF5JpUCaEqEI9o=
+go.opentelemetry.io/otel/sdk/metric v1.35.0/go.mod h1:is6XYCUMpcKi+ZsOvfluY5YstFnhW0BidkR+gL+qN+w=
+go.opentelemetry.io/otel/trace v1.35.0 h1:dPpEfJu1sDIqruz7BHFG3c7528f6ddfSWfFDVt/xgMs=
+go.opentelemetry.io/otel/trace v1.35.0/go.mod h1:WUk7DtFp1Aw2MkvqGdwiXYDZZNvA/1J8o6xRXLrIkyc=
+golang.org/x/exp v0.0.0-20250620022241-b7579e27df2b h1:M2rDM6z3Fhozi9O7NWsxAkg/yqS/lQJ6PmkyIV3YP+o=
+golang.org/x/exp v0.0.0-20250620022241-b7579e27df2b/go.mod h1:3//PLf8L/X+8b4vuAfHzxeRUl04Adcb341+IGKfnqS8=
+golang.org/x/mod v0.25.0 h1:n7a+ZbQKQA/Ysbyb0/6IbB1H/X41mKgbhfv7AfG/44w=
+golang.org/x/mod v0.25.0/go.mod h1:IXM97Txy2VM4PJ3gI61r1YEk/gAj6zAHN3AdZt6S9Ww=
+golang.org/x/net v0.41.0 h1:vBTly1HeNPEn3wtREYfy4GZ/NECgw2Cnl+nK6Nz3uvw=
+golang.org/x/net v0.41.0/go.mod h1:B/K4NNqkfmg07DQYrbwvSluqCJOOXwUjeb/5lOisjbA=
+golang.org/x/sync v0.15.0 h1:KWH3jNZsfyT6xfAfKiz6MRNmd46ByHDYaZ7KSkCtdW8=
+golang.org/x/sync v0.15.0/go.mod h1:1dzgHSNfp02xaA81J2MS99Qcpr2w7fw1gpm99rleRqA=
 golang.org/x/sys v0.33.0 h1:q3i8TbbEz+JRD9ywIRlyRAQbM0qF7hu24q3teo2hbuw=
 golang.org/x/sys v0.33.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
-golang.org/x/text v0.25.0 h1:qVyWApTSYLk/drJRO5mDlNYskwQznZmkpV2c8q9zls4=
-golang.org/x/text v0.25.0/go.mod h1:WEdwpYrmk1qmdHvhkSTNPm3app7v4rsT8F2UD6+VHIA=
-golang.org/x/tools v0.33.0 h1:4qz2S3zmRxbGIhDIAgjxvFutSvH5EfnsYrRBj0UI0bc=
-golang.org/x/tools v0.33.0/go.mod h1:CIJMaWEY88juyUfo7UbgPqbC8rU2OqfAV1h2Qp0oMYI=
+golang.org/x/text v0.26.0 h1:P42AVeLghgTYr4+xUnTRKDMqpar+PtX7KWuNQL21L8M=
+golang.org/x/text v0.26.0/go.mod h1:QK15LZJUUQVJxhz7wXgxSy/CJaTFjd0G+YLonydOVQA=
+golang.org/x/tools v0.34.0 h1:qIpSLOxeCYGg9TrcJokLBG4KFA6d795g0xkBkiESGlo=
+golang.org/x/tools v0.34.0/go.mod h1:pAP9OwEaY1CAW3HOmg3hLZC5Z0CCmzjAF2UQMSqNARg=
 golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
 golang.org/x/xerrors v0.0.0-20240903120638-7835f813f4da h1:noIWHXmPHxILtqtCOPIhSt0ABwskkZKjD3bXGnZGpNY=
 golang.org/x/xerrors v0.0.0-20240903120638-7835f813f4da/go.mod h1:NDW/Ps6MPRej6fsCIbMTohpP40sJ/P/vI1MoTEGwX90=
-gonum.org/v1/gonum v0.12.0 h1:xKuo6hzt+gMav00meVPUlXwSdoEJP46BR+wdxQEFK2o=
-gonum.org/v1/gonum v0.12.0/go.mod h1:73TDxJfAAHeA8Mk9mf8NlIppyhQNo5GLTcYeqgo2lvY=
-google.golang.org/genproto/googleapis/rpc v0.0.0-20250528174236-200df99c418a h1:v2PbRU4K3llS09c7zodFpNePeamkAwG3mPrAery9VeE=
-google.golang.org/genproto/googleapis/rpc v0.0.0-20250528174236-200df99c418a/go.mod h1:qQ0YXyHHx3XkvlzUtpXDkS29lDSafHMZBAZDc03LQ3A=
-google.golang.org/grpc v1.72.2 h1:TdbGzwb82ty4OusHWepvFWGLgIbNo1/SUynEN0ssqv8=
-google.golang.org/grpc v1.72.2/go.mod h1:wH5Aktxcg25y1I3w7H69nHfXdOG3UiadoBtjh3izSDM=
+gonum.org/v1/gonum v0.16.0 h1:5+ul4Swaf3ESvrOnidPp4GZbzf0mxVQpDCYUQE7OJfk=
+gonum.org/v1/gonum v0.16.0/go.mod h1:fef3am4MQ93R2HHpKnLk4/Tbh/s0+wqD5nfa6Pnwy4E=
+google.golang.org/genproto/googleapis/rpc v0.0.0-20250603155806-513f23925822 h1:fc6jSaCT0vBduLYZHYrBBNY4dsWuvgyff9noRNDdBeE=
+google.golang.org/genproto/googleapis/rpc v0.0.0-20250603155806-513f23925822/go.mod h1:qQ0YXyHHx3XkvlzUtpXDkS29lDSafHMZBAZDc03LQ3A=
+google.golang.org/grpc v1.73.0 h1:VIWSmpI2MegBtTuFt5/JWy2oXxtjJ/e89Z70ImfD2ok=
+google.golang.org/grpc v1.73.0/go.mod h1:50sbHOUqWoCQGI8V2HQLJM0B+LMlIUjNSZmow7EVBQc=
 google.golang.org/protobuf v1.36.6 h1:z1NpPI8ku2WgiWnf+t9wTPsn6eP1L7ksHUlkfLvd9xY=
 google.golang.org/protobuf v1.36.6/go.mod h1:jduwjTPXsFjZGTmRluh+L6NjiWu7pchiJ2/5YcXBHnY=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/check.v1 v1.0.0-20200227125254-8fa46927fb4f/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
+gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
 gopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI=
 gopkg.in/yaml.v3 v3.0.0-20200615113413-eeeca48fe776/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
--- a/database/influxdb/mapper.go
+++ b/database/influxdb/mapper.go
@@ -0,0 +1,96 @@
+package influxdb
+
+import (
+	"context"
+
+	"github.com/InfluxCommunity/influxdb3-go/v2/influxdb3"
+)
+
+// Mapper 数据转换的接口
+type Mapper[T any] interface {
+	// ToPoint 将数据转换为InfluxDB的Point格式
+	ToPoint(data *T) *influxdb3.Point
+
+	// ToData 将InfluxDB的Point转换为原始数据
+	ToData(point *influxdb3.Point) *T
+}
+
+// Insert 插入数据
+func Insert[T any](ctx context.Context, c *Client, data *T, mapper Mapper[T]) error {
+	if c.cli == nil {
+		return ErrClientNotConnected
+	}
+
+	if data == nil {
+		return ErrEmptyData
+	}
+
+	point := mapper.ToPoint(data)
+	if point == nil {
+		return ErrInvalidPoint
+	}
+
+	err := c.Insert(ctx, point)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+// BatchInsert 批量插入数据
+func BatchInsert[T any](ctx context.Context, c *Client, data []*T, mapper Mapper[T]) error {
+	if c.cli == nil {
+		return ErrClientNotConnected
+	}
+
+	if len(data) == 0 {
+		return ErrEmptyData
+	}
+
+	points := make([]*influxdb3.Point, len(data))
+	for i, d := range data {
+		point := mapper.ToPoint(d)
+		if point == nil {
+			return ErrInvalidPoint
+		}
+		points[i] = point
+	}
+
+	err := c.BatchInsert(ctx, points)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+// Query 查询数据
+func Query[T any](ctx context.Context, c *Client, query string, mapper Mapper[T]) ([]*T, error) {
+	if c.cli == nil {
+		return nil, ErrClientNotConnected
+	}
+
+	iterator, err := c.Query(ctx, query)
+	if err != nil {
+		return nil, err
+	}
+
+	var dataset []*T
+
+	for iterator.Next() {
+		point, _ := iterator.AsPoints().AsPoint()
+		if point == nil {
+			return nil, ErrInvalidPoint
+		}
+
+		data := mapper.ToData(point)
+		dataset = append(dataset, data)
+	}
+
+	if iterator.Err() != nil {
+		return nil, iterator.Err()
+	}
+
+	return dataset, nil
+}