feat(dashboard): 接入预测结果聚合

This commit is contained in:
2026-05-11 22:21:57 +08:00
parent b11d37e9d8
commit 1a2ff19cf4
4 changed files with 125 additions and 12 deletions
+14 -2
View File
@@ -1,11 +1,23 @@
import { createBatteriesResponse, createDashboardSnapshot } from '@/domain/battery'
import { os } from '@/server/api/server'
import { getBatteryHistory, getLatestBatteryPerDevice } from '@/server/battery/mysql'
import { getBatteryHistory, getBatteryPredictionHistory, getLatestBatteryPerDevice } from '@/server/battery/mysql'
import { isPredictionEnabled, predictSoh } from '@/server/prediction/client'
export const dashboard = os.battery.dashboard.handler(async () => {
const items = await getLatestBatteryPerDevice()
const predictionEntries = isPredictionEnabled()
? await Promise.all(
items.map(async (item) => {
const history = await getBatteryPredictionHistory(item.mac)
const prediction = await predictSoh(item, history)
return createDashboardSnapshot(items)
return prediction ? ([item.mac, prediction] as const) : null
}),
)
: []
const predictions = new Map(predictionEntries.filter((entry) => entry !== null))
return createDashboardSnapshot(items, new Date(), predictions)
})
export const batteries = os.battery.batteries.handler(async ({ input }) => {
+16
View File
@@ -4,6 +4,7 @@ import { type BatteryInfo, type BatteryInfoSourceRow, toBatteryInfo } from '@/do
import { env } from '@/env'
const historyLimit = 500
const predictionHistoryLimit = 10
type BatteryInfoMysqlRow = RowDataPacket & BatteryInfoSourceRow
@@ -55,6 +56,21 @@ export async function getBatteryHistory(mac: string): Promise<BatteryInfo[]> {
return rows.map(toBatteryInfo)
}
export async function getBatteryPredictionHistory(mac: string): Promise<BatteryInfo[]> {
const [rows] = await getBatteryPool().query<BatteryInfoMysqlRow[]>(
`
SELECT ${sourceColumns}
FROM ls_battery_info
WHERE mac = :mac
ORDER BY create_time DESC, id DESC
LIMIT :limit
`,
{ mac, limit: predictionHistoryLimit },
)
return rows.map(toBatteryInfo).reverse()
}
export async function getLatestBatteryPerDevice(): Promise<BatteryInfo[]> {
const [rows] = await getBatteryPool().query<BatteryInfoMysqlRow[]>(`
SELECT ${sourceColumns}