Changed product catalog to use json, so that it slows down and generates nice profiles

This commit is contained in:
morganmclean 2018-07-17 21:35:11 -07:00
parent 2713f47bbb
commit dbc9523cea

View file

@ -10,7 +10,9 @@ import (
"time"
pb "./genproto"
"cloud.google.com/go/profiler"
"github.com/golang/protobuf/jsonpb"
"go.opencensus.io/exporter/stackdriver"
"go.opencensus.io/plugin/ocgrpc"
"go.opencensus.io/trace"
@ -21,71 +23,107 @@ import (
var port = flag.Int("port", 3550, "port to listen at")
var catalog = []*pb.Product{
const catalogJSON = `{
"products": [
{
Id: "OLJCESPC7Z",
Name: "Vintage Typewriter",
Description: "This typewriter looks good in your living room.",
Picture: "/static/img/products/typewriter.jpg",
PriceUsd: &pb.Money{CurrencyCode: "USD", Units: 67, Nanos: 990000000},
},
{
Id: "66VCHSJNUP",
Name: "Vintage Camera Lens",
Description: "You won't have a camera to use it and it probably doesn't work anyway.",
Picture: "/static/img/products/camera-lens.jpg",
PriceUsd: &pb.Money{CurrencyCode: "USD", Units: 12, Nanos: 490000000},
},
{
Id: "1YMWWN1N4O",
Name: "Home Barista Kit",
Description: "Always wanted to brew coffee with Chemex and Aeropress at home?",
Picture: "/static/img/products/barista-kit.jpg",
PriceUsd: &pb.Money{CurrencyCode: "USD", Units: 124, Nanos: 0},
},
{
Id: "L9ECAV7KIM",
Name: "Terrarium",
Description: "This terrarium will looks great in your white painted living room.",
Picture: "/static/img/products/terrarium.jpg",
PriceUsd: &pb.Money{CurrencyCode: "USD", Units: 36, Nanos: 450000000},
},
{
Id: "2ZYFJ3GM2N",
Name: "Film Camera",
Description: "This camera looks like it's a film camera, but it's actually digital.",
Picture: "/static/img/products/film-camera.jpg",
PriceUsd: &pb.Money{CurrencyCode: "USD", Units: 2245, Nanos: 00000000},
},
{
Id: "0PUK6V6EV0",
Name: "Vintage Record Player",
Description: "It still works.",
Picture: "/static/img/products/record-player.jpg",
PriceUsd: &pb.Money{CurrencyCode: "USD", Units: 65, Nanos: 500000000},
},
{
Id: "LS4PSXUNUM",
Name: "Metal Camping Mug",
Description: "You probably don't go camping that often but this is better than plastic cups.",
Picture: "/static/img/products/camp-mug.jpg",
PriceUsd: &pb.Money{CurrencyCode: "USD", Units: 24, Nanos: 330000000},
},
{
Id: "9SIQT8TOJO",
Name: "City Bike",
Description: "This single gear bike probably cannot climb the hills of San Francisco.",
Picture: "/static/img/products/city-bike.jpg",
PriceUsd: &pb.Money{CurrencyCode: "USD", Units: 789, Nanos: 500000000},
},
{
Id: "6E92ZMYYFZ",
Name: "Air Plant",
Description: "Have you ever wondered whether air plants need water? Buy one and figure out.",
Picture: "/static/img/products/air-plant.jpg",
PriceUsd: &pb.Money{CurrencyCode: "USD", Units: 12, Nanos: 300000000},
},
"id": "OLJCESPC7Z",
"name": "Vintage Typewriter",
"description": "This typewriter looks good in your living room.",
"picture": "/static/img/products/typewriter.jpg",
"priceUsd": {
"currencyCode": "USD",
"units": 67,
"nanos": 990000000
}
},
{
"id": "66VCHSJNUP",
"name": "Vintage Camera Lens",
"description": "You won't have a camera to use it and it probably doesn't work anyway.",
"picture": "/static/img/products/camera-lens.jpg",
"priceUsd": {
"currencyCode": "USD",
"units": 12,
"nanos": 490000000
}
},
{
"id": "1YMWWN1N4O",
"name": "Home Barista Kit",
"description": "Always wanted to brew coffee with Chemex and Aeropress at home?",
"picture": "/static/img/products/barista-kit.jpg",
"priceUsd": {
"currencyCode": "USD",
"units": 124
}
},
{
"id": "L9ECAV7KIM",
"name": "Terrarium",
"description": "This terrarium will looks great in your white painted living room.",
"picture": "/static/img/products/terrarium.jpg",
"priceUsd": {
"currencyCode": "USD",
"units": 36,
"nanos": 450000000
}
},
{
"id": "2ZYFJ3GM2N",
"name": "Film Camera",
"description": "This camera looks like it's a film camera, but it's actually digital.",
"picture": "/static/img/products/film-camera.jpg",
"priceUsd": {
"currencyCode": "USD",
"units": 2245
}
},
{
"id": "0PUK6V6EV0",
"name": "Vintage Record Player",
"description": "It still works.",
"picture": "/static/img/products/record-player.jpg",
"priceUsd": {
"currencyCode": "USD",
"units": 65,
"nanos": 500000000
}
},
{
"id": "LS4PSXUNUM",
"name": "Metal Camping Mug",
"description": "You probably don't go camping that often but this is better than plastic cups.",
"picture": "/static/img/products/camp-mug.jpg",
"priceUsd": {
"currencyCode": "USD",
"units": 24,
"nanos": 330000000
}
},
{
"id": "9SIQT8TOJO",
"name": "City Bike",
"description": "This single gear bike probably cannot climb the hills of San Francisco.",
"picture": "/static/img/products/city-bike.jpg",
"priceUsd": {
"currencyCode": "USD",
"units": 789,
"nanos": 500000000
}
},
{
"id": "6E92ZMYYFZ",
"name": "Air Plant",
"description": "Have you ever wondered whether air plants need water? Buy one and figure out.",
"picture": "/static/img/products/air-plant.jpg",
"priceUsd": {
"currencyCode": "USD",
"units": 12,
"nanos": 300000000
}
}
]
}`
func main() {
go initTracing()
@ -136,36 +174,51 @@ func initProfiling(service, version string) {
Service: service,
ServiceVersion: version,
// ProjectID must be set if not running on GCP.
// ProjectID: "my-project",
ProjectID: "oval-time-515",
}); 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)
continue
}
return
}
log.Printf("warning: could not initialize stackdriver profiler after retrying, giving up")
}
type productCatalog struct{}
func (p *productCatalog) catalog() []*pb.Product {
var cat pb.ListProductsResponse
if err := jsonpb.UnmarshalString(catalogJSON, &cat); err != nil {
log.Printf("warning: failed to parse the product catalog: %v", err)
return nil
}
return cat.Products
}
func (p *productCatalog) ListProducts(context.Context, *pb.Empty) (*pb.ListProductsResponse, error) {
return &pb.ListProductsResponse{Products: catalog}, nil
return &pb.ListProductsResponse{Products: p.catalog()}, nil
}
func (p *productCatalog) GetProduct(ctx context.Context, req *pb.GetProductRequest) (*pb.Product, error) {
for _, p := range catalog {
if req.Id == p.Id {
return p, nil
var found *pb.Product
for i := 0; i < len(p.catalog()); i++ {
if req.Id == p.catalog()[i].Id {
found = p.catalog()[i]
}
}
if found == nil {
return nil, status.Errorf(codes.NotFound, "no product with ID %s", req.Id)
}
return found, nil
}
func (p *productCatalog) SearchProducts(ctx context.Context, req *pb.SearchProductsRequest) (*pb.SearchProductsResponse, error) {
// Intepret query as a substring match in name or description.
var ps []*pb.Product
for _, p := range catalog {
for _, p := range p.catalog() {
if strings.Contains(strings.ToLower(p.Name), strings.ToLower(req.Query)) ||
strings.Contains(strings.ToLower(p.Description), strings.ToLower(req.Query)) {
ps = append(ps, p)