InMemoryStatsRecord Interface

src/Adapters/InMemory/InMemoryClientAdapter.php

@@ -6,6 +6,7 @@
 use Cosmastech\StatsDClientAdapter\Adapters\Concerns\TagNormalizerAwareTrait;
 use Cosmastech\StatsDClientAdapter\Adapters\Concerns\TimeClosureTrait;
 use Cosmastech\StatsDClientAdapter\Adapters\Contracts\TagNormalizerAware;
+use Cosmastech\StatsDClientAdapter\Adapters\InMemory\Models\Contracts\InMemoryStatsRecordInterface;
 use Cosmastech\StatsDClientAdapter\Adapters\InMemory\Models\InMemoryCountRecord;
 use Cosmastech\StatsDClientAdapter\Adapters\InMemory\Models\InMemoryDistributionRecord;
 use Cosmastech\StatsDClientAdapter\Adapters\InMemory\Models\InMemoryGaugeRecord;
@@ -31,13 +32,13 @@ class InMemoryClientAdapter implements StatsDClientAdapter, TagNormalizerAware
 
     /**
      * @param  array<mixed, mixed>  $defaultTags
-     * @param  InMemoryStatsRecord  $inMemoryStatsRecord
+     * @param  InMemoryStatsRecordInterface  $inMemoryStatsRecord
      * @param  TagNormalizer  $tagNormalizer
      * @param  ClockInterface  $clock
      */
     public function __construct(
         array $defaultTags = [],
-        InMemoryStatsRecord $inMemoryStatsRecord = new InMemoryStatsRecord(),
+        InMemoryStatsRecordInterface $inMemoryStatsRecord = new InMemoryStatsRecord(),
         TagNormalizer $tagNormalizer = new NoopTagNormalizer(),
         ClockInterface $clock = new Clock()
     ) {
@@ -179,7 +180,7 @@ public function updateStats(array|string $stats, int $delta = 1, float $sampleRa
     /**
      * Get all recorded stats.
      */
-    public function getStats(): InMemoryStatsRecord
+    public function getStats(): InMemoryStatsRecordInterface
     {
         return $this->stats;
     }

src/Adapters/InMemory/Models/Contracts/InMemoryStatsRecordInterface.php

@@ -0,0 +1,58 @@
+<?php
+
+namespace Cosmastech\StatsDClientAdapter\Adapters\InMemory\Models\Contracts;
+
+use Cosmastech\StatsDClientAdapter\Adapters\InMemory\Models\InMemoryCountRecord;
+use Cosmastech\StatsDClientAdapter\Adapters\InMemory\Models\InMemoryDistributionRecord;
+use Cosmastech\StatsDClientAdapter\Adapters\InMemory\Models\InMemoryGaugeRecord;
+use Cosmastech\StatsDClientAdapter\Adapters\InMemory\Models\InMemoryHistogramRecord;
+use Cosmastech\StatsDClientAdapter\Adapters\InMemory\Models\InMemorySetRecord;
+use Cosmastech\StatsDClientAdapter\Adapters\InMemory\Models\InMemoryTimingRecord;
+
+interface InMemoryStatsRecordInterface
+{
+    public function recordTiming(InMemoryTimingRecord $inMemoryTimingRecord): void;
+
+    /**
+     * @return array<int, InMemoryTimingRecord>|iterable<InMemoryTimingRecord>
+     */
+    public function getTimings();
+
+    public function recordCount(InMemoryCountRecord $inMemoryCountRecord): void;
+
+    /**
+     * @return array<int, InMemoryCountRecord>|iterable<InMemoryCountRecord>
+     */
+    public function getCounts();
+
+    public function recordGauge(InMemoryGaugeRecord $inMemoryGaugeRecord): void;
+
+    /**
+     * @return array<int, InMemoryGaugeRecord>|iterable<InMemoryGaugeRecord>
+     */
+    public function getGauges();
+
+
+    public function recordSet(InMemorySetRecord $inMemorySetRecord): void;
+
+    /**
+     * @return array<int, InMemorySetRecord>|iterable<InMemorySetRecord>
+     */
+    public function getSets();
+
+    public function recordHistogram(InMemoryHistogramRecord $inMemoryHistogramRecord): void;
+
+    /**
+     * @return array<int, InMemoryHistogramRecord>|iterable<InMemoryHistogramRecord>
+     */
+    public function getHistograms();
+
+    public function recordDistribution(InMemoryDistributionRecord $inMemoryDistributionRecord): void;
+
+    /**
+     * @return array<int, InMemoryDistributionRecord>|iterable<InMemoryDistributionRecord>
+     */
+    public function getDistributions();
+
+    public function flush(): void;
+}

src/Adapters/InMemory/Models/InMemoryStatsRecord.php

@@ -3,12 +3,13 @@
 namespace Cosmastech\StatsDClientAdapter\Adapters\InMemory\Models;
 
 use Cosmastech\StatsDClientAdapter\Adapters\InMemory\Models\Concerns\GetAndSetRecordsTrait;
+use Cosmastech\StatsDClientAdapter\Adapters\InMemory\Models\Contracts\InMemoryStatsRecordInterface;
 use Cosmastech\StatsDClientAdapter\Adapters\InMemory\Models\Contracts\RecordInterface;
 
 /**
  * Container class for storing all stats.
  */
-class InMemoryStatsRecord implements RecordInterface
+class InMemoryStatsRecord implements RecordInterface, InMemoryStatsRecordInterface
 {
     use GetAndSetRecordsTrait;
 

tests/Adapters/InMemory/InMemoryTimingTest.php

@@ -29,6 +29,7 @@ public function storesTimingRecord(): void
         $inMemoryClient->timing("timing-stat", 199, 0.2, ["timing" => "some-value"]);
 
         // Then
+        /** @var InMemoryStatsRecord $statsRecord */
         $statsRecord = $inMemoryClient->getStats();
         self::assertCount(1, $statsRecord->getTimings());