feat(db): 新增 portfolio_snapshots 表与每日快照记录 Action

2026-04-29 11:37:11 +08:00 · 2026-04-29 11:37:11 +08:00 · 4c4e6ab565
commit 4c4e6ab565
parent 955b01fd79
6 changed files with 525 additions and 1 deletions
--- a/Memory.md
+++ b/Memory.md
@ -11,11 +11,13 @@
 - 完成核心 `transactions` (交易流水) 表的建立，并严格运用了 `numeric(36,18)` 的高精度配置。
 - `assets` 表完成多次业务演进：新增 `latestPrice` (支持现价追踪)、`exchange` (显式交易所绑定) 以及 `name` (中文名称解析) 字段。
 - `exchange_rates` (汇率表) 已建立，支持联合主键与跨币种交叉汇率架构。
+- **引入 `portfolio_snapshots` 表**：用于每日记录投资组合快照，字段包括 `date` (唯一日期)、`total_value_cny` (当日总市值)、`total_cost_cny` (当日总投入本金)，为历史净值走势图奠定底层数据结构。

 ## 核心业务与服务端逻辑 (Server Actions)
 - 完成高精度交易流水与资产的 Server Actions 开发，成功实现字符串级别的高精度防腐层拦截（基于 Zod & Big.js）。
 - 补全资产与流水的全栈增删改查 (CRUD) 操作，`createTransaction` 现已支持根据 `exchange` 自动判定并锁定 `txCurrency`。
 - **估值与 P&L 引擎：** 完成底层估值引擎升级，打通交叉汇率换算逻辑；实现原币种 (Native) 与本位币 (CNY Base) 双轨制的历史成本追溯与真实盈亏 (P&L) 计算引擎。
+- **快照记录引擎：** 新增 `src/actions/snapshots.ts`，`recordDailySnapshot()` 函数基于 `getPortfolioPositions()` 实时计算总市值与总成本，使用 `Asia/Shanghai` 时区获取当日日期，执行 Upsert 逻辑确保每天仅存一条记录；`getSnapshots()` 支持按日期范围与数量限制查询历史快照数据。

 ## 外部行情接口与网络 (Market Data Engines)
 - **股票行情引擎：** 彻底抛弃低效海外接口，自主研发智能路由接入腾讯财经 (`qt.gtimg.cn`) 极速接口。引入原生 `ArrayBuffer` 与 `TextDecoder(gbk)` 彻底解决历史中文乱码问题，实现沪、深、港、美四大市场毫秒级实时同步。
