138 lines
3.9 KiB
Go
138 lines
3.9 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"time"
|
|
|
|
"cloud.google.com/go/profiler"
|
|
"go.opencensus.io/exporter/stackdriver"
|
|
"go.opencensus.io/plugin/ocgrpc"
|
|
"go.opencensus.io/trace"
|
|
"golang.org/x/net/context"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/reflection"
|
|
|
|
pb "./genproto"
|
|
)
|
|
|
|
const (
|
|
defaultPort = "50051"
|
|
)
|
|
|
|
func main() {
|
|
go initTracing()
|
|
go initProfiling("shippingservice", "1.0.0")
|
|
|
|
port := defaultPort
|
|
if value, ok := os.LookupEnv("APP_PORT"); ok {
|
|
port = value
|
|
}
|
|
port = fmt.Sprintf(":%s", port)
|
|
|
|
if err := profiler.Start(profiler.Config{
|
|
Service: "shippingservice",
|
|
ServiceVersion: "1.0.0",
|
|
// ProjectID must be set if not running on GCP.
|
|
// ProjectID: "my-project",
|
|
}); err != nil {
|
|
log.Fatalf("failed to start profiler: %+v", err)
|
|
}
|
|
|
|
lis, err := net.Listen("tcp", port)
|
|
if err != nil {
|
|
log.Fatalf("failed to listen: %v", err)
|
|
}
|
|
s := grpc.NewServer(grpc.StatsHandler(&ocgrpc.ServerHandler{}))
|
|
pb.RegisterShippingServiceServer(s, &server{})
|
|
log.Printf("Shipping Service listening on port %s", port)
|
|
|
|
// Register reflection service on gRPC server.
|
|
reflection.Register(s)
|
|
if err := s.Serve(lis); err != nil {
|
|
log.Fatalf("failed to serve: %v", err)
|
|
}
|
|
}
|
|
|
|
// server controls RPC service responses.
|
|
type server struct{}
|
|
|
|
// GetQuote produces a shipping quote (cost) in USD.
|
|
func (s *server) GetQuote(ctx context.Context, in *pb.GetQuoteRequest) (*pb.GetQuoteResponse, error) {
|
|
log.Printf("[GetQuote] received request")
|
|
defer log.Printf("[GetQuote] completed request")
|
|
|
|
// 1. Our quote system requires the total number of items to be shipped.
|
|
count := 0
|
|
for _, item := range in.Items {
|
|
count += int(item.Quantity)
|
|
}
|
|
|
|
// 2. Generate a quote based on the total number of items to be shipped.
|
|
quote := CreateQuoteFromCount(count)
|
|
|
|
// 3. Generate a response.
|
|
return &pb.GetQuoteResponse{
|
|
CostUsd: &pb.Money{
|
|
CurrencyCode: "USD",
|
|
Units: int64(quote.Dollars),
|
|
Nanos: int32(quote.Cents * 10000000)},
|
|
}, nil
|
|
|
|
}
|
|
|
|
// ShipOrder mocks that the requested items will be shipped.
|
|
// It supplies a tracking ID for notional lookup of shipment delivery status.
|
|
func (s *server) ShipOrder(ctx context.Context, in *pb.ShipOrderRequest) (*pb.ShipOrderResponse, error) {
|
|
log.Printf("[ShipOrder] received request")
|
|
defer log.Printf("[ShipOrder] completed request")
|
|
// 1. Create a Tracking ID
|
|
baseAddress := fmt.Sprintf("%s, %s, %s", in.Address.StreetAddress, in.Address.City, in.Address.State)
|
|
id := CreateTrackingId(baseAddress)
|
|
|
|
// 2. Generate a response.
|
|
return &pb.ShipOrderResponse{
|
|
TrackingId: id,
|
|
}, nil
|
|
}
|
|
|
|
func initTracing() {
|
|
// TODO(ahmetb) this method is duplicated in other microservices using Go
|
|
// since they are not sharing packages.
|
|
for i := 1; i <= 3; i++ {
|
|
exporter, err := stackdriver.NewExporter(stackdriver.Options{})
|
|
if err != nil {
|
|
log.Printf("info: failed to initialize stackdriver exporter: %+v", err)
|
|
} else {
|
|
trace.RegisterExporter(exporter)
|
|
trace.ApplyConfig(trace.Config{DefaultSampler: trace.AlwaysSample()})
|
|
log.Print("registered stackdriver tracing")
|
|
return
|
|
}
|
|
d := time.Second * 10 * time.Duration(i)
|
|
log.Printf("sleeping %v to retry initializing stackdriver exporter", d)
|
|
time.Sleep(d)
|
|
}
|
|
log.Printf("warning: could not initialize stackdriver exporter after retrying, giving up")
|
|
}
|
|
|
|
func initProfiling(service, version string) {
|
|
// TODO(ahmetb) this method is duplicated in other microservices using Go
|
|
// since they are not sharing packages.
|
|
for i := 1; i <= 3; i++ {
|
|
if err := profiler.Start(profiler.Config{
|
|
Service: service,
|
|
ServiceVersion: version,
|
|
// ProjectID must be set if not running on GCP.
|
|
// ProjectID: "my-project",
|
|
}); err != nil {
|
|
log.Printf("warn: failed to start profiler: %+v", err)
|
|
}
|
|
d := time.Second * 10 * time.Duration(i)
|
|
log.Printf("sleeping %v to retry initializing stackdriver profiler", d)
|
|
time.Sleep(d)
|
|
}
|
|
log.Printf("warning: could not initialize stackdriver profiler after retrying, giving up")
|
|
}
|