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)
}
}