You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
109 lines
2.9 KiB
109 lines
2.9 KiB
package memcache
|
|
|
|
import (
|
|
"context"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/bilibili/kratos/pkg/log"
|
|
"github.com/bilibili/kratos/pkg/net/trace"
|
|
)
|
|
|
|
const (
|
|
_traceFamily = "memcache"
|
|
_traceSpanKind = "client"
|
|
_traceComponentName = "library/cache/memcache"
|
|
_tracePeerService = "memcache"
|
|
_slowLogDuration = time.Millisecond * 250
|
|
)
|
|
|
|
type traceConn struct {
|
|
Conn
|
|
ctx context.Context
|
|
address string
|
|
}
|
|
|
|
func (t *traceConn) setTrace(action, statement string) func(error) error {
|
|
now := time.Now()
|
|
parent, ok := trace.FromContext(t.ctx)
|
|
if !ok {
|
|
return func(err error) error { return err }
|
|
}
|
|
span := parent.Fork(_traceFamily, "Memcache:"+action)
|
|
span.SetTag(
|
|
trace.String(trace.TagSpanKind, _traceSpanKind),
|
|
trace.String(trace.TagComponent, _traceComponentName),
|
|
trace.String(trace.TagPeerService, _tracePeerService),
|
|
trace.String(trace.TagPeerAddress, t.address),
|
|
trace.String(trace.TagDBStatement, action+" "+statement),
|
|
)
|
|
return func(err error) error {
|
|
span.Finish(&err)
|
|
t := time.Since(now)
|
|
if t > _slowLogDuration {
|
|
log.Warn("%s slow log action: %s key: %s time: %v", _traceFamily, action, statement, t)
|
|
}
|
|
return err
|
|
}
|
|
}
|
|
|
|
func (t *traceConn) WithContext(ctx context.Context) Conn {
|
|
t.ctx = ctx
|
|
t.Conn = t.Conn.WithContext(ctx)
|
|
return t
|
|
}
|
|
|
|
func (t *traceConn) Add(item *Item) error {
|
|
finishFn := t.setTrace("Add", item.Key)
|
|
return finishFn(t.Conn.Add(item))
|
|
}
|
|
|
|
func (t *traceConn) Set(item *Item) error {
|
|
finishFn := t.setTrace("Set", item.Key)
|
|
return finishFn(t.Conn.Set(item))
|
|
}
|
|
|
|
func (t *traceConn) Replace(item *Item) error {
|
|
finishFn := t.setTrace("Replace", item.Key)
|
|
return finishFn(t.Conn.Replace(item))
|
|
}
|
|
|
|
func (t *traceConn) Get(key string) (*Item, error) {
|
|
finishFn := t.setTrace("Get", key)
|
|
item, err := t.Conn.Get(key)
|
|
return item, finishFn(err)
|
|
}
|
|
|
|
func (t *traceConn) GetMulti(keys []string) (map[string]*Item, error) {
|
|
finishFn := t.setTrace("GetMulti", strings.Join(keys, " "))
|
|
items, err := t.Conn.GetMulti(keys)
|
|
return items, finishFn(err)
|
|
}
|
|
|
|
func (t *traceConn) Delete(key string) error {
|
|
finishFn := t.setTrace("Delete", key)
|
|
return finishFn(t.Conn.Delete(key))
|
|
}
|
|
|
|
func (t *traceConn) Increment(key string, delta uint64) (newValue uint64, err error) {
|
|
finishFn := t.setTrace("Increment", key+" "+strconv.FormatUint(delta, 10))
|
|
newValue, err = t.Conn.Increment(key, delta)
|
|
return newValue, finishFn(err)
|
|
}
|
|
|
|
func (t *traceConn) Decrement(key string, delta uint64) (newValue uint64, err error) {
|
|
finishFn := t.setTrace("Decrement", key+" "+strconv.FormatUint(delta, 10))
|
|
newValue, err = t.Conn.Decrement(key, delta)
|
|
return newValue, finishFn(err)
|
|
}
|
|
|
|
func (t *traceConn) CompareAndSwap(item *Item) error {
|
|
finishFn := t.setTrace("CompareAndSwap", item.Key)
|
|
return finishFn(t.Conn.CompareAndSwap(item))
|
|
}
|
|
|
|
func (t *traceConn) Touch(key string, seconds int32) (err error) {
|
|
finishFn := t.setTrace("Touch", key+" "+strconv.Itoa(int(seconds)))
|
|
return finishFn(t.Conn.Touch(key, seconds))
|
|
}
|
|
|