From 94559712c8a4e63cda99063cbb97d54df5c00a08 Mon Sep 17 00:00:00 2001
From: Eden Kirin <eden.kirin@intis.hr>
Date: Wed, 11 Oct 2023 21:07:02 +0200
Subject: [PATCH] Msgspec benchmark

---
 benchmark/factories.py                   |  2 +-
 benchmark/msgspec_benchmark/benchmark.py | 13 ++++++
 benchmark/msgspec_benchmark/models.py    | 51 ++++++++++++++++++++++++
 main.py                                  |  4 ++
 4 files changed, 69 insertions(+), 1 deletion(-)

diff --git a/benchmark/factories.py b/benchmark/factories.py
index aa9a8c9..c5e9c62 100644
--- a/benchmark/factories.py
+++ b/benchmark/factories.py
@@ -9,7 +9,7 @@ from benchmark.pydantic_benchmark.models import (
 )
 
 
-COLUMNS_COUNT = 100
+COLUMNS_COUNT = 1000
 PLANOGRAMS_COUNT = 1000
 
 
diff --git a/benchmark/msgspec_benchmark/benchmark.py b/benchmark/msgspec_benchmark/benchmark.py
index e69de29..116b0b5 100644
--- a/benchmark/msgspec_benchmark/benchmark.py
+++ b/benchmark/msgspec_benchmark/benchmark.py
@@ -0,0 +1,13 @@
+import msgspec
+from benchmark.base import BenchmarkBase
+from benchmark.msgspec_benchmark.models import PlanogramsBulkInputPayload
+
+
+class MsgSpecBenchmark(BenchmarkBase):
+    def _benchmark(self) -> None:
+        test_data = self._read_test_file()
+
+        # decoder = msgspec.json.Decoder(PlanogramsBulkInputPayload).decode
+        # decoder.loads(test_data)
+
+        data = msgspec.json.decode(test_data, type=PlanogramsBulkInputPayload)
diff --git a/benchmark/msgspec_benchmark/models.py b/benchmark/msgspec_benchmark/models.py
index e69de29..c2e2ea4 100644
--- a/benchmark/msgspec_benchmark/models.py
+++ b/benchmark/msgspec_benchmark/models.py
@@ -0,0 +1,51 @@
+from enum import Enum
+from typing import Annotated, Optional
+from uuid import uuid4
+from msgspec import Meta, Struct, field
+
+
+QuantityInt = Annotated[int, Meta(ge=0, le=2147483647)]
+StrictSmallInt = Annotated[int, Meta(ge=0, le=32767)]
+PriceFloat = Annotated[float, Meta(ge=0, le=99999999.99)]
+ExternalId = Annotated[str, Meta(min_length=1, max_length=32)]
+
+
+class ColumnItemType(str, Enum):
+    PRODUCT = "PRODUCT"
+    COMPONENT = "COMPONENT"
+
+
+class CorrelationId(Struct, rename="camel"):
+    correlation_id: str = field(default_factory=lambda: uuid4().hex)
+
+
+class ColumnsInput(Struct, rename="camel"):
+    column_number: StrictSmallInt
+    external_product_id: Optional[ExternalId] = field(default=None)
+    old_qty: Optional[QuantityInt] = field(default_factory=lambda: None)
+    new_qty: Optional[QuantityInt] = field(default_factory=lambda: None)
+    old_price: Optional[PriceFloat] = field(default_factory=lambda: None)
+    new_price: Optional[PriceFloat] = field(default_factory=lambda: None)
+    select_map: Optional[list[StrictSmallInt]] = field(default_factory=lambda: None)
+    item_type: Optional[ColumnItemType] = field(
+        default_factory=lambda: ColumnItemType.PRODUCT
+    )
+
+
+class PlanogramInput(CorrelationId, Struct, rename="camel"):
+    machine_external_id: ExternalId = field(default="")
+    columns: list[ColumnsInput] = field(default_factory=list)
+
+    # class Config:
+    #     title = "Planogram"
+    #     alias_generator = to_camel_case
+    #     populate_by_name = True
+    #     str_strip_whitespace = True
+
+
+class PlanogramsBulkInputPayload(Struct, rename="camel"):
+    planograms: list[PlanogramInput] = field(default_factory=list)
+
+    # class Config:
+    #     populate_by_name = True
+    #     alias_generator = to_camel_case
diff --git a/main.py b/main.py
index d944575..5865f61 100644
--- a/main.py
+++ b/main.py
@@ -1,5 +1,6 @@
 from pathlib import Path
 from benchmark.factories import create_test_file
+from benchmark.msgspec_benchmark.benchmark import MsgSpecBenchmark
 from benchmark.pydantic_benchmark.benchmark import PydanticBenchmark
 
 
@@ -10,6 +11,9 @@ def main() -> None:
     pydantic_benchmark = PydanticBenchmark(TEST_DATA_FILE)
     pydantic_benchmark.execute()
 
+    msgspec_benchmark = MsgSpecBenchmark(TEST_DATA_FILE)
+    msgspec_benchmark.execute()
+
 
 if __name__ == "__main__":
     # create_test_file(TEST_DATA_FILE)