fix docker compose

Author: leordev

metrics-collector/.dockerignore

@@ -5,4 +5,5 @@ build
 .git
 .vscode
 *metrics.csv
-*metrics.json
+*metrics.json
+data

metrics-collector/.gitignore

@@ -4,3 +4,4 @@ temp
 *metrics.json
 *metrics.csv
 build
+data

metrics-collector/Dockerfile

@@ -19,5 +19,8 @@ COPY . .
 # Build the TypeScript code
 RUN pnpm run build
 
+# Create a directory for storing state data
+RUN mkdir -p /usr/src/app/data
+
 # Expose the port the app runs on
 EXPOSE 3001

metrics-collector/docker-compose.yml

@@ -1,5 +1,3 @@
-version: "3"
-
 services:
   postgres:
     image: postgres:15-alpine
@@ -8,6 +6,11 @@ services:
       POSTGRES_USER: admin
       POSTGRES_PASSWORD: supersecret
       POSTGRES_DB: metrics
+    healthcheck:
+      test: ["CMD-SHELL", "pg_isready -U admin -d metrics"]
+      interval: 5s
+      timeout: 5s
+      retries: 5
     volumes:
       - pgdata:/var/lib/postgresql/data
     ports:
@@ -24,9 +27,12 @@ services:
       DB_PASSWORD: supersecret
       DB_NAME: metrics
     depends_on:
-      - postgres
+      postgres:
+        condition: service_healthy
     ports:
       - "3001:3001"
+    volumes:
+      - metrics-data:/usr/src/app/data
 
   grafana:
     image: grafana/grafana:latest
@@ -44,3 +50,4 @@ services:
 volumes:
   pgdata:
   grafana-storage:
+  metrics-data:

metrics-collector/src/index.ts

@@ -12,7 +12,7 @@ import {
 } from "./sonatype-metrics";
 import { getYesterdayDate, readJsonFile } from "./utils";
 import { readFile, writeFile } from "fs/promises";
-import { existsSync } from "fs";
+import { existsSync, mkdirSync } from "fs";
 
 const isLocalPersistence = process.env.PERSIST_LOCAL_FILES === "true";
 
@@ -152,8 +152,7 @@ async function initialLoad(
 }
 
 export const getLastSavedState = async (metricName: string) => {
-  const stateDir = process.env.LAST_SAVED_STATE_PATH || "./";
-  const filePath = `${stateDir}/last-saved-state-${metricName}`;
+  const filePath = `./data/last-saved-state-${metricName}`;
   if (!existsSync(filePath)) {
     return undefined;
   }
@@ -162,8 +161,11 @@ export const getLastSavedState = async (metricName: string) => {
 };
 
 export const saveLastSavedState = (metricName: string, date: Date) => {
-  const stateDir = process.env.LAST_SAVED_STATE_PATH || "./";
-  const filePath = `${stateDir}/last-saved-state-${metricName}`;
+  const dataDir = "./data";
+  if (!existsSync(dataDir)) {
+    mkdirSync(dataDir, { recursive: true });
+  }
+  const filePath = `${dataDir}/last-saved-state-${metricName}`;
   return writeFile(filePath, date.toISOString());
 };