110 lines
3.3 KiB
Go
110 lines
3.3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/pkg/errors"
|
|
"github.com/sirupsen/logrus"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
const (
|
|
port = "8080"
|
|
defaultCurrency = "USD"
|
|
cookieMaxAge = 60 * 60 * 48
|
|
|
|
cookiePrefix = "shop_"
|
|
cookieSessionID = cookiePrefix + "session-id"
|
|
cookieCurrency = cookiePrefix + "currency"
|
|
)
|
|
|
|
var (
|
|
whitelistedCurrencies = map[string]bool{
|
|
"USD": true, "EUR": true, "CAD": true, "JPY": true}
|
|
)
|
|
|
|
type ctxKeySessionID struct{}
|
|
|
|
type frontendServer struct {
|
|
productCatalogSvcAddr string
|
|
productCatalogSvcConn *grpc.ClientConn
|
|
|
|
currencySvcAddr string
|
|
currencySvcConn *grpc.ClientConn
|
|
|
|
cartSvcAddr string
|
|
cartSvcConn *grpc.ClientConn
|
|
|
|
recommendationSvcAddr string
|
|
recommendationSvcConn *grpc.ClientConn
|
|
|
|
checkoutSvcAddr string
|
|
checkoutSvcConn *grpc.ClientConn
|
|
|
|
shippingSvcAddr string
|
|
shippingSvcConn *grpc.ClientConn
|
|
}
|
|
|
|
func main() {
|
|
ctx := context.Background()
|
|
log := logrus.New()
|
|
log.Level = logrus.DebugLevel
|
|
log.Formatter = &logrus.TextFormatter{}
|
|
|
|
srvPort := port
|
|
if os.Getenv("PORT") != "" {
|
|
srvPort = os.Getenv("PORT")
|
|
}
|
|
addr := os.Getenv("LISTEN_ADDR")
|
|
svc := new(frontendServer)
|
|
mustMapEnv(&svc.productCatalogSvcAddr, "PRODUCT_CATALOG_SERVICE_ADDR")
|
|
mustMapEnv(&svc.currencySvcAddr, "CURRENCY_SERVICE_ADDR")
|
|
mustMapEnv(&svc.cartSvcAddr, "CART_SERVICE_ADDR")
|
|
mustMapEnv(&svc.recommendationSvcAddr, "RECOMMENDATION_SERVICE_ADDR")
|
|
mustMapEnv(&svc.checkoutSvcAddr, "CHECKOUT_SERVICE_ADDR")
|
|
mustMapEnv(&svc.shippingSvcAddr, "SHIPPING_SERVICE_ADDR")
|
|
|
|
mustConnGRPC(ctx, &svc.currencySvcConn, svc.currencySvcAddr)
|
|
mustConnGRPC(ctx, &svc.productCatalogSvcConn, svc.productCatalogSvcAddr)
|
|
mustConnGRPC(ctx, &svc.cartSvcConn, svc.cartSvcAddr)
|
|
mustConnGRPC(ctx, &svc.recommendationSvcConn, svc.recommendationSvcAddr)
|
|
mustConnGRPC(ctx, &svc.shippingSvcConn, svc.shippingSvcAddr)
|
|
mustConnGRPC(ctx, &svc.checkoutSvcConn, svc.checkoutSvcAddr)
|
|
|
|
r := mux.NewRouter()
|
|
r.HandleFunc("/", svc.homeHandler).Methods(http.MethodGet, http.MethodHead)
|
|
r.HandleFunc("/product/{id}", svc.productHandler).Methods(http.MethodGet, http.MethodHead)
|
|
r.HandleFunc("/cart", svc.viewCartHandler).Methods(http.MethodGet, http.MethodHead)
|
|
r.HandleFunc("/cart", svc.addToCartHandler).Methods(http.MethodPost)
|
|
r.HandleFunc("/cart/empty", svc.emptyCartHandler).Methods(http.MethodPost)
|
|
r.HandleFunc("/setCurrency", svc.setCurrencyHandler).Methods(http.MethodPost)
|
|
r.HandleFunc("/logout", svc.logoutHandler).Methods(http.MethodGet)
|
|
r.HandleFunc("/cart/checkout", svc.placeOrderHandler).Methods(http.MethodPost)
|
|
r.PathPrefix("/static/").Handler(http.StripPrefix("/static/", http.FileServer(http.Dir("./static/"))))
|
|
|
|
log.Infof("starting server on " + addr + ":" + srvPort)
|
|
loggedHandler := &logHandler{log: log, next: r}
|
|
log.Fatal(http.ListenAndServe(addr+":"+srvPort,
|
|
http.HandlerFunc(ensureSessionID(loggedHandler))))
|
|
}
|
|
|
|
func mustMapEnv(target *string, envKey string) {
|
|
v := os.Getenv(envKey)
|
|
if v == "" {
|
|
panic(fmt.Sprintf("environment variable %q not set", envKey))
|
|
}
|
|
*target = v
|
|
}
|
|
|
|
func mustConnGRPC(ctx context.Context, conn **grpc.ClientConn, addr string) {
|
|
var err error
|
|
*conn, err = grpc.DialContext(ctx, addr, grpc.WithInsecure(), grpc.WithTimeout(time.Second*3))
|
|
if err != nil {
|
|
panic(errors.Wrapf(err, "grpc: failed to connect %s", addr))
|
|
}
|
|
}
|