forked from mirror/ledisdb
121 lines
2.6 KiB
Go
121 lines
2.6 KiB
Go
package cmd
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"runtime"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"github.com/ledisdb/ledisdb/config"
|
|
"github.com/ledisdb/ledisdb/server"
|
|
)
|
|
|
|
var addr = flag.String("addr", "", "ledisdb listen address")
|
|
var dataDir = flag.String("data_dir", "", "ledisdb base data dir")
|
|
var dbName = flag.String("db_name", "", "select a db to use, it will overwrite the config's db name")
|
|
var usePprof = flag.Bool("pprof", false, "enable pprof")
|
|
var pprofPort = flag.Int("pprof_port", 6060, "pprof http port")
|
|
var slaveof = flag.String("slaveof", "", "make the server a slave of another instance")
|
|
var readonly = flag.Bool("readonly", false, "set readonly mode, slave server is always readonly")
|
|
var rpl = flag.Bool("rpl", false, "enable replication or not, slave server is always enabled")
|
|
var rplSync = flag.Bool("rpl_sync", false, "enable sync replication or not")
|
|
var ttlCheck = flag.Int("ttl_check", 0, "TTL check interval")
|
|
var databases = flag.Int("databases", 0, "ledisdb maximum database number")
|
|
|
|
func Server() {
|
|
var configFile = flag.String("config", "", "ledisdb config file")
|
|
|
|
runtime.GOMAXPROCS(runtime.NumCPU())
|
|
|
|
flag.Parse()
|
|
|
|
var cfg *config.Config
|
|
var err error
|
|
|
|
if len(*configFile) == 0 {
|
|
println("no config set, using default config")
|
|
cfg = config.NewConfigDefault()
|
|
} else {
|
|
cfg, err = config.NewConfigWithFile(*configFile)
|
|
}
|
|
|
|
if err != nil {
|
|
println(err.Error())
|
|
return
|
|
}
|
|
|
|
if len(*addr) > 0 {
|
|
cfg.Addr = *addr
|
|
}
|
|
|
|
if len(*dataDir) > 0 {
|
|
cfg.DataDir = *dataDir
|
|
}
|
|
|
|
if len(*dbName) > 0 {
|
|
cfg.DBName = *dbName
|
|
}
|
|
|
|
if *databases > 0 {
|
|
cfg.Databases = *databases
|
|
}
|
|
|
|
// check bool flag, use it.
|
|
for _, arg := range os.Args {
|
|
arg := strings.ToLower(arg)
|
|
switch arg {
|
|
case "-rpl", "-rpl=true", "-rpl=false":
|
|
cfg.UseReplication = *rpl
|
|
case "-readonly", "-readonly=true", "-readonly=false":
|
|
cfg.Readonly = *readonly
|
|
case "-rpl_sync", "-rpl_sync=true", "-rpl_sync=false":
|
|
cfg.Replication.Sync = *rplSync
|
|
}
|
|
}
|
|
|
|
if len(*slaveof) > 0 {
|
|
cfg.SlaveOf = *slaveof
|
|
cfg.Readonly = true
|
|
cfg.UseReplication = true
|
|
}
|
|
|
|
if *ttlCheck > 0 {
|
|
cfg.TTLCheckInterval = *ttlCheck
|
|
}
|
|
|
|
var app *server.App
|
|
app, err = server.NewApp(cfg)
|
|
if err != nil {
|
|
println(err.Error())
|
|
return
|
|
}
|
|
|
|
sc := make(chan os.Signal, 1)
|
|
signal.Notify(sc,
|
|
os.Kill,
|
|
os.Interrupt,
|
|
syscall.SIGHUP,
|
|
syscall.SIGINT,
|
|
syscall.SIGTERM,
|
|
syscall.SIGQUIT)
|
|
|
|
if *usePprof {
|
|
go func() {
|
|
log.Println(http.ListenAndServe(fmt.Sprintf(":%d", *pprofPort), nil))
|
|
}()
|
|
}
|
|
|
|
go app.Run()
|
|
|
|
<-sc
|
|
|
|
println("ledis-server is closing")
|
|
app.Close()
|
|
println("ledis-server is closed")
|
|
}
|