ledisdb/server/cmd_server.go

package server

import (
	"github.com/siddontang/go/hack"
	"github.com/siddontang/go/num"

	"strconv"
	"strings"
	"time"

	"git.internal/re/ledisdb/config"
)

func pingCommand(c *client) error {
	c.resp.writeStatus(PONG)
	return nil
}

func defaultAuth(c *config.Config, password string) bool {
	return c.AuthPassword == password
}

func authCommand(c *client) error {
	if len(c.args) != 1 {
		return ErrCmdParams
	}

	method := defaultAuth
	if c.app.cfg.AuthMethod != nil {
		method = c.app.cfg.AuthMethod
	}

	if method(c.app.cfg, string(c.args[0])) {
		c.isAuthed = true
		c.resp.writeStatus(OK)
		return nil
	}
	c.isAuthed = false
	return ErrAuthenticationFailure
}

func echoCommand(c *client) error {
	if len(c.args) != 1 {
		return ErrCmdParams
	}

	c.resp.writeBulk(c.args[0])
	return nil
}

func selectCommand(c *client) error {
	if len(c.args) != 1 {
		return ErrCmdParams
	}

	index, err := strconv.Atoi(hack.String(c.args[0]))
	if err != nil {
		return err
	}
	// if c.db.IsInMulti() {
	// 	if err := c.script.Select(index); err != nil {
	// 		return err
	// 	} else {
	// 		c.db = c.script.DB
	// 	}
	// } else {
	// 	if db, err := c.ldb.Select(index); err != nil {
	// 		return err
	// 	} else {
	// 		c.db = db
	// 	}
	// }

	db, err := c.ldb.Select(index)
	if err != nil {
		return err
	}
	c.db = db
	c.resp.writeStatus(OK)

	return nil
}

func infoCommand(c *client) error {
	if len(c.args) > 1 {
		return ErrCmdParams
	}
	var section string
	if len(c.args) == 1 {
		section = strings.ToLower(hack.String(c.args[0]))
	}

	buf := c.app.info.Dump(section)
	c.resp.writeBulk(buf)

	return nil
}

func flushallCommand(c *client) error {
	err := c.ldb.FlushAll()
	if err != nil {
		return err
	}

	//we will restart the replication from master if possible
	c.app.tryReSlaveof()

	c.resp.writeStatus(OK)
	return nil
}

func flushdbCommand(c *client) error {
	_, err := c.db.FlushAll()
	if err != nil {
		return err
	}

	c.resp.writeStatus(OK)
	return nil
}

func timeCommand(c *client) error {
	if len(c.args) != 0 {
		return ErrCmdParams
	}

	t := time.Now()

	//seconds
	s := t.Unix()
	n := t.UnixNano()

	//micro seconds
	m := (n - s*1e9) / 1e3

	ay := []interface{}{
		num.FormatInt64ToSlice(s),
		num.FormatInt64ToSlice(m),
	}

	c.resp.writeArray(ay)
	return nil
}

func configGetCommand(c *client) error {
	args := c.args
	if len(args) != 2 {
		return ErrCmdParams
	}

	ay := make([][]byte, 0, 2)
	key := hack.String(args[1])
	switch key {
	case "databases":
		ay = append(ay, []byte("databases"), num.FormatIntToSlice(c.app.cfg.Databases))
	}

	c.resp.writeSliceArray(ay)
	return nil
}

func configCommand(c *client) error {
	if len(c.args) < 1 {
		return ErrCmdParams
	}

	switch strings.ToLower(hack.String(c.args[0])) {
	case "rewrite":
		if err := c.app.cfg.Rewrite(); err != nil {
			return err
		}
		c.resp.writeStatus(OK)
		return nil
	case "get":
		return configGetCommand(c)
	default:
		return ErrCmdParams
	}
}

func init() {
	register("auth", authCommand)
	register("ping", pingCommand)
	register("echo", echoCommand)
	register("select", selectCommand)
	register("info", infoCommand)
	register("flushall", flushallCommand)
	register("flushdb", flushdbCommand)
	register("time", timeCommand)
	register("config", configCommand)
}