Initial commit

Server/services/user_server.go

@@ -0,0 +1,74 @@
+package services
+
+import (
+	"fmt"
+	"context"
+	"log"
+	"errors"
+	
+	"google.golang.org/grpc"
+	pb "tldemo/services/User"
+)
+
+type UserAddress struct {
+	Address string
+	PostalCode string
+}
+
+type UserData struct {
+	Id int
+	Name string
+	Gender pb.GENDER_ENUM
+	Addresses [] *UserAddress
+}
+
+type UserServer struct {
+	pb.UnimplementedUserServiceServer
+	UserArray []*UserData
+}
+
+func (s *UserServer) Register(ctx context.Context, in *pb.RegisterRequest) (*pb.RegisterResponse, error) {
+	userData := &UserData{Id:len(s.UserArray), Name:in.Name, Gender:in.Gender}
+	userData.Addresses = make([]*UserAddress, 0)
+	for i:=0; i<len(in.Address); i++ {
+		userData.Addresses = append(userData.Addresses, &UserAddress{Address : in.Address[i].Address, PostalCode: in.Address[i].PostalCode})
+	}
+	
+	s.UserArray = append(s.UserArray, userData)
+	
+	replyMessage := fmt.Sprintf("Hello %s, Your Id is %v",  userData.Name, userData.Id)
+	log.Printf("[Register User] %s", replyMessage)
+	result := &pb.RegisterResponse{Id: uint64(userData.Id), Msg: replyMessage}
+	return result, nil
+}
+
+func (s *UserServer) Query(ctx context.Context, in *pb.QueryRequest) (*pb.QueryResponse, error) {
+	log.Printf("[Query User] %d", in.Id)
+	
+	if(in.Id<0 || int(in.Id)>=len(s.UserArray)) {
+		return nil, errors.New("Out of range!")
+	}
+	
+	var userData = s.UserArray[int(in.Id)]
+	
+	address := make([]*pb.Address, 0)
+	for i:=0; i<len(userData.Addresses); i++ {
+		address = append(address, &pb.Address{ Address: userData.Addresses[i].Address, PostalCode: userData.Addresses[i].PostalCode})
+	}
+	//address[0] = &pb.Address{ Address: "xxxx", PostalCode: "1234"}
+	
+	result := &pb.QueryResponse{
+		Name: userData.Name,
+		Gender: userData.Gender,
+		Address : address,
+	}
+	return result, nil
+}
+
+func RegisterUserServer(s *grpc.Server) {
+	userServer := &UserServer{}
+	userServer.UserArray = make([]*UserData, 0)
+	
+	pb.RegisterUserServiceServer(s, userServer)
+}
+