--- a/drizzle/0003_watery_xorn.sql
+++ b/drizzle/0003_watery_xorn.sql
@ -0,0 +1,24 @@
+CREATE TABLE "exchange_rates" (
+	"id" uuid PRIMARY KEY NOT NULL,
+	"from_currency" varchar(10) NOT NULL,
+	"to_currency" varchar(10) NOT NULL,
+	"rate" numeric(20, 8) NOT NULL,
+	"updated_at" timestamp with time zone DEFAULT now()
+);
+--> statement-breakpoint
+CREATE TABLE "portfolio_snapshots" (
+	"id" uuid PRIMARY KEY NOT NULL,
+	"date" date NOT NULL,
+	"total_value_cny" numeric(36, 18) NOT NULL,
+	"total_cost_cny" numeric(36, 18) NOT NULL,
+	"created_at" timestamp with time zone DEFAULT now() NOT NULL,
+	"updated_at" timestamp with time zone DEFAULT now() NOT NULL,
+	CONSTRAINT "portfolio_snapshots_date_unique" UNIQUE("date")
+);
+--> statement-breakpoint
+ALTER TABLE "assets" ALTER COLUMN "id" DROP DEFAULT;--> statement-breakpoint
+ALTER TABLE "users" ALTER COLUMN "id" DROP DEFAULT;--> statement-breakpoint
+ALTER TABLE "assets" ADD COLUMN "name" varchar(100);--> statement-breakpoint
+ALTER TABLE "assets" ADD COLUMN "exchange" varchar(10) DEFAULT 'US';--> statement-breakpoint
+ALTER TABLE "assets" ADD COLUMN "latest_price" numeric(36, 18) DEFAULT '0' NOT NULL;--> statement-breakpoint
+CREATE UNIQUE INDEX "currency_pair_idx" ON "exchange_rates" USING btree ("from_currency","to_currency");
--- a/drizzle/meta/0003_snapshot.json
+++ b/drizzle/meta/0003_snapshot.json
@ -0,0 +1,375 @@
+{
+  "id": "26b392bf-03db-4f71-86d5-395d5c07fae5",
+  "prevId": "18d8b1e1-2700-4d54-adf8-493d526d5bf5",
+  "version": "7",
+  "dialect": "postgresql",
+  "tables": {
+    "public.assets": {
+      "name": "assets",
+      "schema": "",
+      "columns": {
+        "id": {
+          "name": "id",
+          "type": "uuid",
+          "primaryKey": true,
+          "notNull": true
+        },
+        "symbol": {
+          "name": "symbol",
+          "type": "varchar(20)",
+          "primaryKey": false,
+          "notNull": true
+        },
+        "name": {
+          "name": "name",
+          "type": "varchar(100)",
+          "primaryKey": false,
+          "notNull": false
+        },
+        "type": {
+          "name": "type",
+          "type": "asset_type_enum",
+          "typeSchema": "public",
+          "primaryKey": false,
+          "notNull": true
+        },
+        "exchange": {
+          "name": "exchange",
+          "type": "varchar(10)",
+          "primaryKey": false,
+          "notNull": false,
+          "default": "'US'"
+        },
+        "base_currency": {
+          "name": "base_currency",
+          "type": "varchar(10)",
+          "primaryKey": false,
+          "notNull": true
+        },
+        "latest_price": {
+          "name": "latest_price",
+          "type": "numeric(36, 18)",
+          "primaryKey": false,
+          "notNull": true,
+          "default": "'0'"
+        },
+        "created_at": {
+          "name": "created_at",
+          "type": "timestamp with time zone",
+          "primaryKey": false,
+          "notNull": false,
+          "default": "now()"
+        }
+      },
+      "indexes": {},
+      "foreignKeys": {},
+      "compositePrimaryKeys": {},
+      "uniqueConstraints": {
+        "assets_symbol_unique": {
+          "name": "assets_symbol_unique",
+          "nullsNotDistinct": false,
+          "columns": [
+            "symbol"
+          ]
+        }
+      },
+      "policies": {},
+      "checkConstraints": {},
+      "isRLSEnabled": false
+    },
+    "public.exchange_rates": {
+      "name": "exchange_rates",
+      "schema": "",
+      "columns": {
+        "id": {
+          "name": "id",
+          "type": "uuid",
+          "primaryKey": true,
+          "notNull": true
+        },
+        "from_currency": {
+          "name": "from_currency",
+          "type": "varchar(10)",
+          "primaryKey": false,
+          "notNull": true
+        },
+        "to_currency": {
+          "name": "to_currency",
+          "type": "varchar(10)",
+          "primaryKey": false,
+          "notNull": true
+        },
+        "rate": {
+          "name": "rate",
+          "type": "numeric(20, 8)",
+          "primaryKey": false,
+          "notNull": true
+        },
+        "updated_at": {
+          "name": "updated_at",
+          "type": "timestamp with time zone",
+          "primaryKey": false,
+          "notNull": false,
+          "default": "now()"
+        }
+      },
+      "indexes": {
+        "currency_pair_idx": {
+          "name": "currency_pair_idx",
+          "columns": [
+            {
+              "expression": "from_currency",
+              "isExpression": false,
+              "asc": true,
+              "nulls": "last"
+            },
+            {
+              "expression": "to_currency",
+              "isExpression": false,
+              "asc": true,
+              "nulls": "last"
+            }
+          ],
+          "isUnique": true,
+          "concurrently": false,
+          "method": "btree",
+          "with": {}
+        }
+      },
+      "foreignKeys": {},
+      "compositePrimaryKeys": {},
+      "uniqueConstraints": {},
+      "policies": {},
+      "checkConstraints": {},
+      "isRLSEnabled": false
+    },
+    "public.portfolio_snapshots": {
+      "name": "portfolio_snapshots",
+      "schema": "",
+      "columns": {
+        "id": {
+          "name": "id",
+          "type": "uuid",
+          "primaryKey": true,
+          "notNull": true
+        },
+        "date": {
+          "name": "date",
+          "type": "date",
+          "primaryKey": false,
+          "notNull": true
+        },
+        "total_value_cny": {
+          "name": "total_value_cny",
+          "type": "numeric(36, 18)",
+          "primaryKey": false,
+          "notNull": true
+        },
+        "total_cost_cny": {
+          "name": "total_cost_cny",
+          "type": "numeric(36, 18)",
+          "primaryKey": false,
+          "notNull": true
+        },
+        "created_at": {
+          "name": "created_at",
+          "type": "timestamp with time zone",
+          "primaryKey": false,
+          "notNull": true,
+          "default": "now()"
+        },
+        "updated_at": {
+          "name": "updated_at",
+          "type": "timestamp with time zone",
+          "primaryKey": false,
+          "notNull": true,
+          "default": "now()"
+        }
+      },
+      "indexes": {},
+      "foreignKeys": {},
+      "compositePrimaryKeys": {},
+      "uniqueConstraints": {
+        "portfolio_snapshots_date_unique": {
+          "name": "portfolio_snapshots_date_unique",
+          "nullsNotDistinct": false,
+          "columns": [
+            "date"
+          ]
+        }
+      },
+      "policies": {},
+      "checkConstraints": {},
+      "isRLSEnabled": false
+    },
+    "public.transactions": {
+      "name": "transactions",
+      "schema": "",
+      "columns": {
+        "id": {
+          "name": "id",
+          "type": "uuid",
+          "primaryKey": true,
+          "notNull": true
+        },
+        "asset_id": {
+          "name": "asset_id",
+          "type": "uuid",
+          "primaryKey": false,
+          "notNull": true
+        },
+        "tx_type": {
+          "name": "tx_type",
+          "type": "transaction_type_enum",
+          "typeSchema": "public",
+          "primaryKey": false,
+          "notNull": true
+        },
+        "quantity": {
+          "name": "quantity",
+          "type": "numeric(36, 18)",
+          "primaryKey": false,
+          "notNull": true
+        },
+        "price": {
+          "name": "price",
+          "type": "numeric(36, 18)",
+          "primaryKey": false,
+          "notNull": true
+        },
+        "fee": {
+          "name": "fee",
+          "type": "numeric(36, 18)",
+          "primaryKey": false,
+          "notNull": true,
+          "default": "'0'"
+        },
+        "tx_currency": {
+          "name": "tx_currency",
+          "type": "varchar(10)",
+          "primaryKey": false,
+          "notNull": true
+        },
+        "exchange_rate": {
+          "name": "exchange_rate",
+          "type": "numeric(20, 8)",
+          "primaryKey": false,
+          "notNull": true,
+          "default": "'1'"
+        },
+        "executed_at": {
+          "name": "executed_at",
+          "type": "timestamp with time zone",
+          "primaryKey": false,
+          "notNull": true
+        },
+        "created_at": {
+          "name": "created_at",
+          "type": "timestamp with time zone",
+          "primaryKey": false,
+          "notNull": false,
+          "default": "now()"
+        }
+      },
+      "indexes": {},
+      "foreignKeys": {
+        "transactions_asset_id_assets_id_fk": {
+          "name": "transactions_asset_id_assets_id_fk",
+          "tableFrom": "transactions",
+          "tableTo": "assets",
+          "columnsFrom": [
+            "asset_id"
+          ],
+          "columnsTo": [
+            "id"
+          ],
+          "onDelete": "no action",
+          "onUpdate": "no action"
+        }
+      },
+      "compositePrimaryKeys": {},
+      "uniqueConstraints": {},
+      "policies": {},
+      "checkConstraints": {},
+      "isRLSEnabled": false
+    },
+    "public.users": {
+      "name": "users",
+      "schema": "",
+      "columns": {
+        "id": {
+          "name": "id",
+          "type": "uuid",
+          "primaryKey": true,
+          "notNull": true
+        },
+        "username": {
+          "name": "username",
+          "type": "varchar(50)",
+          "primaryKey": false,
+          "notNull": true
+        },
+        "password_hash": {
+          "name": "password_hash",
+          "type": "varchar(255)",
+          "primaryKey": false,
+          "notNull": true
+        },
+        "created_at": {
+          "name": "created_at",
+          "type": "timestamp with time zone",
+          "primaryKey": false,
+          "notNull": true,
+          "default": "now()"
+        }
+      },
+      "indexes": {},
+      "foreignKeys": {},
+      "compositePrimaryKeys": {},
+      "uniqueConstraints": {
+        "users_username_unique": {
+          "name": "users_username_unique",
+          "nullsNotDistinct": false,
+          "columns": [
+            "username"
+          ]
+        }
+      },
+      "policies": {},
+      "checkConstraints": {},
+      "isRLSEnabled": false
+    }
+  },
+  "enums": {
+    "public.asset_type_enum": {
+      "name": "asset_type_enum",
+      "schema": "public",
+      "values": [
+        "STOCK",
+        "CRYPTO",
+        "CASH"
+      ]
+    },
+    "public.transaction_type_enum": {
+      "name": "transaction_type_enum",
+      "schema": "public",
+      "values": [
+        "BUY",
+        "SELL",
+        "DIVIDEND",
+        "AIRDROP",
+        "FEE"
+      ]
+    }
+  },
+  "schemas": {},
+  "sequences": {},
+  "roles": {},
+  "policies": {},
+  "views": {},
+  "_meta": {
+    "columns": {},
+    "schemas": {},
+    "tables": {}
+  }
+}
--- a/drizzle/meta/_journal.json
+++ b/drizzle/meta/_journal.json
@ -22,6 +22,13 @@
      "when": 1777289592183,
      "tag": "0002_rapid_invaders",
      "breakpoints": true
