mirror of
https://github.com/HDT3213/godis.git
synced 2025-09-26 21:01:17 +08:00
77 lines
1.7 KiB
Go
77 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/hdt3213/godis/cluster"
|
|
"github.com/hdt3213/godis/config"
|
|
"github.com/hdt3213/godis/database"
|
|
idatabase "github.com/hdt3213/godis/interface/database"
|
|
"github.com/hdt3213/godis/lib/logger"
|
|
"github.com/hdt3213/godis/lib/utils"
|
|
"github.com/hdt3213/godis/redis/server/gnet"
|
|
stdserver "github.com/hdt3213/godis/redis/server/std"
|
|
)
|
|
|
|
var banner = `
|
|
______ ___
|
|
/ ____/___ ____/ (_)____
|
|
/ / __/ __ \/ __ / / ___/
|
|
/ /_/ / /_/ / /_/ / (__ )
|
|
\____/\____/\__,_/_/____/
|
|
`
|
|
|
|
var defaultProperties = &config.ServerProperties{
|
|
Bind: "0.0.0.0",
|
|
Port: 6399,
|
|
AppendOnly: false,
|
|
AppendFilename: "",
|
|
MaxClients: 1000,
|
|
RunID: utils.RandString(40),
|
|
}
|
|
|
|
func fileExists(filename string) bool {
|
|
info, err := os.Stat(filename)
|
|
return err == nil && !info.IsDir()
|
|
}
|
|
|
|
func main() {
|
|
print(banner)
|
|
logger.Setup(&logger.Settings{
|
|
Path: "logs",
|
|
Name: "godis",
|
|
Ext: "log",
|
|
TimeFormat: "2006-01-02",
|
|
})
|
|
configFilename := os.Getenv("CONFIG")
|
|
if configFilename == "" {
|
|
if fileExists("redis.conf") {
|
|
config.SetupConfig("redis.conf")
|
|
} else {
|
|
config.Properties = defaultProperties
|
|
}
|
|
} else {
|
|
config.SetupConfig(configFilename)
|
|
}
|
|
listenAddr := fmt.Sprintf("%s:%d", config.Properties.Bind, config.Properties.Port)
|
|
|
|
var err error
|
|
if config.Properties.UseGnet {
|
|
var db idatabase.DB
|
|
if config.Properties.ClusterEnable {
|
|
db = cluster.MakeCluster()
|
|
} else {
|
|
db = database.NewStandaloneServer()
|
|
}
|
|
server := gnet.NewGnetServer(db)
|
|
err = server.Run(listenAddr)
|
|
} else {
|
|
handler := stdserver.MakeHandler()
|
|
err = stdserver.Serve(listenAddr, handler)
|
|
}
|
|
if err != nil {
|
|
logger.Errorf("start server failed: %v", err)
|
|
}
|
|
}
|