Extend service with GetConsignment method

sleeping-barber · Sep 6, 2018 · 536308e · 536308e
1 parent 5368c36
commit 536308e
Show file tree

Hide file tree

Showing 3 changed files with 119 additions and 33 deletions.
diff --git a/consignment-service/main.go b/consignment-service/main.go
@@ -17,6 +17,7 @@ const (
 
 type IRepository interface {
 	Create(*pb.Consignment) (*pb.Consignment, error)
+	GetAll() []*pb.Consignment
 }
 
 // Dummy repository implementation
@@ -30,6 +31,10 @@ func (repo *Repository) Create(consignment *pb.Consignment) (*pb.Consignment, er
 	return consignment, nil
 }
 
+func (repo *Repository) GetAll() []*pb.Consignment {
+	return repo.consignement
+}
+
 // Service satisfy all methods from the proto interface
 type Service struct {
 	repo IRepository
@@ -45,6 +50,12 @@ func (s *Service) CreateConsignment(ctx context.Context, request *pb.Consignment
 	return &pb.Response{Created: true, Consignment: consignment}, nil
 }
 
+func (s *Service) GetConsignment(ctx context.Context) (*pb.Response, error) {
+	consignments := s.repo.GetAll()
+
+	return &pb.Response{Consignments: consignments}, nil
+}
+
 func main() {
 
 	repo := &Repository{}

diff --git a/consignment-service/proto/consignment/consignment.pb.go b/consignment-service/proto/consignment/consignment.pb.go
diff --git a/consignment-service/proto/consignment/consignment.proto b/consignment-service/proto/consignment/consignment.proto
@@ -5,7 +5,7 @@ package go.micro.srv.consignment;
 
 service ShippingService {
     rpc CreateConsignment (Consignment) returns (Response) {}
-    rpc GetConsignment (GetRequest) returns (Response)
+    rpc GetConsignment (GetRequest) returns (Response) {}
 }
 
 message Consignment {