+    },
+    {
+      "idx": 3,
+      "version": "7",
+      "when": 1777433725731,
+      "tag": "0003_watery_xorn",
+      "breakpoints": true
    }
  ]
 }
--- a/src/actions/snapshots.ts
+++ b/src/actions/snapshots.ts
@ -0,0 +1,103 @@
+'use server';
+
+import { db } from '@/db';
+import { portfolioSnapshots } from '@/db/schema';
+import { getPortfolioPositions } from './portfolio';
+import { eq, sql } from 'drizzle-orm';
+
+function getTodayInShanghai(): string {
+  const now = new Date();
+  const utcStr = now.toLocaleString('en-US', { timeZone: 'UTC' });
+  const utcDate = new Date(utcStr);
+  const shanghaiOffset = 8 * 60 * 60 * 1000;
+  const shanghaiDate = new Date(utcDate.getTime() + shanghaiOffset);
+  const year = shanghaiDate.getFullYear();
+  const month = String(shanghaiDate.getMonth() + 1).padStart(2, '0');
+  const day = String(shanghaiDate.getDate()).padStart(2, '0');
+  return `${year}-${month}-${day}`;
+}
+
+export async function recordDailySnapshot() {
+  const positions = await getPortfolioPositions();
+
+  const totalValueCny = positions.reduce(
+    (sum, pos) => sum.plus(pos.cnyValue || '0'),
+    0 as number | string
+  );
+
+  const totalCostCny = positions.reduce(
+    (sum, pos) => sum.plus(pos.totalCostCny || '0'),
+    0 as number | string
+  );
+
+  const dateStr = getTodayInShanghai();
+
+  const existing = await db
+    .select()
+    .from(portfolioSnapshots)
+    .where(eq(portfolioSnapshots.date, dateStr))
+    .limit(1);
+
+  const now = new Date();
+
+  if (existing.length > 0) {
+    await db
+      .update(portfolioSnapshots)
+      .set({
+        totalValueCny: String(totalValueCny),
+        totalCostCny: String(totalCostCny),
+        updatedAt: now,
+      })
+      .where(eq(portfolioSnapshots.date, dateStr));
+
+    return {
+      success: true,
+      action: 'updated',
+      date: dateStr,
+      totalValueCny: String(totalValueCny),
+      totalCostCny: String(totalCostCny),
+    };
+  }
+
+  await db
+    .insert(portfolioSnapshots)
+    .values({
+      date: dateStr,
+      totalValueCny: String(totalValueCny),
+      totalCostCny: String(totalCostCny),
+      createdAt: now,
+      updatedAt: now,
+    });
+
+  return {
+    success: true,
+    action: 'inserted',
+    date: dateStr,
+    totalValueCny: String(totalValueCny),
+    totalCostCny: String(totalCostCny),
+  };
+}
+
+export async function getSnapshots(params?: {
+  limit?: number;
+  startDate?: string;
+  endDate?: string;
+}) {
+  const { limit = 365, startDate, endDate } = params || {};
+
+  let query = db
+    .select()
+    .from(portfolioSnapshots)
+    .orderBy(sql`"${date}" DESC`);
+
+  if (startDate) {
+    query = query.where(sql`"${date}" >= ${startDate}`);
+  }
+  if (endDate) {
+    query = query.where(sql`"${date}" <= ${endDate}`);
+  }
+
+  const snapshots = await query.limit(limit);
+
+  return snapshots.reverse();
+}
--- a/src/db/schema.ts
+++ b/src/db/schema.ts
@ -1,4 +1,4 @@
-import { pgTable, uuid, varchar, timestamp, pgEnum, numeric, uniqueIndex } from "drizzle-orm/pg-core";
+import { pgTable, uuid, varchar, timestamp, pgEnum, numeric, uniqueIndex, date } from "drizzle-orm/pg-core";

 export const users = pgTable("users", {
  id: uuid("id").primaryKey().$defaultFn(() => crypto.randomUUID()),
@ -65,3 +65,16 @@ export const exchangeRates = pgTable("exchange_rates", {
 }, (table) => [
  uniqueIndex("currency_pair_idx").on(table.fromCurrency, table.toCurrency),
 ]);
+
+export const portfolioSnapshots = pgTable("portfolio_snapshots", {
+  id: uuid("id").primaryKey().$defaultFn(() => crypto.randomUUID()),
+  date: date("date", { mode: "string" }).notNull().unique(),
+  totalValueCny: numeric("total_value_cny", { precision: 36, scale: 18 }).notNull(),
+  totalCostCny: numeric("total_cost_cny", { precision: 36, scale: 18 }).notNull(),
+  createdAt: timestamp("created_at", { withTimezone: true, mode: "date" })
+    .defaultNow()
+    .notNull(),
+  updatedAt: timestamp("updated_at", { withTimezone: true, mode: "date" })
+    .defaultNow()
+    .notNull(),
+});