mirror of https://github.com/ledisdb/ledisdb.git
63 lines
1.1 KiB
Go
63 lines
1.1 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bufio"
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
"github.com/siddontang/ledisdb/server"
|
||
|
|
"net"
|
||
|
|
"os"
|
||
|
|
)
|
||
|
|
|
||
|
|
var host = flag.String("host", "127.0.0.1", "ledis server host")
|
||
|
|
var port = flag.Int("port", 6380, "ledis server port")
|
||
|
|
var sock = flag.String("sock", "", "ledis unix socket domain")
|
||
|
|
var dumpFile = flag.String("o", "./ledis.dump", "dump file to save")
|
||
|
|
|
||
|
|
var fullSyncCmd = []byte("*1\r\n$8\r\nfullsync\r\n") //fullsync
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
flag.Parse()
|
||
|
|
|
||
|
|
var c net.Conn
|
||
|
|
var err error
|
||
|
|
var f *os.File
|
||
|
|
|
||
|
|
if f, err = os.OpenFile(*dumpFile, os.O_CREATE|os.O_WRONLY, os.ModePerm); err != nil {
|
||
|
|
println(err.Error())
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
defer f.Close()
|
||
|
|
|
||
|
|
if len(*sock) != 0 {
|
||
|
|
c, err = net.Dial("unix", *sock)
|
||
|
|
} else {
|
||
|
|
addr := fmt.Sprintf("%s:%d", *host, *port)
|
||
|
|
c, err = net.Dial("tcp", addr)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
println(err.Error())
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
defer c.Close()
|
||
|
|
|
||
|
|
println("dump begin")
|
||
|
|
|
||
|
|
if _, err = c.Write(fullSyncCmd); err != nil {
|
||
|
|
println(err.Error())
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
rb := bufio.NewReaderSize(c, 16*1024)
|
||
|
|
|
||
|
|
if err = server.ReadBulkTo(rb, f); err != nil {
|
||
|
|
println(err.Error())
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
println("dump end")
|
||
|
|
}
|