@heyhru/app-dms-server 0.6.0 → 0.7.0

package/dist/index.js

@@ -4,6 +4,42 @@ import { createPgDb } from "@heyhru/server-plugin-pg";
 // src/app.ts
 import Fastify from "fastify";
 import cors from "@fastify/cors";
+import { metricsPlugin } from "@heyhru/server-plugin-metrics";
+
+// src/metrics.ts
+import { Histogram, Counter, Gauge, register } from "@heyhru/server-plugin-metrics";
+import { getPoolStats } from "@heyhru/business-dms-datasource";
+var sqlDuration = new Histogram({
+  name: "dms_sql_duration_seconds",
+  help: "SQL execution duration in seconds",
+  labelNames: ["data_source_id", "sql_type", "status"],
+  buckets: [0.1, 0.5, 1, 2, 5, 10, 30],
+  registers: [register]
+});
+var authLoginTotal = new Counter({
+  name: "dms_auth_login_total",
+  help: "Total login attempts",
+  labelNames: ["status"],
+  registers: [register]
+});
+var dbPoolActive = new Gauge({
+  name: "dms_db_pool_active",
+  help: "Active connections per data source pool",
+  labelNames: ["data_source_id"],
+  registers: [register]
+});
+var dbPoolWaiting = new Gauge({
+  name: "dms_db_pool_waiting",
+  help: "Waiting requests per data source pool",
+  labelNames: ["data_source_id"],
+  registers: [register]
+});
+function refreshPoolMetrics() {
+  for (const { dataSourceId, active, waiting } of getPoolStats()) {
+    if (active !== null) dbPoolActive.set({ data_source_id: dataSourceId }, active);
+    if (waiting !== null) dbPoolWaiting.set({ data_source_id: dataSourceId }, waiting);
+  }
+}
 
 // src/auth/auth.middleware.ts
 import { verifyToken as jwtVerify } from "@heyhru/server-plugin-jwt";
@@ -122,10 +158,12 @@ async function authLogin(req, reply) {
   const user = await getUserByUsername(username);
   if (!user || !await verifyPassword(password, user.password_hash)) {
     req.log.warn("Login failed for user: %s", username);
+    authLoginTotal.inc({ status: "fail" });
     return reply.code(401).send({ error: "\u7528\u6237\u540D\u6216\u5BC6\u7801\u9519\u8BEF" });
   }
   const payload = { id: user.id, username: user.username, role: user.role };
   const token = signToken(payload, config.jwtSecret, config.jwtExpiresIn);
+  authLoginTotal.inc({ status: "success" });
   req.log.info("User logged in: %s", username);
   return reply.send({ ...payload, token });
 }
@@ -237,7 +275,15 @@ async function sqlExecute({ dataSourceId, database, sql, userId, ip }) {
     return ds ? getPool(ds) : null;
   })();
   if (!pool) throw new Error("\u6570\u636E\u6E90\u672A\u627E\u5230");
-  const rows = await pool.execute(sql);
+  const end = sqlDuration.startTimer({ data_source_id: dataSourceId, sql_type: category });
+  let rows;
+  try {
+    rows = await pool.execute(sql);
+    end({ status: "success" });
+  } catch (err) {
+    end({ status: "failed" });
+    throw err;
+  }
   await writeAuditLog({
     userId,
     dataSourceId,
@@ -345,6 +391,10 @@ function savedSqlController(app) {
 async function buildApp() {
   const app = Fastify({ loggerInstance: logger });
   await app.register(cors, { origin: true, credentials: true });
+  await app.register(metricsPlugin);
+  app.addHook("onResponse", () => {
+    refreshPoolMetrics();
+  });
   app.decorateRequest("user", null);
   app.addHook("preHandler", authHook);
   app.addHook("onError", (req, _reply, error, done) => {

package/package.json

@@ -3,7 +3,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "0.6.0",
+  "version": "0.7.0",
   "description": "DMS backend API server built on Fastify",
   "type": "module",
   "main": "./dist/index.mjs",
@@ -19,15 +19,16 @@
   },
   "dependencies": {
     "@fastify/cors": "^11.2.0",
-    "@heyhru/business-dms-approval": "0.6.0",
-    "@heyhru/business-dms-audit": "0.6.0",
-    "@heyhru/business-dms-datasource": "0.6.0",
-    "@heyhru/business-dms-saved-sql": "0.6.0",
-    "@heyhru/business-dms-user": "0.6.0",
+    "@heyhru/business-dms-approval": "0.6.1",
+    "@heyhru/business-dms-audit": "0.6.1",
+    "@heyhru/business-dms-datasource": "0.7.0",
+    "@heyhru/business-dms-saved-sql": "0.6.1",
+    "@heyhru/business-dms-user": "0.6.1",
     "@heyhru/common-util-logger": "0.6.0",
     "@heyhru/server-plugin-jwt": "0.6.0",
-    "@heyhru/server-plugin-mysql": "0.6.0",
-    "@heyhru/server-plugin-pg": "0.6.0",
+    "@heyhru/server-plugin-metrics": "0.7.0",
+    "@heyhru/server-plugin-mysql": "0.7.0",
+    "@heyhru/server-plugin-pg": "0.7.0",
     "@heyhru/server-util-crypto": "0.6.0",
     "fastify": "^5.8.4",
     "node-sql-parser": "^5.4.0",
@@ -41,5 +42,5 @@
     "typescript": "^6.0.2",
     "vitest": "^4.1.4"
   },
-  "gitHead": "c25e6fe17f0b15e07ba534e712cc723be02051b3"
+  "gitHead": "b9097b9ad29ac4c0c3610a39cb0db83f74a11fb8"
 }