ledisdb/server/app.go

116 lines
1.6 KiB
Go
Raw Normal View History

package server
2014-05-02 13:08:20 +04:00
import (
2014-06-06 07:25:13 +04:00
"fmt"
"github.com/siddontang/ledisdb/ledis"
2014-05-02 13:08:20 +04:00
"net"
2014-06-11 12:48:11 +04:00
"path"
2014-05-02 13:08:20 +04:00
"strings"
)
type App struct {
cfg *Config
listener net.Listener
2014-05-20 04:41:24 +04:00
ldb *ledis.Ledis
2014-05-08 06:54:33 +04:00
closed bool
quit chan struct{}
2014-06-08 12:43:59 +04:00
2014-06-11 12:48:11 +04:00
access *accessLog
2014-06-08 12:43:59 +04:00
//for slave replication
2014-06-09 13:23:32 +04:00
m *master
2014-05-02 13:08:20 +04:00
}
func NewApp(cfg *Config) (*App, error) {
2014-06-06 07:25:13 +04:00
if len(cfg.DataDir) == 0 {
return nil, fmt.Errorf("must set data_dir first")
}
if len(cfg.DB.DataDir) == 0 {
cfg.DB.DataDir = cfg.DataDir
}
2014-05-02 13:08:20 +04:00
app := new(App)
app.quit = make(chan struct{})
2014-05-08 06:54:33 +04:00
app.closed = false
2014-05-02 13:08:20 +04:00
app.cfg = cfg
var err error
if strings.Contains(cfg.Addr, "/") {
app.listener, err = net.Listen("unix", cfg.Addr)
} else {
app.listener, err = net.Listen("tcp", cfg.Addr)
}
if err != nil {
return nil, err
}
2014-06-11 12:48:11 +04:00
if len(cfg.AccessLog) > 0 {
if path.Dir(cfg.AccessLog) == "." {
app.access, err = newAcessLog(path.Join(cfg.DataDir, cfg.AccessLog))
} else {
app.access, err = newAcessLog(cfg.AccessLog)
}
if err != nil {
return nil, err
}
}
if app.ldb, err = ledis.OpenWithConfig(&cfg.DB); err != nil {
2014-05-02 13:08:20 +04:00
return nil, err
}
2014-06-09 13:23:32 +04:00
app.m = newMaster(app)
2014-05-02 13:08:20 +04:00
return app, nil
}
func (app *App) Close() {
2014-05-08 06:54:33 +04:00
if app.closed {
return
}
app.closed = true
close(app.quit)
2014-05-02 13:08:20 +04:00
app.listener.Close()
2014-06-09 13:23:32 +04:00
app.m.Close()
2014-06-11 12:48:11 +04:00
if app.access != nil {
app.access.Close()
}
2014-05-20 04:41:24 +04:00
app.ldb.Close()
2014-05-02 13:08:20 +04:00
}
func (app *App) Run() {
2014-06-08 12:43:59 +04:00
if len(app.cfg.SlaveOf) > 0 {
2014-06-09 13:23:32 +04:00
app.slaveof(app.cfg.SlaveOf)
}
2014-05-08 06:54:33 +04:00
for !app.closed {
2014-05-02 13:08:20 +04:00
conn, err := app.listener.Accept()
if err != nil {
continue
}
newClient(conn, app)
2014-05-02 13:08:20 +04:00
}
}
2014-06-11 10:52:34 +04:00
2014-06-11 12:48:11 +04:00
func (app *App) Ledis() *ledis.Ledis {
2014-06-11 10:52:34 +04:00
return app.ldb
}