Initial commit

Server/services/greeter_server.go

@@ -0,0 +1,109 @@
+package services
+
+import (
+	"fmt"
+	"context"
+	"log"
+	"time"
+
+	"google.golang.org/grpc"
+	pb "tldemo/services/Greeter"
+)
+
+type GreeterServer struct {
+	pb.UnimplementedGreeterServiceServer
+}
+
+func (s *GreeterServer) Hello(ctx context.Context, in *pb.HelloRequest) (*pb.HelloResponse, error) {
+	currentTime := time.Now()
+	replyMessage := fmt.Sprintf("Hello %s, time is %s", in.Name, currentTime.Format("15:04:05"))
+	log.Printf("[Hello User] %s", in.Name)
+	result := &pb.HelloResponse{ReplyMessage: replyMessage}
+	return result, nil
+}
+
+type TimeServer struct {
+	pb.UnimplementedTimeServiceServer
+}
+
+func (s *TimeServer) Now(ctx context.Context, in *pb.Nothing) (*pb.NowResponse, error) {
+	currentTime := uint64(time.Now().UTC().UnixMilli())
+	log.Printf("[Now]send time: %d", currentTime)
+	
+	result := &pb.NowResponse{Now: currentTime}
+	return result, nil
+}
+
+func (s *TimeServer) Ticktok(in *pb.TicktokRequest, stream pb.TimeService_TicktokServer) error {
+	ctx := stream.Context()
+	response := &pb.NowResponse{}
+
+	sendNow := func(counts int32) {
+		response.Now = uint64(time.Now().UTC().UnixMilli())
+		response.Counts = counts
+		log.Printf("[Ticktok]send time %d/%d", response.Now, counts)
+		if err := stream.Send(response); err != nil {
+			return
+		}
+	}
+	counts := int32(in.Counts)
+	if(counts < 0) {
+		counts=0
+	}
+	
+	for {
+		sendNow(counts)
+		
+		if(in.Counts>0) {
+			counts-=1
+			
+			if(counts <= 0) {
+				return nil
+			}
+		}
+		
+		select {
+		case <-ctx.Done():
+			return nil
+		default:
+			time.Sleep(1.0 * time.Second)
+		}
+	}
+	return nil
+}
+
+func (s *TimeServer) Watch(stream pb.TimeService_WatchServer) error {
+	ctx := stream.Context()
+	response := &pb.NowResponse{}
+
+	sendNow := func() error {
+		response.Now = uint64(time.Now().UTC().UnixMilli())
+		log.Printf("[Watch]send time %d", response.Now)
+		return stream.Send(response)
+	}
+
+	for {
+		select {
+		case <-ctx.Done():
+			return nil
+		default:
+		}
+
+		if _, err := stream.Recv(); err != nil {
+			return err
+		} else if err := sendNow(); err != nil {
+			return err
+		}
+	}
+}
+
+
+func RegisterGreeterServer(s *grpc.Server) {
+	greeterServer := &GreeterServer{}
+	pb.RegisterGreeterServiceServer(s, greeterServer)
+	
+	timeServer := &TimeServer{}
+	pb.RegisterTimeServiceServer(s, timeServer)
+}
+
+

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)
+}
+