diff --git a/client/client.go b/client/client.go
index fcb5d8de2f6c2d88fec581d868550c0c84e2d684..e6a13030f1415256128f3ee18705a700a52f44b2 100644 (file)
--- a/client/client.go
+++ b/client/client.go
//
-// Copyright (C) 2014 Sebastian 'tokkee' Harl <sh@tokkee.org>
+// Copyright (C) 2014-2015 Sebastian 'tokkee' Harl <sh@tokkee.org>
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
}
defer c.Close()
+Then, it can issue requests to the server:
+
+ major, minor, patch, extra, err := c.ServerVersion()
+ if err != nil {
+ // handle error
+ }
+ fmt.Printf("Connected to SysDB %d.%d.%d%s\n", major, minor, patch, extra)
+
+or:
+
+ res, err := c.Call(&proto.Message{Type: proto.ConnectionServerVersion})
+ if err != nil {
+ // handle error
+ }
+ fmt.Printf("%v\n", res)
+
+or execute queries:
+
+ q, err := client.QueryString("FETCH %s %s", client.Identifier(typ), name)
+ if err != nil {
+ // handle error
+ }
+ res, err := c.Query(q)
+ if err != nil {
+ // handle error
+ }
+
+ // res is one of the object types defined in the sysdb package.
+ switch typ {
+ case "host":
+ host := res.(*sysdb.Host)
+ // do something
+ // ...
+ }
+
+Each client maintains multiple connections to a SysDB server allowing to
+handle multiple requests in parallel. The SysDB server is able to handle that
+easily making it a cheap approach. The low-level Dial function creates a
+single connection to a SysDB server allowing to perform low-level operations:
+
+ conn, err := client.Dial("unix:/var/run/sysdbd.sock", "username")
+ if err != nil {
+ // handle error
+ }
+ defer conn.Close()
+
The github.com/sysdb/go/proto package provides support for handling requests
and responses. Use the Send and Receive functions to communicate with the
server:
Type: proto.ConnectionQuery,
Raw: []byte{"LOOKUP hosts MATCHING attribute.architecture = 'amd64';"},
}
- if err := c.Send(m); err != nil {
+ if err := conn.Send(m); err != nil {
// handle error
}
- m, err := c.Receive()
+ m, err := conn.Receive()
if err != nil {
// handle error
}
package client
import (
+ "encoding/binary"
"fmt"
- "net"
- "strings"
+ "log"
+ "runtime"
"github.com/sysdb/go/proto"
)
-// A Conn is a connection to a SysDB server instance.
+// A Client is a client for SysDB.
//
-// Multiple goroutines may invoke methods on a Conn simultaneously but since
-// the SysDB protocol requires a strict ordering of request and response
-// messages, the communication with the server will usually happen
-// sequentially.
-type Conn struct {
- c net.Conn
- network, addr, user string
+// A client may be used from multiple goroutines in parallel.
+type Client struct {
+ conns chan *Conn
}
-func (c *Conn) connect() (err error) {
- if c.c, err = net.Dial(c.network, c.addr); err != nil {
- return err
- }
- defer func() {
+// Connect creates a new client connected to a SysDB server instance at the
+// specified address using the specified user.
+//
+// The address may be a IP address or a UNIX domain socket, either prefixed
+// with 'unix:' or specifying an absolute file-system path.
+func Connect(addr, user string) (*Client, error) {
+ c := &Client{conns: make(chan *Conn, 2*runtime.NumCPU())}
+
+ for i := 0; i < cap(c.conns); i++ {
+ conn, err := Dial(addr, user)
if err != nil {
- c.Close()
+ return nil, err
}
- }()
-
- m := &proto.Message{
- Type: proto.ConnectionStartup,
- Raw: []byte(c.user),
- }
- if err := c.Send(m); err != nil {
- return err
- }
-
- m, err = c.Receive()
- if err != nil {
- return err
+ c.conns <- conn
}
- if m.Type == proto.ConnectionError {
- return fmt.Errorf("failed to startup session: %s", string(m.Raw))
- }
- if m.Type != proto.ConnectionOK {
- return fmt.Errorf("failed to startup session: unsupported")
- }
- return nil
+ return c, nil
}
-// Connect sets up a client connection to a SysDB server instance at the
-// specified address using the specified user.
+// Close closes a client connection. It may not be further used after calling
+// this function.
//
-// The address may be a UNIX domain socket, either prefixed with 'unix:' or
-// specifying an absolute file-system path.
-func Connect(addr, user string) (*Conn, error) {
- network := "tcp"
- if strings.HasPrefix(addr, "unix:") {
- network = "unix"
- addr = addr[len("unix:"):]
- } else if addr[0] == '/' {
- network = "unix"
- }
-
- c := &Conn{network: network, addr: addr, user: user}
- if err := c.connect(); err != nil {
- return nil, err
- }
- return c, nil
+// The function waits for all pending operations to finish.
+func (c *Client) Close() {
+ for i := 0; i < cap(c.conns); i++ {
+ conn := <-c.conns
+ conn.Close()
+ }
+ close(c.conns)
+ c.conns = nil
}
-// Close closes the client connection.
-//
-// Any blocked Send or Receive operations will be unblocked and return errors.
-func (c *Conn) Close() {
- if c.c == nil {
- return
+// Call sends the specified request to the server and waits for its reply. It
+// blocks until the full reply has been received.
+func (c *Client) Call(req *proto.Message) (*proto.Message, error) {
+ conn := <-c.conns
+ defer func() { c.conns <- conn }()
+
+ err := conn.Send(req)
+ if err != nil {
+ return nil, err
}
- c.c.Close()
- c.c = nil
-}
-// Send sends the specified raw message to the server.
-//
-// Send operations block until the full message could be written to the
-// underlying sockets. This ensures that server and client don't get out of
-// sync.
-func (c *Conn) Send(m *proto.Message) error {
- var err error
- if c.c != nil {
- err = proto.Write(c.c, m)
- if err == nil {
- return nil
+ for {
+ res, err := conn.Receive()
+ switch {
+ case err != nil:
+ return nil, err
+ case res.Type == proto.ConnectionError:
+ return nil, fmt.Errorf("request failed: %s", string(res.Raw))
+ case res.Type != proto.ConnectionLog:
+ return res, err
}
- c.Close()
- }
- // Try to reconnect.
- if e := c.connect(); e == nil {
- return proto.Write(c.c, m)
- } else if err == nil {
- err = e
+ if len(res.Raw) > 4 {
+ log.Println(string(res.Raw[4:]))
+ }
}
- return err
+
+ // Not reached; needed for Go1 compatibility.
+ return nil, nil
}
-// Receive waits for a reply from the server and returns the raw message.
-//
-// Receive operations block until a full message could be read from the
-// underlying socket. This ensures that server and client don't get out of
-// sync.
-func (c *Conn) Receive() (*proto.Message, error) {
- var err error
- if c.c != nil {
- var m *proto.Message
- m, err = proto.Read(c.c)
+// ServerVersion queries and returns the version of the remote server.
+func (c *Client) ServerVersion() (major, minor, patch int, extra string, err error) {
+ res, err := c.Call(&proto.Message{Type: proto.ConnectionServerVersion})
+ if err != nil || res.Type != proto.ConnectionOK {
if err == nil {
- return m, err
+ err = fmt.Errorf("SERVER_VERSION command failed with status %d", res.Type)
}
- c.Close()
+ return 0, 0, 0, "", err
}
-
- // Try to reconnect.
- if e := c.connect(); e == nil {
- return proto.Read(c.c)
- } else if err == nil {
- err = e
+ if len(res.Raw) < 4 {
+ return 0, 0, 0, "", fmt.Errorf("SERVER_VERSION reply is too short")
+ }
+ version := int(binary.BigEndian.Uint32(res.Raw[:4]))
+ major = version / 10000
+ minor = version/100 - 100*major
+ patch = version - 10000*major - 100*minor
+ if len(res.Raw) > 4 {
+ extra = string(res.Raw[4:])
}
- return nil, err
+ return major, minor, patch, extra, nil
}
// vim: set tw=78 sw=4 sw=4 noexpandtab :