tile38/internal/server/readonly.go

45 lines
849 B
Go
Raw Normal View History

package server
2016-03-04 16:08:16 -07:00
import (
"strings"
2016-03-29 05:53:53 -07:00
"time"
2016-03-04 16:08:16 -07:00
2017-10-05 08:20:40 -07:00
"github.com/tidwall/resp"
"github.com/tidwall/tile38/internal/log"
2016-03-04 16:08:16 -07:00
)
func (c *Server) cmdReadOnly(msg *Message) (res resp.Value, err error) {
2016-03-29 05:53:53 -07:00
start := time.Now()
vs := msg.Args[1:]
2016-03-04 16:08:16 -07:00
var arg string
2016-03-29 05:53:53 -07:00
var ok bool
2017-10-05 08:20:40 -07:00
2016-03-29 05:53:53 -07:00
if vs, arg, ok = tokenval(vs); !ok || arg == "" {
return NOMessage, errInvalidNumberOfArguments
2016-03-04 16:08:16 -07:00
}
2016-03-29 05:53:53 -07:00
if len(vs) != 0 {
return NOMessage, errInvalidNumberOfArguments
2016-03-04 16:08:16 -07:00
}
2016-03-29 05:53:53 -07:00
update := false
2016-03-04 16:08:16 -07:00
switch strings.ToLower(arg) {
default:
return NOMessage, errInvalidArgument(arg)
2016-03-04 16:08:16 -07:00
case "yes":
2017-09-29 18:11:05 -07:00
if !c.config.readOnly() {
2016-03-29 05:53:53 -07:00
update = true
2017-09-29 18:11:05 -07:00
c.config.setReadOnly(true)
2016-03-29 05:53:53 -07:00
log.Info("read only")
2016-03-04 16:08:16 -07:00
}
case "no":
2017-09-29 18:11:05 -07:00
if c.config.readOnly() {
2016-03-29 05:53:53 -07:00
update = true
2017-09-29 18:11:05 -07:00
c.config.setReadOnly(false)
2016-03-29 05:53:53 -07:00
log.Info("read write")
2016-03-04 16:08:16 -07:00
}
}
2016-03-29 05:53:53 -07:00
if update {
2017-09-29 18:11:05 -07:00
c.config.write(false)
2016-03-04 16:08:16 -07:00
}
return OKMessage(msg, start), nil
2016-03-04 16:08:16 -07:00
}