add partitioning to pg_telemetry SimpleMetric model

Author: matt-codecov

shared/django_apps/db_settings.py

@@ -161,9 +161,7 @@
 POSTGRES_EXTRA_DB_BACKEND_BASE: "django_prometheus.db.backends.postgresql"  # type: ignore
 
 # Allows to use the pgpartition command
-PSQLEXTRA_PARTITIONING_MANAGER = (
-    "shared.django_apps.user_measurements.partitioning.manager"
-)
+PSQLEXTRA_PARTITIONING_MANAGER = "shared.django_apps.partitioning.manager"
 
 DATABASE_ROUTERS = [
     "shared.django_apps.db_routers.MultiDatabaseRouter",

shared/django_apps/user_measurements/partitioning.py shared/django_apps/partitioning.py

@@ -6,6 +6,7 @@
 )
 from psqlextra.partitioning.config import PostgresPartitioningConfig
 
+from shared.django_apps.pg_telemetry.models import SimpleMetric
 from shared.django_apps.user_measurements.models import UserMeasurement
 
 # Overlapping partitions will cause errors - https://www.postgresql.org/docs/current/ddl-partitioning.html#DDL-PARTITIONING-DECLARATIVE -> "create partitions"
@@ -23,5 +24,13 @@
                 max_age=relativedelta(months=12),
             ),
         ),
+        PostgresPartitioningConfig(
+            model=SimpleMetric,
+            strategy=PostgresCurrentTimePartitioningStrategy(
+                size=PostgresTimePartitionSize(months=1),
+                count=3,
+                max_age=relativedelta(months=6),
+            ),
+        ),
     ]
 )

shared/django_apps/pg_telemetry/models.py

@@ -1,8 +1,10 @@
 from django.conf import settings
 from django.db import models
+from psqlextra.models import PostgresPartitionedModel
+from psqlextra.types import PostgresPartitioningMethod
 
 
-class BaseModel(models.Model):
+class BaseModel(PostgresPartitionedModel):
     """
     Base model for timeseries metrics. It provides a timestamp field which
     represents the time that the data sample was captured at and a few metadata
@@ -13,6 +15,10 @@ class BaseModel(models.Model):
     Timescale for a time, we'll pick one.
     """
 
+    class PartitioningMeta:
+        method = PostgresPartitioningMethod.RANGE
+        key = ["timestamp"]
+
     class Meta:
         abstract = True