Add benchmarks

benchmark/carsales_proto.py

@@ -0,0 +1,97 @@
+#!/usr/bin/env python
+
+import carsales_pb2
+from common import rand_int, rand_double, rand_bool, from_bytes_helper
+from random import choice
+
+MAKES = ["Toyota", "GM", "Ford", "Honda", "Tesla"]
+MODELS = ["Camry", "Prius", "Volt", "Accord", "Leaf", "Model S"]
+COLORS = ["black", "white", "red", "green", "blue", "cyan", "magenta", "yellow", "silver"]
+
+def random_car(car):
+    car.make = choice(MAKES)
+    car.model = choice(MODELS)
+    car.color = rand_int(len(COLORS))
+
+    car.seats = 2 + rand_int(6)
+    car.doors = 2 + rand_int(3)
+
+    for _ in range(4):
+        wheel = car.wheel.add()
+        wheel.diameter = 25 + rand_int(15)
+        wheel.air_pressure = 30 + rand_double(20)
+        wheel.snow_tires = rand_int(16) == 0
+
+    car.length = 170 + rand_int(150)
+    car.width = 48 + rand_int(36)
+    car.height = 54 + rand_int(48)
+    car.weight = car.length * car.width * car.height / 200
+
+    engine = car.engine
+    engine.horsepower = 100 * rand_int(400)
+    engine.cylinders = 4 + 2 * rand_int(3)
+    engine.cc = 800 + rand_int(10000)
+    engine.uses_gas = True
+    engine.uses_electric = rand_bool()
+
+    car.fuel_capacity = 10.0 + rand_double(30.0)
+    car.fuel_level = rand_double(car.fuel_capacity)
+    car.has_power_windows = rand_bool()
+    car.has_power_steering = rand_bool()
+    car.has_cruise_control = rand_bool()
+    car.cup_holders = rand_int(12)
+    car.has_nav_system = rand_bool()
+
+def calc_value(car):
+    result = 0
+
+    result += car.seats * 200
+    result += car.doors * 350
+    for wheel in car.wheel:
+        result += wheel.diameter * wheel.diameter
+        result += 100 if wheel.snow_tires else 0
+
+    result += car.length * car.width * car.height / 50
+
+    engine = car.engine
+    result += engine.horsepower * 40
+    if engine.uses_electric:
+        if engine.uses_gas:
+          result += 5000
+        else:
+          result += 3000
+
+    result += 100 if car.has_power_windows else 0
+    result += 200 if car.has_power_steering else 0
+    result += 400 if car.has_cruise_control else 0
+    result += 2000 if car.has_nav_system else 0
+
+    result += car.cup_holders * 25
+
+    return result
+
+class Benchmark:
+    def __init__(self, compression):
+        self.Request = carsales_pb2.ParkingLot
+        self.Response = carsales_pb2.TotalValue
+        self.from_bytes_request = from_bytes_helper(carsales_pb2.ParkingLot)
+        self.from_bytes_response = from_bytes_helper(carsales_pb2.TotalValue)
+        self.to_bytes = lambda x: x.SerializeToString()
+
+    def setup(self, request):
+        result = 0
+        for _ in range(rand_int(200)):
+          car = request.car.add()
+          random_car(car)
+          result += calc_value(car)
+        return result
+
+    def handle(self, request, response):
+        result = 0
+        for car in request.car:
+          result += calc_value(car)
+
+        response.amount = result
+
+    def check(self, response, expected):
+        return response.amount == expected