Implement client

client/args.go

@@ -0,0 +1,46 @@
+package main
+
+import (
+	"flag"
+)
+
+type Config struct {
+	port    int
+	host    string
+	dump    bool
+	restore bool
+}
+
+var config Config
+
+func init() {
+	const (
+		defaultPort = 9090
+		portDesc    = "The port to connect to the server"
+
+		defaultHost = "127.0.0.1"
+		hostDesc    = "The host to connect to the server"
+
+		defaultDump = false
+		dumpDesc    = "Dump the whole database to the JSON format on STDOUT"
+
+		defaultRestore = false
+		restoreDesc    = "Restore the database from the dumped file"
+	)
+
+	shorhand := " (shorthand)"
+
+	flag.IntVar(&config.port, "port", defaultPort, portDesc)
+	flag.IntVar(&config.port, "p", defaultPort, portDesc+shorhand)
+
+	flag.StringVar(&config.host, "host", defaultHost, hostDesc)
+	flag.StringVar(&config.host, "h", defaultHost, hostDesc+shorhand)
+
+	flag.BoolVar(&config.dump, "dump", defaultDump, dumpDesc)
+	flag.BoolVar(&config.dump, "d", defaultDump, dumpDesc+shorhand)
+
+	flag.BoolVar(&config.restore, "restore", defaultRestore, restoreDesc)
+	flag.BoolVar(&config.restore, "r", defaultRestore, restoreDesc+shorhand)
+
+	flag.Parse()
+}

client/main.go

@@ -0,0 +1,57 @@
+package main
+
+import (
+	"bufio"
+	"fmt"
+	"log"
+	"net"
+	"os"
+	"strconv"
+
+	_ "github.com/SiarheiKresik/go-kvdb/common"
+)
+
+func main() {
+
+	const protocol = "tcp"
+
+	addr := net.JoinHostPort(config.host, strconv.Itoa(config.port))
+	// TODO check if host and port are valid
+
+	log.Println("Connecting to server ", addr, "...")
+
+	conn, err := net.Dial(protocol, addr)
+	if err != nil {
+		log.Fatalln(err)
+	}
+	defer conn.Close()
+
+	remoteAddr := conn.RemoteAddr()
+	reader := bufio.NewReader(os.Stdin)
+
+	for {
+		// command line prompt
+		fmt.Print("server: ", remoteAddr, " > ")
+
+		// read in input from stdin
+		text, err := reader.ReadString('\n')
+		if err != nil {
+			log.Fatalln("text error:", err)
+		}
+
+		// send to socket
+		_, prnterr := fmt.Fprintf(conn, text)
+		if prnterr != nil {
+			log.Fatalln(prnterr)
+		}
+
+		// listen for reply
+		message, readerr := bufio.NewReader(conn).ReadString('\n')
+		if readerr != nil {
+			log.Fatalln(readerr)
+			break
+		}
+		fmt.Print(message)
+
+	}
+}