Add ability to override with custom model

Author: stevebauman

src/DatabaseMetricManager.php

@@ -12,13 +12,30 @@ class DatabaseMetricManager implements MetricManager
      */
     protected bool $capturing = false;
 
+    /**
+     * The metric model to use.
+     *
+     * @var class-string<\Illuminate\Database\Eloquent\Model>
+     */
+    public static string $model = Metric::class;
+
     /**
      * Constructor.
      */
     public function __construct(
         protected MetricRepository $repository
     ) {}
 
+    /**
+     * Set the metric model to use.
+     *
+     * @param  class-string<\Illuminate\Database\Eloquent\Model>  $model
+     */
+    public static function useModel(string $model): void
+    {
+        static::$model = $model;
+    }
+
     /**
      * {@inheritDoc}
      */

src/Jobs/RecordMetric.php

@@ -2,6 +2,7 @@
 
 namespace DirectoryTree\Metrics\Jobs;
 
+use DirectoryTree\Metrics\DatabaseMetricManager;
 use DirectoryTree\Metrics\Measurable;
 use DirectoryTree\Metrics\Metric;
 use Illuminate\Bus\Queueable;
@@ -19,7 +20,7 @@ class RecordMetric implements ShouldQueue
      */
     public function __construct(
         /** @var Collection<Measurable>|Measurable */
-        public Collection|Measurable $metrics,
+        public Collection|Measurable $metrics
     ) {}
 
     /**
@@ -38,9 +39,13 @@ public function handle(): void
             fn (Measurable $metric) => $metric->value()
         );
 
-        Metric::query()->getConnection()->transaction(
-            function (ConnectionInterface $connection) use ($metric, $value) {
-                $model = Metric::query()->firstOrCreate([
+        /** @var \Illuminate\Database\Eloquent\Model $model */
+        $model = new DatabaseMetricManager::$model;
+
+        $model->getConnection()->transaction(
+            function (ConnectionInterface $connection) use ($metric, $value, $model) {
+                $instance = $model->newQuery()->firstOrCreate([
+                    ...$metric->additional(),
                     'name' => $metric->name(),
                     'category' => $metric->category(),
                     'year' => $metric->year(),
@@ -50,7 +55,7 @@ function (ConnectionInterface $connection) use ($metric, $value) {
                     'measurable_id' => $metric->measurable()?->getKey(),
                 ], ['value' => 0]);
 
-                Metric::query()->whereKey($model->getKey())->update([
+                $model->newQuery()->whereKey($instance->getKey())->update([
                     'value' => $connection->raw('value + '.$value),
                 ]);
             }