package main import ( // "bufio" "flag" "fmt" // "io" "math/rand" "net" "os" // "strings" ) var options struct { host string port int password string } func usage(status int) { fmt.Println("usage: rsload [filename]") os.Exit(status) } func randomString(n int) string { var alpha = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789" buf := make([]byte, n) for i := 0; i < len(buf); i++ { buf[i] = alpha[rand.Intn(len(alpha)-1)] } return string(buf) } func main() { flag.Parse() args := flag.Args() if len(args) < 1 { usage(1) } fname := args[0] conn, err := net.Dial("tcp", fmt.Sprintf("%s:%d", options.host, options.port)) if err != nil { fmt.Printf("unable to connect to redis: %v\n", err) os.Exit(1) } defer conn.Close() if options.password != "" { auth(options.password).Write(conn) // fmt.Fprintf(conn, "*2\r\n$4\r\nauth\r\n$%d\r\n%s\r\n", len(options.password), options.password) v, err := readValue(conn) if err != nil { fmt.Printf("unable to auth: %v\n", err) os.Exit(1) } if !isOK(v) { fmt.Printf("not OK: %v\n", v) os.Exit(1) } } infile, err := os.Open(fname) if err != nil { fmt.Printf("unable to open file %s: %v\n", fname, err) os.Exit(1) } defer infile.Close() c, e := make(chan value), make(chan error) sent := make(chan value) go streamValues(infile, c, e) go func() { for { select { case v, ok := <-c: if !ok { return } v.Write(conn) sent <- v case err, ok := <-e: if !ok { return } fmt.Println(err) } } }() type pair struct { request value response value } cc, ee := make(chan value), make(chan error) go streamValues(conn, cc, ee) ReadResponses: for { select { case response, ok := <-cc: if !ok { break ReadResponses } request := <-sent fmt.Println(pair{request, response}) case err, ok := <-ee: if !ok { break ReadResponses } request := <-sent fmt.Printf("fuck %v %v\n", request, err) } } } func init() { flag.StringVar(&options.host, "h", "127.0.0.1", "hostname") flag.IntVar(&options.port, "p", 6379, "port") flag.StringVar(&options.password, "a", "", "password") } /* -h Server hostname (default: 127.0.0.1) -p Server port (default: 6379) -s Server socket (overrides hostname and port) -a Password to use when connecting to the server -r Execute specified command N times -i When -r is used, waits seconds per command. It is possible to specify sub-second times like -i 0.1 -n Database number -x Read last argument from STDIN -d Multi-bulk delimiter in for raw formatting (default: \n) -c Enable cluster mode (follow -ASK and -MOVED redirections) --raw Use raw formatting for replies (default when STDOUT is not a tty) --latency Enter a special mode continuously sampling latency --latency-history Like --latency but tracking latency changes over time. Default time interval is 15 sec. Change it using -i. --slave Simulate a slave showing commands received from the master --rdb Transfer an RDB dump from remote server to local file. --pipe Transfer raw Redis protocol from stdin to server --bigkeys Sample Redis keys looking for big keys --eval Send an EVAL command using the Lua script at --help Output this help and exit --version Output version and exit */