mirror of
https://github.com/gogf/gf.git
synced 2025-04-05 11:18:50 +08:00
81 lines
2.1 KiB
Go
81 lines
2.1 KiB
Go
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
package gredis
|
|
|
|
import (
|
|
"context"
|
|
"reflect"
|
|
|
|
"github.com/gogf/gf/v2"
|
|
"github.com/gogf/gf/v2/container/gvar"
|
|
"github.com/gogf/gf/v2/internal/json"
|
|
"github.com/gogf/gf/v2/internal/reflection"
|
|
"github.com/gogf/gf/v2/os/gtime"
|
|
"go.opentelemetry.io/otel"
|
|
"go.opentelemetry.io/otel/trace"
|
|
)
|
|
|
|
// RedisConn is a connection of redis client.
|
|
type RedisConn struct {
|
|
conn Conn
|
|
redis *Redis
|
|
}
|
|
|
|
// Do send a command to the server and returns the received reply.
|
|
// It uses json.Marshal for struct/slice/map type values before committing them to redis.
|
|
func (c *RedisConn) Do(ctx context.Context, command string, args ...interface{}) (reply *gvar.Var, err error) {
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
for k, v := range args {
|
|
var (
|
|
reflectInfo = reflection.OriginTypeAndKind(v)
|
|
)
|
|
switch reflectInfo.OriginKind {
|
|
case
|
|
reflect.Struct,
|
|
reflect.Map,
|
|
reflect.Slice,
|
|
reflect.Array:
|
|
// Ignore slice type of: []byte.
|
|
if _, ok := v.([]byte); !ok {
|
|
if args[k], err = json.Marshal(v); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Trace span start.
|
|
tr := otel.GetTracerProvider().Tracer(traceInstrumentName, trace.WithInstrumentationVersion(gf.VERSION))
|
|
_, span := tr.Start(ctx, "Redis."+command, trace.WithSpanKind(trace.SpanKindInternal))
|
|
defer span.End()
|
|
|
|
timestampMilli1 := gtime.TimestampMilli()
|
|
reply, err = c.conn.Do(ctx, command, args...)
|
|
timestampMilli2 := gtime.TimestampMilli()
|
|
|
|
// Trace span end.
|
|
c.traceSpanEnd(ctx, span, &traceItem{
|
|
err: err,
|
|
command: command,
|
|
args: args,
|
|
costMilli: timestampMilli2 - timestampMilli1,
|
|
})
|
|
return
|
|
}
|
|
|
|
// Receive receives a single reply as gvar.Var from the Redis server.
|
|
func (c *RedisConn) Receive(ctx context.Context) (*gvar.Var, error) {
|
|
return c.conn.Receive(ctx)
|
|
}
|
|
|
|
// Close puts the connection back to connection pool.
|
|
func (c *RedisConn) Close(ctx context.Context) error {
|
|
return c.conn.Close(ctx)
|
|
}
|