func main() { addr := flag.String("listen", "0.0.0.0:9002", "Endpoint address") redisEndpoint := flag.String("redis", "127.0.0.1:6379", "Redis endpoint") bankEndpoint := flag.String("bank", "localhost:9001", "Endpoint for bank service") services.ParseBalanceEndpointsFlags() flag.Parse() services.ParseLoadBalanceConfig() services.EnableResponseLogging() golib.ConfigureOpenFilesLimit() bank := bankApi.NewHttpBank(*bankEndpoint) redisClient, err := services.ConnectRedis(*redisEndpoint) if err != nil { log.Fatalln(err) } if err := services.RegisterLockScripts(redisClient); err != nil { log.Fatalln("Failed to register redis scripts", err) } payments := &Payments{ bank: bank, redis: redisClient, accountName: "store", redisLockValue: services.EndpointLockValue(*addr), } mux := mux.NewRouter() mux.HandleFunc("/payment", payments.new_payment).Methods("POST").MatcherFunc(services.MatchFormKeys("user", "value", "ts")) mux.HandleFunc("/payment/{id}", payments.show_payment).Methods("GET") mux.HandleFunc("/payment/{id}/commit", payments.commit_payment).Methods("POST") mux.HandleFunc("/payment/{id}/cancel", payments.cancel_payment).Methods("POST") services.L.Warnf("Running on " + *addr) if err := http.ListenAndServe(*addr, mux); err != nil { log.Fatal(err) } }
func main() { num_users := flag.Uint64("users", 1000, "Number of users to check") verbose := flag.Bool("v", false, "Print processing details") bankEndpoint := flag.String("bank", "localhost:9001", "Bank endpoint") var shopEndpoint string flag.StringVar(&shopEndpoint, "shop", "localhost:9004", "Shop endpoint") flag.Parse() golib.ConfigureOpenFilesLimit() bank := bankApi.NewHttpBank(*bankEndpoint) inconsistent := false allItems, err := shopApi.AllItems(shopEndpoint) itemMap := make(map[string]*shopApi.Item) var totalEarned float64 var totalShipped uint64 for _, item := range allItems { itemMap[item.Name] = item total := item.Reserved + item.Shipped + item.Stock initial := initialItems[item.Name] expectedTotal := initial + item.Refills if total != expectedTotal { fmt.Printf("%v inconsistent: %v + %v + %v = %v, expected %v + %v = %v\n", item.Name, item.Reserved, item.Shipped, item.Stock, total, initial, item.Refills, expectedTotal) inconsistent = true } if item.Reserved != 0 { fmt.Printf("%v still has %v reserved items\n", item.Name, item.Reserved) inconsistent = true } totalShipped += item.Shipped totalEarned += float64(item.Shipped) * item.Cost } if !inconsistent { fmt.Println("All items are consistent!") } var cancelledOrders uint64 var processedOrders uint64 var totalEarnedOrders float64 var totalShippedOrders uint64 var totalProcessingOrders uint64 for i := uint64(0); i < *num_users; i++ { user := fmt.Sprintf("User%v", i) orders, err := shopApi.AllOrders(shopEndpoint, user) check(err) if *verbose { fmt.Printf("Checking %v orders of user %v (%v of %v users)\n", len(orders), user, i, *num_users) } for _, order := range orders { if strings.HasPrefix(order.Status, "Order processed successfully") { item := itemMap[order.Item] totalShippedOrders += order.Quantity totalEarnedOrders += float64(order.Quantity) * item.Cost processedOrders++ } else if strings.HasPrefix(order.Status, "Cancelling because of:") { cancelledOrders++ } else if order.Status == shopApi.OrderStatusProcessing { totalProcessingOrders++ } else { fmt.Println("Unknown order status:", order.Status) inconsistent = true break } } } fmt.Println("Orders processed:", processedOrders, "orders cancelled:", cancelledOrders) if totalProcessingOrders > 0 { fmt.Printf("UNFINISHED: There are still %v orders to process\n", totalProcessingOrders) inconsistent = true } if totalShipped != totalShippedOrders { fmt.Printf("Inconsistent: shipped %v, shipped orders %v\n", totalShipped, totalShippedOrders) inconsistent = true } balance, err := bank.Balance("store") check(err) balance = round(balance) totalEarnedOrders = round(totalEarnedOrders) totalEarned = round(totalEarned) if balance != totalEarned { fmt.Printf("Inconsistent earnings bank vs. items. Expected %v, have %v\n", totalEarned, balance) inconsistent = true if balance > totalEarned { fmt.Println(balance-totalEarned, "too much") } else { fmt.Println("Missing", totalEarned-balance) } } if balance != totalEarnedOrders { fmt.Printf("Inconsistent earnings bank vs. orders. Expected %v, have %v\n", totalEarnedOrders, balance) inconsistent = true if balance > totalEarnedOrders { fmt.Println(balance-totalEarnedOrders, "too much") } else { fmt.Println("Missing", totalEarnedOrders-balance) } } fmt.Printf("Total shipped items %v, total earnings %v\n", totalShipped, totalEarned) if inconsistent { fmt.Printf("MAYBE THERE WERE MORE THAN %v USERS?\n", *num_users) os.Exit(1) } else { os.Exit(0) } }