/
main.go
69 lines (53 loc) · 1.49 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
package main
import (
"context"
"fmt"
"log"
"net"
"github.com/J-Thompson12/planet-express/crew"
"github.com/J-Thompson12/planet-express/delivery"
pb "github.com/J-Thompson12/planet-express/ship/pkg/planetexpress"
"google.golang.org/grpc"
"google.golang.org/protobuf/types/known/emptypb"
)
const port = 10000
type planetExpressShipServer struct {
pb.UnimplementedPlanetExpressServer
}
func newServer() *planetExpressShipServer {
s := &planetExpressShipServer{}
return s
}
func (s *planetExpressShipServer) GetShip(ctx context.Context, empty *emptypb.Empty) (*pb.GetShipResponse, error) {
crew, err := crew.GetCrew()
if err != nil {
return nil, err
}
return &pb.GetShipResponse{
Ship: &pb.Ship{
Name: "planet express ship",
Location: "omicron persei 8",
FuelLevel: pb.Ship_FULL,
Crew: crew.Crew,
},
}, nil
}
func (s *planetExpressShipServer) GetDelivery(ctx context.Context, deliveryRequest *pb.GetDeliveryRequest) (*pb.GetDeliveryResponse, error) {
delivered, err := delivery.GetDeliveryResponse(deliveryRequest)
if err != nil {
return nil, err
}
return delivered, nil
}
func main() {
log.Println("Planet Express Ship")
lis, err := net.Listen("tcp", fmt.Sprintf("localhost:%d", port))
if err != nil {
log.Fatalf("failed to listen: %v", err)
}
var opts []grpc.ServerOption
grpcServer := grpc.NewServer(opts...)
pb.RegisterPlanetExpressServer(grpcServer, newServer())
log.Printf("Ship listening on localhost:%d\n", port)
grpcServer.Serve(lis)
}