add grpc demo

Author: harleyzhao(赵政)

example_code/grpc_demo/client.go

@@ -0,0 +1,110 @@
+// client/main.go
+package main
+
+import (
+	"fmt"
+	"io"
+	"log"
+
+	"golang.org/x/net/context"
+	"google.golang.org/grpc"
+
+	pb "./customer"
+)
+
+const (
+	address = "localhost:50051"
+)
+
+// createCustomer calls the RPC method CreateCustomer of CustomerServer
+func createCustomer(client pb.CustomerClient, customer *pb.CustomerRequest) {
+	resp, err := client.CreateCustomer(context.Background(), customer)
+	if err != nil {
+		log.Fatalf("Could not create Customer: %v", err)
+	}
+	if resp.Success {
+		log.Printf("A new Customer has been added with id: %d", resp.Id)
+	}
+}
+
+// GetCustomers calls the RPC method GetCustomers of CustomerServer
+func getCustomers(client pb.CustomerClient, filter *pb.CustomerFilter) {
+	// calling the streaming API
+	stream, err := client.GetCustomers(context.Background(), filter)
+	if err != nil {
+		log.Fatal("Error on get customers: %v", err)
+	}
+	for {
+		customer, err := stream.Recv()
+		if err == io.EOF {
+			break
+		}
+
+		if err != nil {
+			log.Fatal("%v.GetCustomers(_) = _, %v", client, err)
+		}
+		log.Printf("Customer: %v", customer)
+	}
+}
+
+func main() {
+	// Set up a connection to the RPC server
+	conn, err := grpc.Dial(address, grpc.WithInsecure())
+	if err != nil {
+		log.Fatal("did not connect: %v", err)
+	}
+	defer conn.Close()
+	// creates a new CustomerClient
+	client := pb.NewCustomerClient(conn)
+
+	customer := &pb.CustomerRequest{
+		Id:    101,
+		Name:  "Shiju Varghese",
+		Email: "[email protected]",
+		Phone: "732-757-2923",
+		Addresses: []*pb.CustomerRequest_Address{
+			&pb.CustomerRequest_Address{
+				Street:            "1 Mission Street",
+				City:              "San Francisco",
+				State:             "CA",
+				Zip:               "94105",
+				IsShippingAddress: false,
+			},
+			&pb.CustomerRequest_Address{
+				Street:            "Greenfield",
+				City:              "Kochi",
+				State:             "KL",
+				Zip:               "68356",
+				IsShippingAddress: true,
+			},
+		},
+	}
+
+	// Create a new customer
+	createCustomer(client, customer)
+
+	customer = &pb.CustomerRequest{
+		Id:    102,
+		Name:  "Irene Rose",
+		Email: "[email protected]",
+		Phone: "732-757-2924",
+		Addresses: []*pb.CustomerRequest_Address{
+			&pb.CustomerRequest_Address{
+				Street:            "1 Mission Street",
+				City:              "San Francisco",
+				State:             "CA",
+				Zip:               "94105",
+				IsShippingAddress: true,
+			},
+		},
+	}
+
+	// Create a new customer
+	createCustomer(client, customer)
+	//Filter with an empty Keyword
+	filter := &pb.CustomerFilter{Keyword: ""}
+
+	fmt.Println("=====")
+	getCustomers(client, filter)
+
+}