func (t TextProt) Prepend(rw *bufio.ReadWriter, key []byte, value []byte) error {
if _, err := fmt.Fprintf(rw, "prepend %s 0 0 %v\r\n%s\r\n", string(key), len(value), string(value)); err != nil {
return err
}
rw.Flush()
_, err := rw.ReadString('\n')
return err
}
func (t TextProt) BatchGet(rw *bufio.ReadWriter, keys [][]byte) ([][]byte, error) {
if VERBOSE {
fmt.Printf("Getting keys %v\n", keys)
}
cmd := []byte("get")
space := byte(' ')
end := []byte("\r\n")
for _, key := range keys {
cmd = append(cmd, space)
cmd = append(cmd, key...)
}
cmd = append(cmd, end...)
if _, err := fmt.Fprint(rw, string(cmd)); err != nil {
return nil, err
}
rw.Flush()
var ret [][]byte
for {
// read the header line
response, err := rw.ReadString('\n')
if err != nil {
return nil, err
}
if VERBOSE {
fmt.Println(response)
}
if strings.TrimSpace(response) == "END" {
if VERBOSE {
fmt.Println("End of batch response")
}
return ret, nil
}
// then read the value
response, err = rw.ReadString('\n')
if err != nil {
return nil, err
}
if VERBOSE {
fmt.Println(response)
}
ret = append(ret, []byte(response))
}
}
/* read client data, if client closed,then close server conn
*
*/
func read(rwr *bufio.ReadWriter, userId int64) {
var p byte = '\n'
var err error
status := "Ok"
if _, err = rwr.ReadString(p); err != nil {
conn := us.conns[userId]
if closeErr := conn.Close(); closeErr != nil {
status = "Fail"
}
delete(us.conns, userId)
del(userId)
log.Debug("close%s:\t%d", status, userId)
}
}
// Run performs a single synchronous command on the GarageFeed.
func (g *GarageFeed) run(req garageReq, rw *bufio.ReadWriter) (result string, err error) {
if rw.Reader.Buffered() > 0 {
panic("should be no bytes pending")
}
// Write the command, flush the buffer.
_, err = rw.WriteString(fmt.Sprintf("%s\n", req.command))
if err != nil {
return "", err
}
rw.Flush()
// Now, wait for a single line respnse.
return rw.ReadString('\n')
}
func (t TextProt) Touch(rw *bufio.ReadWriter, key []byte) error {
strKey := string(key)
if VERBOSE {
fmt.Printf("Touching key %s\n", strKey)
}
if _, err := fmt.Fprintf(rw, "touch %s %v\r\n", strKey, common.Exp()); err != nil {
return err
}
rw.Flush()
response, err := rw.ReadString('\n')
if err != nil {
return err
}
if VERBOSE {
fmt.Println(response)
fmt.Printf("Touched key %s\n", strKey)
}
return nil
}
func (t TextProt) Delete(rw *bufio.ReadWriter, key []byte) error {
strKey := string(key)
if VERBOSE {
fmt.Printf("Deleting key %s\n", strKey)
}
if _, err := fmt.Fprintf(rw, "delete %s\r\n", strKey); err != nil {
return err
}
rw.Flush()
response, err := rw.ReadString('\n')
if err != nil {
return err
}
if VERBOSE {
fmt.Println(response)
fmt.Printf("Deleted key %s\r\n", strKey)
}
return nil
}
func (t TextProt) Replace(rw *bufio.ReadWriter, key []byte, value []byte) error {
strKey := string(key)
if VERBOSE {
fmt.Printf("Replacing key %s, value of length %v\n", strKey, len(value))
}
if _, err := fmt.Fprintf(rw, "replace %s 0 0 %v\r\n%s\r\n", strKey, len(value), string(value)); err != nil {
return err
}
rw.Flush()
response, err := rw.ReadString('\n')
if err != nil {
return err
}
if VERBOSE {
fmt.Println(response)
fmt.Printf("Replaced key %s\n", strKey)
}
return nil
}
func (t TextProt) Get(rw *bufio.ReadWriter, key []byte) ([]byte, error) {
strKey := string(key)
if VERBOSE {
fmt.Printf("Getting key %s\n", strKey)
}
if _, err := fmt.Fprintf(rw, "get %s\r\n", strKey); err != nil {
return nil, err
}
rw.Flush()
// read the header line
response, err := rw.ReadString('\n')
if err != nil {
return nil, err
}
if VERBOSE {
fmt.Println(response)
}
if strings.TrimSpace(response) == "END" {
if VERBOSE {
fmt.Println("Empty response / cache miss")
}
return []byte{}, nil
}
// then read the value
response, err = rw.ReadString('\n')
if err != nil {
return nil, err
}
if VERBOSE {
fmt.Println(response)
}
// then read the END
response, err = rw.ReadString('\n')
if err != nil {
return nil, err
}
if VERBOSE {
fmt.Println(response)
fmt.Printf("Got key %s\n", key)
}
return []byte(response), nil
}
