Compare commits
9 Commits
Author | Date | ||
---|---|---|---|
![]() |
ace40d5502 | ||
![]() |
f25236549e | ||
![]() |
dcc42736da | ||
![]() |
66a8d078a0 | ||
![]() |
4089e4d6b7 | ||
![]() |
ee6659ccb1 | ||
![]() |
1e8c7645bb | ||
![]() |
82def756a1 | ||
![]() |
84f521b43d |
13
go.mod
13
go.mod
@ -1,3 +1,14 @@
|
||||
module qoobing.com/gomod/redis
|
||||
|
||||
go 1.16
|
||||
go 1.18
|
||||
|
||||
require (
|
||||
github.com/gomodule/redigo v1.9.2
|
||||
qoobing.com/gomod/log v1.2.8
|
||||
qoobing.com/gomod/str v1.0.5
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/tylerb/gls v0.0.0-20150407001822-e606233f194d // indirect
|
||||
github.com/tylerb/is v2.1.4+incompatible // indirect
|
||||
)
|
||||
|
@ -25,7 +25,8 @@ import (
|
||||
)
|
||||
|
||||
var (
|
||||
_ redis.ConnWithTimeout = (*loggingConn)(nil)
|
||||
_ redis.ConnWithTimeout = (*loggingConn)(nil)
|
||||
logCallDepth = 3
|
||||
)
|
||||
|
||||
// NewLoggingConn returns a logging wrapper around a connection.
|
||||
@ -55,7 +56,7 @@ func (c *loggingConn) Close() error {
|
||||
err := c.Conn.Close()
|
||||
var buf bytes.Buffer
|
||||
fmt.Fprintf(&buf, "%sClose() -> (%v)", c.prefix, err)
|
||||
c.logger.Output(2, buf.String()) // nolint: errcheck
|
||||
c.logger.Output(logCallDepth, buf.String()) // nolint: errcheck
|
||||
return err
|
||||
}
|
||||
|
||||
@ -115,7 +116,7 @@ func (c *loggingConn) print(method, commandName string, args []interface{}, repl
|
||||
buf.WriteString(", ")
|
||||
}
|
||||
fmt.Fprintf(&buf, "%v)", err)
|
||||
c.logger.Output(3, buf.String()) // nolint: errcheck
|
||||
c.logger.Output(logCallDepth+1, buf.String()) // nolint: errcheck
|
||||
}
|
||||
|
||||
func (c *loggingConn) Do(commandName string, args ...interface{}) (interface{}, error) {
|
||||
|
44
redis.go
44
redis.go
@ -1,45 +1,11 @@
|
||||
package redis
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/gomodule/redigo/redis"
|
||||
"qoobing.com/gomod/redis/redis"
|
||||
"qoobing.com/gomod/redis/sentinel"
|
||||
)
|
||||
|
||||
func newSentinelPool() *redis.Pool {
|
||||
sntnl := &sentinel.Sentinel{
|
||||
Addrs: []string{":26379", ":26380", ":26381"},
|
||||
MasterName: "mymaster",
|
||||
Dial: func(addr string) (redis.Conn, error) {
|
||||
timeout := 500 * time.Millisecond
|
||||
c, err := redis.DialTimeout("tcp", addr, timeout, timeout, timeout)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return c, nil
|
||||
},
|
||||
}
|
||||
return &redis.Pool{
|
||||
MaxIdle: 3,
|
||||
MaxActive: 64,
|
||||
Wait: true,
|
||||
IdleTimeout: 240 * time.Second,
|
||||
Dial: func() (redis.Conn, error) {
|
||||
masterAddr, err := sntnl.MasterAddr()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
c, err := redis.Dial("tcp", masterAddr)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if !sentinel.TestRole(c, "master") {
|
||||
c.Close()
|
||||
return nil, fmt.Errorf("%s is not redis master", masterAddr)
|
||||
}
|
||||
return c, nil
|
||||
},
|
||||
}
|
||||
}
|
||||
type Config = sentinel.Config
|
||||
|
||||
var NewPool = redis.NewPool
|
||||
var NewSentinelPool = sentinel.NewPool
|
||||
|
98
redis/redis.go
Normal file
98
redis/redis.go
Normal file
@ -0,0 +1,98 @@
|
||||
package redis
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"github.com/gomodule/redigo/redis"
|
||||
"qoobing.com/gomod/redis/logging"
|
||||
"qoobing.com/gomod/redis/sentinel"
|
||||
)
|
||||
|
||||
type Config = sentinel.Config
|
||||
|
||||
func NewPool(cfg Config) *redis.Pool {
|
||||
if cfg.Master == "" {
|
||||
panic("config is invalid: Master is empty")
|
||||
} else if cfg.Master != "" && cfg.MasterName != "" {
|
||||
//panic("config is invalid: Master & MasterName must not set both")
|
||||
}
|
||||
var (
|
||||
masterAddr = cfg.Master
|
||||
masterUsername = cfg.Username
|
||||
masterPassword = cfg.Password
|
||||
)
|
||||
if cfg.Wait == nil {
|
||||
cfg.Wait = new(bool)
|
||||
*cfg.Wait = true
|
||||
}
|
||||
|
||||
if cfg.IdleTimeout == nil {
|
||||
cfg.IdleTimeout = new(int)
|
||||
*cfg.IdleTimeout = 300
|
||||
} else if *cfg.IdleTimeout <= 0 {
|
||||
*cfg.IdleTimeout = 86400 * 365 * 10
|
||||
}
|
||||
|
||||
if cfg.MaxIdle == nil {
|
||||
cfg.MaxIdle = new(int)
|
||||
*cfg.MaxIdle = 100
|
||||
} else if *cfg.MaxIdle < 0 {
|
||||
*cfg.MaxIdle = 100
|
||||
}
|
||||
|
||||
if cfg.MaxActive == nil {
|
||||
cfg.MaxActive = new(int)
|
||||
*cfg.MaxActive = 100
|
||||
} else if *cfg.MaxActive < 0 {
|
||||
*cfg.MaxActive = 100
|
||||
}
|
||||
|
||||
var (
|
||||
logPrefix = "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,
|
||||
MaxActive: *cfg.MaxActive,
|
||||
Wait: *cfg.Wait,
|
||||
IdleTimeout: time.Duration(*cfg.IdleTimeout) * time.Second,
|
||||
Dial: func() (redis.Conn, error) {
|
||||
c, err := redis.Dial("tcp", masterAddr)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var okstr = "OK"
|
||||
if masterPassword != "" && masterUsername != "" {
|
||||
okstr, err = redis.String(c.Do("AUTH", masterUsername, masterPassword))
|
||||
} else if masterPassword != "" {
|
||||
okstr, err = redis.String(c.Do("AUTH", masterPassword))
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("redis master AUTH failed: <%s>", err.Error())
|
||||
} else if okstr != "OK" {
|
||||
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, logStdLogger, logPrefix)
|
||||
}
|
||||
return c, nil
|
||||
},
|
||||
}
|
||||
}
|
65
redis_test.go
Normal file
65
redis_test.go
Normal file
@ -0,0 +1,65 @@
|
||||
package redis
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"testing"
|
||||
|
||||
"qoobing.com/gomod/redis/sentinel"
|
||||
"qoobing.com/gomod/redis/xstream"
|
||||
)
|
||||
|
||||
func TestNewConsumer(t *testing.T) {
|
||||
cfg := xstream.Config{
|
||||
Group: "group_a",
|
||||
Stream: "testxstream_queue",
|
||||
Sentinel: sentinel.Config{
|
||||
Debug: true, //调试开关(会在日志打印REDIS语句)
|
||||
MasterName: "walletmaster", //REDIS主名称(一个哨兵集群可管理多个REDIS主从结构)
|
||||
Username: "web3wallet", //REDIS连接用户名
|
||||
Password: "web3onthemoon@2023#dev", //REDIS连接用户密码
|
||||
Sentinels: "sentinel-0.redis:5000", //哨兵节点列表,逗号分隔,一般配置三个
|
||||
SentinelUsername: "sentinel", //哨兵节点连接用户名,不填默认default
|
||||
SentinelPassword: "password@mhxzkhl#dev", //哨兵节点连接密码,不填认为集群无密码
|
||||
},
|
||||
}
|
||||
|
||||
for i := 0; i < 15; i++ {
|
||||
i := i
|
||||
name := fmt.Sprintf("name-%d", i)
|
||||
t.Run(name, func(t *testing.T) {
|
||||
t.Parallel()
|
||||
log.Println("name", name)
|
||||
if r := xstream.NewReader(cfg); r == nil {
|
||||
t.Error("create new reader failed")
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestNewPool(t *testing.T) {
|
||||
cfg := Config{
|
||||
Debug: true, //调试开关(会在日志打印REDIS语句)
|
||||
Master: "redis-0.redis:2379", //REDIS主
|
||||
Username: "web3wallet", //REDIS连接用户名
|
||||
Password: "web3onthemoon@2023#dev", //REDIS连接用户密码
|
||||
Sentinels: "sentinel-0.redis:5000", //哨兵节点列表,逗号分隔,一般配置三个
|
||||
SentinelUsername: "sentinel", //哨兵节点连接用户名,不填默认default
|
||||
SentinelPassword: "password@mhxzkhl#dev", //哨兵节点连接密码,不填认为集群无密码
|
||||
}
|
||||
pool := NewPool(cfg)
|
||||
if pool == nil {
|
||||
t.Errorf("new pool failed")
|
||||
}
|
||||
rds := pool.Get()
|
||||
if rds == nil {
|
||||
t.Errorf("get redis failed")
|
||||
}
|
||||
if ret, err := rds.Do("SET", "ABC", "123"); err != nil {
|
||||
t.Errorf("get redis failed: %s", err)
|
||||
} else {
|
||||
fmt.Println("ABC", ret)
|
||||
}
|
||||
fmt.Println(rds.Do("GET", "ABC"))
|
||||
|
||||
}
|
@ -15,15 +15,18 @@ import (
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
Debug bool `toml:"debug"` //调试开关(会在日志打印REDIS语句)
|
||||
Username string `toml:"username"` //REDIS连接用户名
|
||||
Password string `toml:"password"` //REDIS连接用户密码
|
||||
MasterName string `toml:"mastername"` //REDIS主名称(一个哨兵集群可管理多个REDIS主从结构)
|
||||
Sentinels string `toml:"sentinels"` //哨兵节点列表,逗号分隔,一般配置三个
|
||||
Wait *bool `toml:"wait"`
|
||||
MaxIdle *int `toml:"max_idle"`
|
||||
MaxActive *int `toml:"max_active"`
|
||||
IdleTimeout *int `toml:"idle_timeout"`
|
||||
Debug bool `toml:"debug"` //调试开关(会在日志打印REDIS语句)
|
||||
Master string `toml:"master"` //REDIS主host&port 与 MasterName 互斥
|
||||
MasterName string `toml:"mastername"` //REDIS主名称,值不为空是启动sentinal模式, 与Master互斥
|
||||
Username string `toml:"username"` //REDIS连接用户名
|
||||
Password string `toml:"password"` //REDIS连接用户密码
|
||||
Sentinels string `toml:"sentinels"` //哨兵节点列表,逗号分隔,一般配置三个
|
||||
SentinelUsername string `toml:"sentinel_username"` //哨兵节点连接用户名,不填默认default
|
||||
SentinelPassword string `toml:"sentinel_password"` //哨兵节点连接密码,不填认为集群无密码
|
||||
Wait *bool `toml:"wait"` //当无可用连接时,是否等待
|
||||
MaxIdle *int `toml:"max_idle"` //最大空闲连接数
|
||||
MaxActive *int `toml:"max_active"` //最大活跃连接数
|
||||
IdleTimeout *int `toml:"idle_timeout"` //空闲超时时间
|
||||
}
|
||||
|
||||
type Sentinel struct {
|
||||
@ -51,10 +54,18 @@ type Sentinel struct {
|
||||
}
|
||||
|
||||
func NewPool(cfg Config) *redis.Pool {
|
||||
if cfg.MasterName == "" {
|
||||
panic("config is invalid: MasterName is empty")
|
||||
} else if cfg.Master != "" && cfg.MasterName != "" {
|
||||
//panic("config is invalid: Master & MasterName must not set both")
|
||||
}
|
||||
var (
|
||||
masterName = cfg.MasterName
|
||||
masterPassword = cfg.Password
|
||||
sntnl = &Sentinel{
|
||||
masterName = cfg.MasterName
|
||||
masterUsername = cfg.Username
|
||||
masterPassword = cfg.Password
|
||||
sentinelUsername = cfg.SentinelUsername
|
||||
sentinelPassword = cfg.SentinelPassword
|
||||
sntnl = &Sentinel{
|
||||
Addrs: strings.Split(cfg.Sentinels, ","),
|
||||
MasterName: masterName,
|
||||
Dial: func(addr string) (redis.Conn, error) {
|
||||
@ -63,6 +74,18 @@ func NewPool(cfg Config) *redis.Pool {
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var okstr = "OK"
|
||||
if sentinelPassword != "" && sentinelUsername != "" {
|
||||
okstr, err = redis.String(c.Do("AUTH", sentinelUsername, sentinelPassword))
|
||||
} else if sentinelPassword != "" {
|
||||
okstr, err = redis.String(c.Do("AUTH", sentinelPassword))
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("redis sentinel AUTH failed: <%s>", err.Error())
|
||||
} else if okstr != "OK" {
|
||||
return nil, fmt.Errorf("redis sentinel AUTH failed: <%s>", okstr)
|
||||
}
|
||||
return c, nil
|
||||
},
|
||||
}
|
||||
@ -71,7 +94,6 @@ func NewPool(cfg Config) *redis.Pool {
|
||||
cfg.Wait = new(bool)
|
||||
*cfg.Wait = true
|
||||
}
|
||||
|
||||
if cfg.IdleTimeout == nil {
|
||||
cfg.IdleTimeout = new(int)
|
||||
*cfg.IdleTimeout = 300
|
||||
@ -93,6 +115,13 @@ func NewPool(cfg Config) *redis.Pool {
|
||||
*cfg.MaxActive = 100
|
||||
}
|
||||
|
||||
var (
|
||||
logPrefix = "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,
|
||||
MaxActive: *cfg.MaxActive,
|
||||
@ -109,7 +138,13 @@ func NewPool(cfg Config) *redis.Pool {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
okstr, err := redis.String(c.Do("AUTH", masterPassword))
|
||||
var okstr = "OK"
|
||||
if masterPassword != "" && masterUsername != "" {
|
||||
okstr, err = redis.String(c.Do("AUTH", masterUsername, masterPassword))
|
||||
} else if masterPassword != "" {
|
||||
okstr, err = redis.String(c.Do("AUTH", masterPassword))
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("redis master AUTH failed: <%s>", err.Error())
|
||||
} else if okstr != "OK" {
|
||||
@ -118,11 +153,14 @@ func NewPool(cfg Config) *redis.Pool {
|
||||
|
||||
if !TestRole(c, "master") {
|
||||
c.Close()
|
||||
return nil, fmt.Errorf("%s is not redis master", masterAddr)
|
||||
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, log.New(os.Stdout, "XXXXXXX", log.LstdFlags), "coreredis")
|
||||
c = logging.NewLoggingConn(c, logStdLogger, logPrefix)
|
||||
}
|
||||
return c, nil
|
||||
},
|
||||
|
372
xstream/reader.go
Normal file
372
xstream/reader.go
Normal file
@ -0,0 +1,372 @@
|
||||
package xstream
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"net"
|
||||
"time"
|
||||
|
||||
"github.com/gomodule/redigo/redis"
|
||||
"qoobing.com/gomod/log"
|
||||
"qoobing.com/gomod/redis/sentinel"
|
||||
"qoobing.com/gomod/str"
|
||||
)
|
||||
|
||||
//////////////////////////////////////////////////////////////////////////////////
|
||||
///////////////////////// example //////////////////////////////////////////////
|
||||
//////////////////////////////////////////////////////////////////////////////////
|
||||
////// const (
|
||||
////// CORESERVER_NEWACCOUNT_GROUP = "groupid-coreserver-newaccount"
|
||||
////// CORESERVER_NEWACCOUNT_QUEUE = "coreserver.newaccount"
|
||||
////// )
|
||||
//////
|
||||
////// type NewAccountMessage struct {
|
||||
////// Sign string `json:"sign"`
|
||||
////// Appid string `json:"appid"`
|
||||
////// Userid int64 `json:"userid"`
|
||||
////// Account string `json:"account"`
|
||||
////// CreateTime time.Time `json:"createtime"`
|
||||
////// }
|
||||
//////
|
||||
////// func Run(ctx context.Context) {
|
||||
////// var SLEEP_TIME = 10 * time.Second
|
||||
////// go func() {
|
||||
////// for {
|
||||
////// log.Infof("start run newaccount marketing engine ...")
|
||||
////// runNewAccountMarketingEngine(ctx)
|
||||
////// log.Infof("newaccount marketing engine exit unexpected, retry %s later again", SLEEP_TIME.String())
|
||||
////// time.Sleep(SLEEP_TIME)
|
||||
////// }
|
||||
////// }()
|
||||
////// }
|
||||
//////
|
||||
////// func runNewAccountMarketingEngine(ctx context.Context) {
|
||||
////// // r read message from redis
|
||||
////// cfg := config.Instance()
|
||||
////// r := xstream.NewReader(xstream.Config{
|
||||
////// Group: CORESERVER_NEWACCOUNT_GROUP,
|
||||
////// Stream: CORESERVER_NEWACCOUNT_QUEUE,
|
||||
////// Sentinel: cfg.CoreserverRedis,
|
||||
////// })
|
||||
////// defer r.Close()
|
||||
//////
|
||||
////// // w write message to kafka
|
||||
////// w := &kafka.Writer{
|
||||
////// Addr: kafka.TCP(cfg.MarketingKafka.Addresses...),
|
||||
////// Balancer: &kafka.LeastBytes{},
|
||||
////// }
|
||||
////// defer w.Close()
|
||||
//////
|
||||
////// var error_wait_time = 0 * time.Millisecond
|
||||
////// for {
|
||||
////// // Step 0. wait a moment if need
|
||||
////// if error_wait_time > 0 {
|
||||
////// time.Sleep(error_wait_time)
|
||||
////// }
|
||||
//////
|
||||
////// // Step 1. fetch message
|
||||
////// var err error = nil
|
||||
////// var msg xstream.Message
|
||||
////// var newacctMsg NewAccountMessage
|
||||
////// log.Infof("waiting fetch newaccount message from redis queue...")
|
||||
////// if msg, err = r.FetchMessage(ctx); err != nil {
|
||||
////// log.Errorf("failed fetch message from redis queue, err:%s", err)
|
||||
////// //error_wait_time = 1000 * time.Millisecond
|
||||
////// break
|
||||
////// }
|
||||
//////
|
||||
////// // Step 2. decode message
|
||||
////// if err := json.Unmarshal(msg.Value, &newacctMsg); err != nil {
|
||||
////// log.Errorf("failed json.unmarshal message[%s], err:%s", string(msg.Value), err)
|
||||
////// continue
|
||||
////// } else if newacctMsg.Userid == 0 || newacctMsg.Account == "" {
|
||||
////// log.Errorf("invalid newaccount message, userid=[%d]", newacctMsg.Userid)
|
||||
////// continue
|
||||
////// }
|
||||
////// log.Infof("fetch newaccount message success, account=[%s]", newacctMsg.Account)
|
||||
//////
|
||||
////// // Step 3. handle message
|
||||
////// // xxxxxxx handle message xxxxxxxxxxxxxxxxxxxxxxxx
|
||||
////// // xxxxxxx handle message xxxxxxxxxxxxxxxxxxxxxxxx
|
||||
////// // xxxxxxx handle message xxxxxxxxxxxxxxxxxxxxxxxx
|
||||
////// // xxxxxxx handle message xxxxxxxxxxxxxxxxxxxxxxxx
|
||||
////// // xxxxxxx handle message xxxxxxxxxxxxxxxxxxxxxxxx
|
||||
////// // Step 4. send ack to coreserver redis
|
||||
////// if err := r.CommitMessages(ctx, msg); err != nil {
|
||||
////// log.Errorf("failed commit message to redis, err:%s", err)
|
||||
////// error_wait_time = 1000 * time.Millisecond
|
||||
////// continue
|
||||
////// }
|
||||
////// }
|
||||
////// }
|
||||
///////////////////////////// end example ////////////////////////////////////////////
|
||||
|
||||
var (
|
||||
redisPools = map[string]*redis.Pool{}
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
Group string `toml:"group"`
|
||||
Stream string `toml:"stream"`
|
||||
Sentinel sentinel.Config `toml:"sentinel"`
|
||||
}
|
||||
|
||||
type Message struct {
|
||||
Id string
|
||||
Value []byte
|
||||
}
|
||||
|
||||
type xstreamReader struct {
|
||||
cfg Config
|
||||
id string
|
||||
name string
|
||||
conn redis.Conn
|
||||
connPool *redis.Pool
|
||||
closed bool
|
||||
restart bool
|
||||
}
|
||||
|
||||
func NewReader(cfg Config) (r *xstreamReader) {
|
||||
// redis pool
|
||||
poolId := cfg.Sentinel.Sentinels
|
||||
if _, ok := redisPools[poolId]; !ok {
|
||||
redisPools[poolId] = sentinel.NewPool(cfg.Sentinel)
|
||||
}
|
||||
|
||||
// initialize
|
||||
connPool := redisPools[poolId]
|
||||
r = &xstreamReader{
|
||||
cfg: cfg,
|
||||
conn: connPool.Get(),
|
||||
connPool: connPool,
|
||||
closed: false,
|
||||
restart: true,
|
||||
}
|
||||
r.id = str.GetRandomNumString(12)
|
||||
r.name = newName(r.conn, r.id, cfg.Group, cfg.Stream)
|
||||
if err := r.initGroup(); err != nil {
|
||||
panic(fmt.Sprintf("initGroup failed: error:%s", err))
|
||||
}
|
||||
|
||||
// heart beat
|
||||
go func() {
|
||||
beatconn := redisPools[poolId].Get()
|
||||
defer beatconn.Close()
|
||||
for !r.closed {
|
||||
id, err := redis.String(beatconn.Do("GETEX", r.name, "EX", 120))
|
||||
if err != nil {
|
||||
log.Warningf("redis GETEX failed, err:%s", err)
|
||||
r.Close()
|
||||
} else if id != r.id {
|
||||
log.Warningf("name id in redis (%s) not equal to local name id(%s)", id, r.id)
|
||||
r.Close()
|
||||
}
|
||||
time.Sleep(15 * time.Second)
|
||||
}
|
||||
}()
|
||||
|
||||
return r
|
||||
}
|
||||
|
||||
func newName(conn redis.Conn, id, group, xstream string) string {
|
||||
var (
|
||||
mkeys = []any{}
|
||||
prefix = fmt.Sprintf("xstream[%s]-group[%s]-", xstream, group)
|
||||
consumer = func(i int) string { return fmt.Sprintf("%s%03d", prefix, i) }
|
||||
consumers = map[string]string{}
|
||||
MAX_CONSUMER_NUM = 30
|
||||
)
|
||||
|
||||
// get all name
|
||||
for i := 0; i < MAX_CONSUMER_NUM; i++ {
|
||||
mkeys = append(mkeys, consumer(i))
|
||||
}
|
||||
ret, err := redis.Strings(conn.Do("MGET", mkeys...))
|
||||
if err != nil {
|
||||
log.Warningf("redis MGET error: %s", err)
|
||||
panic("redis MGET error")
|
||||
}
|
||||
for i, s := range ret {
|
||||
if s != "" {
|
||||
name := consumer(i)
|
||||
consumers[name] = s
|
||||
}
|
||||
}
|
||||
|
||||
var (
|
||||
i = 0
|
||||
retry = 0
|
||||
consumer_i = ""
|
||||
)
|
||||
LOOP:
|
||||
// get unused name
|
||||
for ; i < MAX_CONSUMER_NUM; i++ {
|
||||
if _, ok := consumers[consumer(i)]; !ok {
|
||||
consumer_i = consumer(i)
|
||||
break
|
||||
}
|
||||
}
|
||||
if i == MAX_CONSUMER_NUM {
|
||||
panic(fmt.Sprintf("too many consumer(more than %d)???", MAX_CONSUMER_NUM))
|
||||
}
|
||||
|
||||
// set to redis (try lock)
|
||||
success, err := redis.Int(conn.Do("SETNX", consumer_i, id))
|
||||
conn.Do("GETEX", consumer_i, "EX", 120) /* for SETNX failed forever */
|
||||
if err != nil && retry >= 2 {
|
||||
log.Warningf("redis SETNX(%s,%s) error: %s", consumer_i, id, err)
|
||||
panic(fmt.Sprintf("redis SETNX error after try [%d] times", retry))
|
||||
}
|
||||
|
||||
// return or retry if something error
|
||||
if err != nil /*&& retry < 2*/ {
|
||||
retry++
|
||||
log.Warningf("redis SETNX(%s,%s) error: %s, retry(%d)", consumer_i, id, err, retry)
|
||||
goto LOOP
|
||||
}
|
||||
if success == 0 {
|
||||
i++
|
||||
retry++
|
||||
log.Warningf("consumer name(%s) used by others, retry(%d)", consumer_i, retry)
|
||||
goto LOOP
|
||||
}
|
||||
log.Infof("success to get new consumer-name:[%s]", consumer_i)
|
||||
return consumer_i
|
||||
}
|
||||
|
||||
func (r *xstreamReader) Close() {
|
||||
// delete client consumer name
|
||||
id, err := redis.String(r.redisDo("GETEX", r.name, "EX", 120))
|
||||
if err == nil && id == r.id {
|
||||
r.redisDo("DEL", r.name)
|
||||
}
|
||||
|
||||
// clonse redis connection(put to pool)
|
||||
r.closed = true
|
||||
r.conn.Close()
|
||||
}
|
||||
|
||||
func (r *xstreamReader) fetchMessage() (ret []any, err error) {
|
||||
// func (r *xstreamReader) fetchMessage(ctx context.Context) (ret []any, err error) {
|
||||
args := []interface{}{
|
||||
"GROUP", r.cfg.Group, r.name,
|
||||
"COUNT", 1,
|
||||
"BLOCK", 864000000,
|
||||
"STREAMS", r.cfg.Stream,
|
||||
}
|
||||
if r.restart {
|
||||
args = append(args, "0-0")
|
||||
} else {
|
||||
args = append(args, ">")
|
||||
}
|
||||
|
||||
ret, err = redis.Values(r.redisDo("XREADGROUP", args...))
|
||||
|
||||
if err != nil {
|
||||
return ret, err
|
||||
} else if ret == nil {
|
||||
return ret, nil
|
||||
}
|
||||
if len(ret) != 1 {
|
||||
panic("len(ret) is not 1")
|
||||
}
|
||||
return ret, nil
|
||||
}
|
||||
|
||||
func (r *xstreamReader) FetchMessage(ctx context.Context) (msg Message, err error) {
|
||||
// func (r *xstreamReader) FetchMessage(ctx context.Context) (msg xstreamMessage, err error) {
|
||||
// Step 1. get result from redis xstream
|
||||
var ret, qmsg, msgs []any = nil, nil, nil
|
||||
for {
|
||||
ret, err = r.fetchMessage()
|
||||
//////////////////////////////////////////////////////////////////////
|
||||
////ret format example:
|
||||
//// [["coreserver.txneedrepair", [["1675180029481-0", ["a", "b"]]]]]
|
||||
//////////////////////////////////////////////////////////////////////
|
||||
if err != nil {
|
||||
log.Warningf("fetchMessage meassage err:%s", err)
|
||||
return msg, errors.New("fetchMessage meassage err")
|
||||
}
|
||||
if len(ret) == 0 {
|
||||
log.Warningf("redis stream result length is 0, which shoud bigger than 0")
|
||||
return msg, errors.New("redis fetchMessage ret length is 0")
|
||||
}
|
||||
|
||||
qmsg, _ = redis.Values(ret[0], nil) // ["coreserver.txneedrepair", [["1675180029481-0", ["a", "b"]]]]
|
||||
msgs, _ = redis.Values(qmsg[1], nil) // [["1675180029481-0", ["a", "b"]]]
|
||||
|
||||
if len(msgs) == 0 && r.restart {
|
||||
r.restart = false
|
||||
continue
|
||||
}
|
||||
if len(msgs) != 1 {
|
||||
log.Warningf("meassage length from xstream is not 1(but %d)", len(msgs))
|
||||
return msg, errors.New("no new msg")
|
||||
}
|
||||
break
|
||||
}
|
||||
|
||||
// Step 2. parse xstream item result
|
||||
var (
|
||||
amsg, _ = redis.Values(msgs[0], nil) // ["1675180029481-0", ["a", "b"]]
|
||||
rmsg, _ = redis.Values(amsg[1], nil) // ["a", "b"]
|
||||
msgid, _ = redis.String(amsg[0], nil) // "1675180029481-0"
|
||||
msgval, _ = redis.String(rmsg[1], nil) // "b"
|
||||
)
|
||||
msg.Id = msgid
|
||||
msg.Value = []byte(msgval)
|
||||
log.Infof("success fetched meassage '%s': %s", msg.Id, msgval)
|
||||
return msg, nil
|
||||
}
|
||||
|
||||
func (r *xstreamReader) CommitMessages(ctx context.Context, msg Message) (err error) {
|
||||
args := []interface{}{
|
||||
r.cfg.Stream, // key
|
||||
r.cfg.Group, // group
|
||||
msg.Id, // id
|
||||
}
|
||||
|
||||
if cnt, err := redis.Int(r.redisDo("XACK", args...)); err != nil {
|
||||
return err
|
||||
} else if cnt != 1 {
|
||||
log.Warningf("attention: 'XACK' return %d (not 1), but we trade as success", cnt)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *xstreamReader) redisDo(cmd string, args ...any) (reply any, err error) {
|
||||
if r.closed {
|
||||
return nil, net.ErrClosed
|
||||
} else if r.conn.Err() != nil {
|
||||
log.Warningf("redis connection closed/unavaliable, we will get new one from pool")
|
||||
oldconn := r.conn
|
||||
newconn := r.connPool.Get()
|
||||
r.conn = newconn
|
||||
oldconn.Close()
|
||||
}
|
||||
|
||||
return r.conn.Do(cmd, args...)
|
||||
}
|
||||
|
||||
func (r *xstreamReader) initGroup() error {
|
||||
groups, err := r.redisDo("XINfO", "GROUPS", r.cfg.Stream)
|
||||
groupInfosInf, _ := groups.([]interface{})
|
||||
if err != nil || len(groupInfosInf) == 0 {
|
||||
log.Infof("create group, name: %s", r.cfg.Group)
|
||||
_, err = r.redisDo("XGROUP", "CREATE", r.cfg.Stream, r.cfg.Group, "$", "MKSTREAM")
|
||||
return err
|
||||
}
|
||||
|
||||
for _, groupInfo := range groupInfosInf {
|
||||
groupInfoInf := groupInfo.([]interface{})
|
||||
groupName := string(groupInfoInf[1].([]uint8))
|
||||
log.Infof("group name: %s", groupName)
|
||||
if groupName == r.cfg.Group {
|
||||
return nil
|
||||
}
|
||||
}
|
||||
log.Infof("create group, name: %s", r.cfg.Group)
|
||||
_, err = r.redisDo("XGROUP", "CREATE", r.cfg.Stream, r.cfg.Group, "$", "MKSTREAM")
|
||||
return err
|
||||
}
|
20
xstream/writer.go
Normal file
20
xstream/writer.go
Normal file
@ -0,0 +1,20 @@
|
||||
package xstream
|
||||
|
||||
//// func Example() {
|
||||
//// var (
|
||||
//// m = model.NewModelDefault()
|
||||
//// queue = "coreserver.newaccount"
|
||||
//// msgval = map[string]interface{}{
|
||||
//// "sign": "",
|
||||
//// "appid": c.AppId,
|
||||
//// "userid": c.UserId,
|
||||
//// "account": input.Address,
|
||||
//// }
|
||||
//// msgstr, _ = json.Marshal(msgval)
|
||||
//// )
|
||||
//// defer m.Close()
|
||||
////
|
||||
//// if _, err = m.Redis.Do("XADD", queue, "*", "msg", msgstr); err != nil {
|
||||
//// log.Warningf("!!!!FAILED add newaccount to redis queue, msg is:'%s'", msgstr)
|
||||
//// }
|
||||
//// }
|
Loading…
Reference in New Issue
Block a user