client.go

package client

import (
	"flag"
	"fmt"
	"log"
	"time"

	pb "github.com/manybooks/deleuze/pb"
	"golang.org/x/net/context"
	"google.golang.org/grpc"
)

var (
	serverAddr = flag.String("server_addr", "127.0.0.1:10000", "The server address in the format of host:port")
)

func Reveal(question string) pb.Answer {
	conn := createConnection()
	defer conn.Close()
	client := pb.NewOracleClient(conn)
	ctx := context.Background()
	q := pb.Question{Question: question, UserId: 100}
	ans, err := client.Reveal(ctx, &q)
	if err != nil {
		log.Fatalf("Fail to get answer. ErrorMessage: %v\n", err)
	}
	return *ans
}

func createConnection() *grpc.ClientConn {
	opts := grpc.WithInsecure()
	conn, err := grpc.Dial("localhost:10000", opts)
	if err != nil {
		log.Fatalf("fail to dail: %v", err)
	}
	return conn
}

func main() {
	flag.Parse()
	opts := grpc.WithInsecure()
	conn, err := grpc.Dial(*serverAddr, opts)
	if err != nil {
		log.Fatalf("fail to dail: %v", err)
	}
	defer conn.Close()
	client := pb.NewOracleClient(conn)
	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
	q := pb.Question{Question: "what time is it now?", UserId: 100}
	ans, err := client.Reveal(ctx, &q)
	if err != nil {
		log.Fatalf("fail to get answer")
	}
	fmt.Println(ans.Answer)
}