ledisdb/ledis/command.go

package ledis

import (
	"fmt"
	"strings"
)

type CommandFunc func(c *client) error

var regCmds = map[string]CommandFunc{}

func register(name string, f CommandFunc) {
	if _, ok := regCmds[strings.ToLower(name)]; ok {
		panic(fmt.Sprintf("%s has been registered", name))
	}

	regCmds[name] = f
}

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

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

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

func init() {
	register("ping", pingCommand)
	register("echo", echoCommand)
}