redis-shake工具
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.
 
 
 

114 lines
2.3 KiB

package client
import (
"bufio"
"crypto/tls"
"github.com/alibaba/RedisShake/internal/client/proto"
"github.com/alibaba/RedisShake/internal/log"
"net"
"time"
)
type Redis struct {
reader *bufio.Reader
writer *bufio.Writer
protoReader *proto.Reader
protoWriter *proto.Writer
}
func NewRedisClient(address string, username string, password string, isTls bool) *Redis {
r := new(Redis)
var conn net.Conn
var dialer net.Dialer
var err error
dialer.Timeout = 3 * time.Second
if isTls {
conn, err = tls.DialWithDialer(&dialer, "tcp", address, &tls.Config{InsecureSkipVerify: true})
} else {
conn, err = dialer.Dial("tcp", address)
}
if err != nil {
log.PanicError(err)
}
r.reader = bufio.NewReader(conn)
r.writer = bufio.NewWriter(conn)
r.protoReader = proto.NewReader(r.reader)
r.protoWriter = proto.NewWriter(r.writer)
// auth
if password != "" {
var reply string
if username != "" {
reply = r.DoWithStringReply("auth", username, password)
} else {
reply = r.DoWithStringReply("auth", password)
}
if reply != "OK" {
log.Panicf("auth failed with reply: %s", reply)
}
log.Infof("auth successful. address=[%s]", address)
} else {
log.Infof("no password. address=[%s]", address)
}
// ping to test connection
reply := r.DoWithStringReply("ping")
if reply != "PONG" {
panic("ping failed with reply: " + reply)
}
return r
}
func (r *Redis) DoWithStringReply(args ...string) string {
r.Send(args...)
replyInterface, err := r.Receive()
if err != nil {
log.PanicError(err)
}
reply := replyInterface.(string)
return reply
}
func (r *Redis) Send(args ...string) {
argsInterface := make([]interface{}, len(args))
for inx, item := range args {
argsInterface[inx] = item
}
err := r.protoWriter.WriteArgs(argsInterface)
if err != nil {
log.PanicError(err)
}
r.flush()
}
func (r *Redis) SendBytes(buf []byte) {
_, err := r.writer.Write(buf)
if err != nil {
log.PanicError(err)
}
r.flush()
}
func (r *Redis) flush() {
err := r.writer.Flush()
if err != nil {
log.PanicError(err)
}
}
func (r *Redis) Receive() (interface{}, error) {
return r.protoReader.ReadReply()
}
func (r *Redis) BufioReader() *bufio.Reader {
return r.reader
}
func (r *Redis) SetBufioReader(rd *bufio.Reader) {
r.reader = rd
r.protoReader = proto.NewReader(r.reader)
}