update
continuous-integration/drone/push Build is passing
Details
continuous-integration/drone/push Build is passing
Details
parent
2fb8077151
commit
a6e9c2db86
@ -0,0 +1,190 @@
|
||||
package ssdb
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"net"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
type Client struct {
|
||||
sock *net.TCPConn
|
||||
recv_buf bytes.Buffer
|
||||
}
|
||||
|
||||
func Connect(ip string, port int) (*Client, error) {
|
||||
addr, err := net.ResolveTCPAddr("tcp", fmt.Sprintf("%s:%d", ip, port))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
sock, err := net.DialTCP("tcp", nil, addr)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var c Client
|
||||
c.sock = sock
|
||||
return &c, nil
|
||||
}
|
||||
|
||||
func (c *Client) Do(args ...interface{}) ([]string, error) {
|
||||
err := c.send(args)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
resp, err := c.recv()
|
||||
return resp, err
|
||||
}
|
||||
|
||||
func (c *Client) Set(key string, val string) (interface{}, error) {
|
||||
resp, err := c.Do("set", key, val)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if len(resp) == 2 && resp[0] == "ok" {
|
||||
return true, nil
|
||||
}
|
||||
return nil, fmt.Errorf("bad response")
|
||||
}
|
||||
|
||||
// TODO: Will somebody write addition semantic methods?
|
||||
func (c *Client) Get(key string) (interface{}, error) {
|
||||
resp, err := c.Do("get", key)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if len(resp) == 2 && resp[0] == "ok" {
|
||||
return resp[1], nil
|
||||
}
|
||||
if resp[0] == "not_found" {
|
||||
return nil, nil
|
||||
}
|
||||
return nil, fmt.Errorf("bad response")
|
||||
}
|
||||
|
||||
func (c *Client) Del(key string) (interface{}, error) {
|
||||
resp, err := c.Do("del", key)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
//response looks like this: [ok 1]
|
||||
if len(resp) > 0 && resp[0] == "ok" {
|
||||
return true, nil
|
||||
}
|
||||
return nil, fmt.Errorf("bad response:resp:%v:", resp)
|
||||
}
|
||||
|
||||
func (c *Client) Send(args ...interface{}) error {
|
||||
return c.send(args)
|
||||
}
|
||||
|
||||
func (c *Client) send(args []interface{}) error {
|
||||
var buf bytes.Buffer
|
||||
for _, arg := range args {
|
||||
var s string
|
||||
switch arg := arg.(type) {
|
||||
case string:
|
||||
s = arg
|
||||
case []byte:
|
||||
s = string(arg)
|
||||
case []string:
|
||||
for _, s := range arg {
|
||||
buf.WriteString(fmt.Sprintf("%d", len(s)))
|
||||
buf.WriteByte('\n')
|
||||
buf.WriteString(s)
|
||||
buf.WriteByte('\n')
|
||||
}
|
||||
continue
|
||||
case int:
|
||||
s = fmt.Sprintf("%d", arg)
|
||||
case int64:
|
||||
s = fmt.Sprintf("%d", arg)
|
||||
case float64:
|
||||
s = fmt.Sprintf("%f", arg)
|
||||
case bool:
|
||||
if arg {
|
||||
s = "1"
|
||||
} else {
|
||||
s = "0"
|
||||
}
|
||||
case nil:
|
||||
s = ""
|
||||
default:
|
||||
return fmt.Errorf("bad arguments")
|
||||
}
|
||||
buf.WriteString(fmt.Sprintf("%d", len(s)))
|
||||
buf.WriteByte('\n')
|
||||
buf.WriteString(s)
|
||||
buf.WriteByte('\n')
|
||||
}
|
||||
buf.WriteByte('\n')
|
||||
_, err := c.sock.Write(buf.Bytes())
|
||||
return err
|
||||
}
|
||||
|
||||
func (c *Client) Recv() ([]string, error) {
|
||||
return c.recv()
|
||||
}
|
||||
|
||||
func (c *Client) recv() ([]string, error) {
|
||||
var tmp [8192]byte
|
||||
for {
|
||||
resp := c.parse()
|
||||
if resp == nil || len(resp) > 0 {
|
||||
return resp, nil
|
||||
}
|
||||
n, err := c.sock.Read(tmp[0:])
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
c.recv_buf.Write(tmp[0:n])
|
||||
}
|
||||
}
|
||||
|
||||
func (c *Client) parse() []string {
|
||||
resp := []string{}
|
||||
buf := c.recv_buf.Bytes()
|
||||
var idx, offset int
|
||||
idx = 0
|
||||
offset = 0
|
||||
|
||||
for {
|
||||
idx = bytes.IndexByte(buf[offset:], '\n')
|
||||
if idx == -1 {
|
||||
break
|
||||
}
|
||||
p := buf[offset : offset+idx]
|
||||
offset += idx + 1
|
||||
//fmt.Printf("> [%s]\n", p);
|
||||
if len(p) == 0 || (len(p) == 1 && p[0] == '\r') {
|
||||
if len(resp) == 0 {
|
||||
continue
|
||||
} else {
|
||||
var new_buf bytes.Buffer
|
||||
new_buf.Write(buf[offset:])
|
||||
c.recv_buf = new_buf
|
||||
return resp
|
||||
}
|
||||
}
|
||||
|
||||
size, err := strconv.Atoi(string(p))
|
||||
if err != nil || size < 0 {
|
||||
return nil
|
||||
}
|
||||
if offset+size >= c.recv_buf.Len() {
|
||||
break
|
||||
}
|
||||
|
||||
v := buf[offset : offset+size]
|
||||
resp = append(resp, string(v))
|
||||
offset += size + 1
|
||||
}
|
||||
|
||||
//fmt.Printf("buf.size: %d packet not ready...\n", len(buf))
|
||||
return []string{}
|
||||
}
|
||||
|
||||
// Close The Client Connection
|
||||
func (c *Client) Close() error {
|
||||
return c.sock.Close()
|
||||
}
|
||||
Loading…
Reference in new issue