Compare commits
No commits in common. "v1.3.11" and "master" have entirely different histories.
15
go.sum
15
go.sum
@ -1,15 +0,0 @@
|
||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||
github.com/gomodule/redigo v1.9.2 h1:HrutZBLhSIU8abiSfW8pj8mPhOyMYjZT/wcA4/L9L9s=
|
||||
github.com/gomodule/redigo v1.9.2/go.mod h1:KsU3hiK/Ay8U42qpaJk+kuNa3C+spxapWpM+ywhcgtw=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/stretchr/testify v1.8.4 h1:CcVxjf3Q8PM0mHUKJCdn+eZZtm5yQwehR5yeSVQQcUk=
|
||||
github.com/tylerb/gls v0.0.0-20150407001822-e606233f194d h1:yYYPFFlbqxF5mrj5sEfETtM/Ssz2LTy0/VKlDdXYctc=
|
||||
github.com/tylerb/gls v0.0.0-20150407001822-e606233f194d/go.mod h1:0MwyId/pXK5wkYYEXe7NnVknX+aNBuF73fLV3U0reU8=
|
||||
github.com/tylerb/is v2.1.4+incompatible h1:BMf2zP0kY2Ykzx2W1fDrjwKj1x1B4E0mELkpjaNy1tM=
|
||||
github.com/tylerb/is v2.1.4+incompatible/go.mod h1:3Bw2NWEEe8Kx7/etYqgm9ug53iNDgabnloch75jjOSc=
|
||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||
qoobing.com/gomod/log v1.2.8 h1:J1L87VhDMHvCgPZH2H/6QuwTLiMorJL97xVx7FJyi2g=
|
||||
qoobing.com/gomod/log v1.2.8/go.mod h1:/ZTN/ukAbSqRb4eMlF9LpfkVgM21xwprbd5y3tcQxpM=
|
||||
qoobing.com/gomod/str v1.0.1/go.mod h1:gbhN2dba/P5gFRGVJvEI57KEJLlMHHAd6Kuuxn4GlMY=
|
||||
qoobing.com/gomod/str v1.0.5 h1:AXEB8k/yhepLK5jVez+WL4sWVuCFb8pWAgmo3nvt96A=
|
||||
qoobing.com/gomod/str v1.0.5/go.mod h1:gbhN2dba/P5gFRGVJvEI57KEJLlMHHAd6Kuuxn4GlMY=
|
||||
@ -18,6 +18,7 @@ import (
|
||||
"bytes"
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
"time"
|
||||
|
||||
"github.com/gomodule/redigo/redis"
|
||||
@ -25,33 +26,37 @@ import (
|
||||
|
||||
var (
|
||||
_ redis.ConnWithTimeout = (*loggingConn)(nil)
|
||||
logCallDepth = 3
|
||||
)
|
||||
|
||||
type Logger interface {
|
||||
Debugf(format string, v ...interface{})
|
||||
Errorf(format string, v ...interface{})
|
||||
}
|
||||
|
||||
type loggingConn struct {
|
||||
redis.Conn
|
||||
logger Logger
|
||||
prefix string
|
||||
logskip func(cmdName string) bool
|
||||
}
|
||||
|
||||
// NewLoggingConn returns a logging wrapper around a connection.
|
||||
func NewLoggingConn(conn redis.Conn, logger Logger, prefix string) redis.Conn {
|
||||
func NewLoggingConn(conn redis.Conn, logger *log.Logger, prefix string) redis.Conn {
|
||||
if prefix != "" {
|
||||
prefix = prefix + "."
|
||||
}
|
||||
return &loggingConn{conn, logger, prefix, nil}
|
||||
}
|
||||
|
||||
// NewLoggingConnFilter returns a logging wrapper around a connection and a filter function.
|
||||
func NewLoggingConnFilter(conn redis.Conn, logger Logger, prefix string, skip func(cmdName string) bool) redis.Conn {
|
||||
func NewLoggingConnFilter(conn redis.Conn, logger *log.Logger, prefix string, skip func(cmdName string) bool) redis.Conn {
|
||||
if prefix != "" {
|
||||
prefix = prefix + "."
|
||||
}
|
||||
return &loggingConn{conn, logger, prefix, skip}
|
||||
}
|
||||
|
||||
type loggingConn struct {
|
||||
redis.Conn
|
||||
logger *log.Logger
|
||||
prefix string
|
||||
skip func(cmdName string) bool
|
||||
}
|
||||
|
||||
func (c *loggingConn) Close() error {
|
||||
err := c.Conn.Close()
|
||||
c.logger.Debugf("%sClose() -> (%v)", c.prefix, err)
|
||||
var buf bytes.Buffer
|
||||
fmt.Fprintf(&buf, "%sClose() -> (%v)", c.prefix, err)
|
||||
c.logger.Output(logCallDepth, buf.String()) // nolint: errcheck
|
||||
return err
|
||||
}
|
||||
|
||||
@ -93,7 +98,7 @@ func (c *loggingConn) printValue(buf *bytes.Buffer, v interface{}) {
|
||||
}
|
||||
|
||||
func (c *loggingConn) print(method, commandName string, args []interface{}, reply interface{}, err error) {
|
||||
if c.logskip != nil && c.logskip(commandName) {
|
||||
if c.skip != nil && c.skip(commandName) {
|
||||
return
|
||||
}
|
||||
var buf bytes.Buffer
|
||||
@ -106,15 +111,12 @@ func (c *loggingConn) print(method, commandName string, args []interface{}, repl
|
||||
}
|
||||
}
|
||||
buf.WriteString(") -> (")
|
||||
if err != nil {
|
||||
fmt.Fprintf(&buf, "error: %v)", err)
|
||||
} else {
|
||||
if method != "Send" {
|
||||
c.printValue(&buf, reply)
|
||||
buf.WriteString(", ")
|
||||
}
|
||||
buf.WriteString(")")
|
||||
}
|
||||
c.logger.Debugf("%s", buf.String())
|
||||
fmt.Fprintf(&buf, "%v)", err)
|
||||
c.logger.Output(logCallDepth+1, buf.String()) // nolint: errcheck
|
||||
}
|
||||
|
||||
func (c *loggingConn) Do(commandName string, args ...interface{}) (interface{}, error) {
|
||||
|
||||
34
redis.go
34
redis.go
@ -1,39 +1,11 @@
|
||||
package redis
|
||||
|
||||
import (
|
||||
redigo "github.com/gomodule/redigo/redis"
|
||||
"qoobing.com/gomod/redis/redis"
|
||||
"qoobing.com/gomod/redis/sentinel"
|
||||
)
|
||||
|
||||
type (
|
||||
Conn = redigo.Conn
|
||||
Pool = redigo.Pool
|
||||
Config = sentinel.Config
|
||||
)
|
||||
type Config = sentinel.Config
|
||||
|
||||
var (
|
||||
NewRedisPool = redis.NewPool
|
||||
NewSentinelPool = sentinel.NewPool
|
||||
Int = redigo.Int
|
||||
Int64 = redigo.Int64
|
||||
Uint64 = redigo.Uint64
|
||||
Float64 = redigo.Float64
|
||||
String = redigo.String
|
||||
Strings = redigo.Strings
|
||||
Bool = redigo.Bool
|
||||
Ints = redigo.Ints
|
||||
Float64s = redigo.Float64s
|
||||
Values = redigo.Values
|
||||
ErrNil = redigo.ErrNil
|
||||
ErrPoolExhausted = redigo.ErrPoolExhausted
|
||||
)
|
||||
|
||||
func NewPool(cfg Config) *redigo.Pool {
|
||||
if cfg.Master != "" {
|
||||
return NewRedisPool(cfg)
|
||||
} else if cfg.MasterName != "" {
|
||||
return NewSentinelPool(cfg)
|
||||
}
|
||||
panic("invalid config: Master & MasterName are both empty")
|
||||
}
|
||||
var NewPool = redis.NewPool
|
||||
var NewSentinelPool = sentinel.NewPool
|
||||
|
||||
@ -2,10 +2,10 @@ package redis
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"qoobing.com/gomod/log"
|
||||
|
||||
"github.com/gomodule/redigo/redis"
|
||||
"qoobing.com/gomod/redis/logging"
|
||||
"qoobing.com/gomod/redis/sentinel"
|
||||
@ -52,7 +52,10 @@ func NewPool(cfg Config) *redis.Pool {
|
||||
|
||||
var (
|
||||
logPrefix = "redis"
|
||||
logLogger = log.New("redis")
|
||||
logStdPrefix = "DBUG "
|
||||
logStdWriter = os.Stdout
|
||||
logStdFlags = log.Ldate | log.Lmicroseconds | log.Lshortfile
|
||||
logStdLogger = log.New(logStdWriter, logStdPrefix, logStdFlags)
|
||||
)
|
||||
return &redis.Pool{
|
||||
MaxIdle: *cfg.MaxIdle,
|
||||
@ -62,7 +65,6 @@ func NewPool(cfg Config) *redis.Pool {
|
||||
Dial: func() (redis.Conn, error) {
|
||||
c, err := redis.Dial("tcp", masterAddr)
|
||||
if err != nil {
|
||||
logLogger.Errorf("connect [%s] error: %s", masterAddr, err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
@ -74,15 +76,21 @@ func NewPool(cfg Config) *redis.Pool {
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
logLogger.Errorf("auth [%s] error: %s", masterAddr, err)
|
||||
return nil, fmt.Errorf("redis master AUTH failed: <%s>", err.Error())
|
||||
} else if okstr != "OK" {
|
||||
logLogger.Errorf("auth [%s] not return ok but '%s'", masterAddr, okstr)
|
||||
return nil, fmt.Errorf("redis master AUTH failed: <%s>", okstr)
|
||||
}
|
||||
|
||||
//// if !TestRole(c, "master") {
|
||||
//// c.Close()
|
||||
//// err = fmt.Errorf(
|
||||
//// "master(%s) got by name '%s' is not redis master",
|
||||
//// masterAddr, masterName)
|
||||
//// return nil, err
|
||||
//// }
|
||||
|
||||
if cfg.Debug {
|
||||
c = logging.NewLoggingConn(c, logLogger, logPrefix)
|
||||
c = logging.NewLoggingConn(c, logStdLogger, logPrefix)
|
||||
}
|
||||
return c, nil
|
||||
},
|
||||
|
||||
@ -3,13 +3,14 @@ package sentinel
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"log"
|
||||
"net"
|
||||
"os"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/gomodule/redigo/redis"
|
||||
"qoobing.com/gomod/log"
|
||||
"qoobing.com/gomod/redis/logging"
|
||||
)
|
||||
|
||||
@ -28,15 +29,6 @@ type Config struct {
|
||||
IdleTimeout *int `toml:"idle_timeout"` //空闲超时时间
|
||||
}
|
||||
|
||||
func (cfg Config) GetSecDsn() (dsn string) {
|
||||
if cfg.Master != "" {
|
||||
dsn += fmt.Sprintf("mode=redis address=%s", cfg.Master)
|
||||
} else if cfg.MasterName != "" {
|
||||
dsn += fmt.Sprintf("mode=sentinel address=%s mastername=%s", cfg.Sentinels, cfg.MasterName)
|
||||
}
|
||||
return dsn
|
||||
}
|
||||
|
||||
type Sentinel struct {
|
||||
// Addrs is a slice with known Sentinel addresses.
|
||||
Addrs []string
|
||||
@ -122,9 +114,13 @@ func NewPool(cfg Config) *redis.Pool {
|
||||
} else if *cfg.MaxActive < 0 {
|
||||
*cfg.MaxActive = 100
|
||||
}
|
||||
|
||||
var (
|
||||
logPrefix = "redis"
|
||||
logLogger = log.New("redis")
|
||||
logStdPrefix = "DBUG "
|
||||
logStdWriter = os.Stdout
|
||||
logStdFlags = log.Ldate | log.Lmicroseconds | log.Lshortfile
|
||||
logStdLogger = log.New(logStdWriter, logStdPrefix, logStdFlags)
|
||||
)
|
||||
return &redis.Pool{
|
||||
MaxIdle: *cfg.MaxIdle,
|
||||
@ -134,13 +130,11 @@ func NewPool(cfg Config) *redis.Pool {
|
||||
Dial: func() (redis.Conn, error) {
|
||||
masterAddr, err := sntnl.MasterAddr()
|
||||
if err != nil {
|
||||
logLogger.Errorf("get master from sentinel error: %s", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
c, err := redis.Dial("tcp", masterAddr)
|
||||
if err != nil {
|
||||
logLogger.Errorf("connect [%s] error: %s", masterAddr, err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
@ -152,10 +146,8 @@ func NewPool(cfg Config) *redis.Pool {
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
logLogger.Errorf("auth [%s] error: %s", masterAddr, err)
|
||||
return nil, fmt.Errorf("redis master AUTH failed: <%s>", err.Error())
|
||||
} else if okstr != "OK" {
|
||||
logLogger.Errorf("auth [%s] not return ok but '%s'", masterAddr, okstr)
|
||||
return nil, fmt.Errorf("redis master AUTH failed: <%s>", okstr)
|
||||
}
|
||||
|
||||
@ -164,12 +156,11 @@ func NewPool(cfg Config) *redis.Pool {
|
||||
err = fmt.Errorf(
|
||||
"master(%s) got by name '%s' is not redis master",
|
||||
masterAddr, masterName)
|
||||
logLogger.Errorf("%s", err.Error())
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if cfg.Debug {
|
||||
c = logging.NewLoggingConn(c, logLogger, logPrefix)
|
||||
c = logging.NewLoggingConn(c, logStdLogger, logPrefix)
|
||||
}
|
||||
return c, nil
|
||||
},
|
||||
|
||||
Loading…
Reference in New Issue
Block a user