81 lines
1.5 KiB
Go
81 lines
1.5 KiB
Go
package goSam
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
)
|
|
|
|
type Client struct {
|
|
SamConn net.Conn
|
|
verbose bool
|
|
}
|
|
|
|
// create a new client, connecting to the default host:port at localhost:7656
|
|
func NewDefaultClient() (*Client, error) {
|
|
return NewClient("localhost:7656")
|
|
}
|
|
|
|
// create a new client, connecting to a specified port
|
|
func NewClient(addr string) (*Client, error) {
|
|
conn, err := net.Dial("tcp", addr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
c := &Client{conn, false}
|
|
return c, c.hello()
|
|
}
|
|
|
|
func (c *Client) ToggleVerbose() {
|
|
c.verbose = !c.verbose
|
|
}
|
|
|
|
// send the initial handshake command and check that the reply is ok
|
|
func (c *Client) hello() (err error) {
|
|
var r *Reply
|
|
|
|
r, err = c.sendCmd("HELLO VERSION MIN=3.0 MAX=3.0")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if r.Topic != "HELLO" {
|
|
return fmt.Errorf("Unknown Reply: %+v\n", r)
|
|
}
|
|
|
|
if r.Pairs["RESULT"] != "OK" || r.Pairs["VERSION"] != "3.0" {
|
|
return fmt.Errorf("Handshake did not succeed\nReply:%+v\n", r)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// helper to send one command and parse the reply by sam
|
|
func (c *Client) sendCmd(cmd string) (r *Reply, err error) {
|
|
if _, err = fmt.Fprintln(c.SamConn, cmd); err != nil {
|
|
return
|
|
}
|
|
|
|
if c.verbose {
|
|
log.Printf(">Send>'%s'\n", cmd)
|
|
}
|
|
|
|
reader := bufio.NewReader(c.SamConn)
|
|
line, err := reader.ReadString('\n')
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
if c.verbose {
|
|
log.Printf("<Rcvd<'%s'\n", line)
|
|
}
|
|
|
|
r, err = parseReply(line)
|
|
return
|
|
}
|
|
|
|
func (c *Client) Close() error {
|
|
return c.SamConn.Close()
|
|
}
|