Move client and CLI to the main repo

client/client.go

@@ -0,0 +1,163 @@
+package client
+
+import (
+	"net"
+	"bytes"
+	"encoding/json"
+	"io/ioutil"
+	"net/http"
+	"strconv"
+	"strings"
+)
+
+const (
+	publishEndpoint   = "/publish"
+	subscribeEndpoint = "/subscribe"
+	statusEndpoit     = "/status"
+	debugEndpoint     = "/debug"
+)
+
+type (
+	Client struct {
+		Config     *Config
+		httpClient *http.Client
+	}
+	Message struct {
+		Queue string
+		Body  []byte
+	}
+	QueueInfo struct {
+		Name        string
+		Messages    int
+		Subscribers int
+	}
+	DebugInfo struct {
+		Version      string                 `json:"version"`
+		Goroutines   int                    `json:"goroutines"`
+		KyotoCabinet map[string]interface{} `json:"kyoto_cabinet"`
+	}
+)
+
+func NewClient(c *Config) *Client {
+	return &Client{
+		Config:     c,
+		httpClient: &http.Client{},
+	}
+}
+
+func (c *Client) Publish(m *Message) (ok bool) {
+	rdr := bytes.NewReader(m.Body)
+	url := c.url(publishEndpoint, "?queue=", m.Queue)
+
+	res, err := http.Post(url, "text/plain", rdr)
+	if err != nil {
+		return
+	}
+
+	defer res.Body.Close()
+	body, err := ioutil.ReadAll(res.Body)
+	if err != nil {
+		return
+	}
+
+	ok = (string(body) == "OK")
+
+	return
+}
+
+func (c *Client) Subscribe(queues ...string) (m *Message) {
+	url := c.url(subscribeEndpoint, "?queues=", strings.Join(queues, ","))
+
+	transport := http.Transport{
+        Dial: net.DialTimeout(network, addr, timeout),
+    }
+
+	rdr := bytes.NewReader([]byte)
+	req := http.NewRequest("GET", url, rdr)
+	req.
+	res, err := http.Get(url)
+	if err != nil {
+		return
+	}
+
+	defer res.Body.Close()
+	body, err := ioutil.ReadAll(res.Body)
+	if err != nil {
+		return
+	}
+
+	m = &Message{
+		Queue: res.Header.Get("Queue"),
+		Body:  body,
+	}
+
+	return
+}
+
+func (c *Client) Status() (stat []*QueueInfo) {
+	url := c.url(statusEndpoit)
+	res, err := http.Get(url)
+	if err != nil {
+		return
+	}
+
+	defer res.Body.Close()
+	body, err := ioutil.ReadAll(res.Body)
+	if err != nil {
+		return
+	}
+
+	tmp := make(map[string]map[string]int)
+	if err := json.Unmarshal(body, &tmp); err != nil {
+		return
+	}
+
+	for queue, info := range tmp {
+		qi := &QueueInfo{
+			Name:        queue,
+			Messages:    info["messages"],
+			Subscribers: info["subscribers"],
+		}
+		stat = append(stat, qi)
+	}
+
+	return
+}
+
+func (c *Client) Debug() *DebugInfo {
+	url := c.url(debugEndpoint)
+	res, err := http.Get(url)
+	if err != nil {
+		return nil
+	}
+
+	defer res.Body.Close()
+	body, err := ioutil.ReadAll(res.Body)
+	if err != nil {
+		return nil
+	}
+
+	dbg := DebugInfo{}
+	if err := json.Unmarshal(body, &dbg); err != nil {
+		return nil
+	}
+
+	return &dbg
+}
+
+func (c *Client) url(path ...string) string {
+	parts := []string{"http://", c.Config.Host, ":", strconv.Itoa(c.Config.Port)}
+	parts = append(parts, path...)
+	return strings.Join(parts, "")
+}
+
+func TimeoutDialer(cTimeout time.Duration, rwTimeout time.Duration) func(net, addr string) (c net.Conn, err error) {
+    return func(netw, addr string) (net.Conn, error) {
+        conn, err := net.DialTimeout(netw, addr, cTimeout)
+        if err != nil {
+            return nil, err
+        }
+        conn.SetDeadline(time.Now().Add(rwTimeout))
+        return conn, nil
+    }
+}

client/config.go

@@ -0,0 +1,19 @@
+package client
+
+import (
+	"time"
+)
+
+type (
+	Config struct {
+		Host    string
+		Port    int
+		Timeout time.Duration
+	}
+)
+
+func (c *Config) UseDefaults() {
+	c.Host = "127.0.0.1"
+	c.Port = 4401
+	c.Timeout = 60 * time.Second
+}