Skip to content

Commit

Permalink
fix: 还没修干净
Browse files Browse the repository at this point in the history
  • Loading branch information
@SALTWOOD
SALTWOOD committed Sep 30, 2024
1 parent 21e3546 commit 12148ff
Show file tree
Hide file tree
Showing 2 changed files with 183 additions and 194 deletions.
166 changes: 76 additions & 90 deletions src/statistics/ClusterStats.ts
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,104 +2,90 @@ import * as fs from 'fs';
import * as path from 'path';
import { Config } from '../Config.js';

export class StatsStorage {
public readonly id: string;
private data: { date: string, hits: number, bytes: number }[];
private filePath: string;
private saveInterval: NodeJS.Timeout;
private dataUpdated: boolean; // 标志是否有数据更新

constructor(id: string) {
this.id = id;
this.filePath = path.join(Config.getInstance().statsDir, `${this.id}.stats.json`); // 改为 .json 文件
this.data = [];
this.dataUpdated = false; // 初始时数据未更新

// 如果文件存在，加载已有的数据
if (fs.existsSync(this.filePath)) {
this.loadFromFile();
}

// 每10分钟保存一次数据（如果有更新）
this.saveInterval = setInterval(() => {
this.maybeWriteToFile();
}, 10 * 60 * 1000); // 10分钟
}

public addData({ hits, bytes }: { hits: number, bytes: number }): void {
const today = new Date().toISOString().split('T')[0];

let todayData = this.data.find(entry => entry.date === today);
if (!todayData) {
todayData = { date: today, hits: 0, bytes: 0 };
this.data.push(todayData);
if (this.data.length > 30) {
this.data.shift(); // 保持只存30天的数据
}
}

todayData.hits += hits;
todayData.bytes += bytes;
this.dataUpdated = true; // 数据已更新
}
const getLocalDateString = (date = new Date()): string => {
return new Intl.DateTimeFormat('sv-SE', { timeZone: 'system', year: 'numeric', month: '2-digit', day: '2-digit' }).format(date);
};

const readFileData = (filePath: string): { date: string, hits: number, bytes: number }[] => {
if (!fs.existsSync(filePath)) return [];
const fileContent = fs.readFileSync(filePath, 'utf8');
return fileContent.split('\n').filter(line => line.trim() !== '').map(line => {
const [date, hits, bytes] = line.split(' ');
return { date, hits: Number(hits), bytes: Number(bytes) };
});
};

const writeFileData = (filePath: string, data: { date: string, hits: number, bytes: number }[]) => {
const fileContent = data.map(entry => `${entry.date} ${entry.hits} ${entry.bytes}`).join('\n');
fs.writeFileSync(filePath, fileContent, 'utf8');
};

public getTodayStats(): { hits: number, bytes: number } {
const today = new Date().toISOString().split('T')[0];
const todayData = this.data.find(entry => entry.date === today);
if (todayData) {
return { hits: todayData.hits, bytes: todayData.bytes };
} else {
return { hits: 0, bytes: 0 };
}
export class StatsStorage {
public readonly id: string;
private data: { date: string, hits: number, bytes: number }[];
private filePath: string;
private saveInterval: NodeJS.Timeout;
private dataUpdated: boolean;

constructor(id: string) {
this.id = id;
this.filePath = path.join(Config.getInstance().statsDir, `${this.id}.stats`);
this.data = readFileData(this.filePath);
this.dataUpdated = false;

this.saveInterval = setInterval(() => this.maybeWriteToFile(), 10 * 60 * 1000);
}

public addData({ hits, bytes }: { hits: number, bytes: number }): void {
const today = getLocalDateString();
let todayData = this.data.find(entry => entry.date === today);

if (!todayData) {
todayData = { date: today, hits: 0, bytes: 0 };
this.data.push(todayData);
if (this.data.length > 30) this.data.shift(); // 保持只存30天的数据
}

public getLast30DaysStats(): { date: string, hits: number, bytes: number }[] {
const now = new Date();

const dateMap: { [key: string]: { hits: number, bytes: number } } = {};

// 填充最近30天的数据
for (let i = 0; i < 30; i++) {
const date = new Date(now.getTime() - i * 24 * 60 * 60 * 1000).toISOString().split('T')[0];
dateMap[date] = { hits: 0, bytes: 0 };
}

// 更新映射中的数据
for (const entry of this.data) {
dateMap[entry.date] = { hits: entry.hits, bytes: entry.bytes };
}

// 将映射转换为数组，并按日期排序
return Object.keys(dateMap)
.sort()
.map(date => ({
date,
hits: dateMap[date].hits,
bytes: dateMap[date].bytes
}));
todayData.hits += hits;
todayData.bytes += bytes;
this.dataUpdated = true;
}

public getTodayStats(): { hits: number, bytes: number } {
const today = getLocalDateString();
const todayData = this.data.find(entry => entry.date === today);
return todayData ? { hits: todayData.hits, bytes: todayData.bytes } : { hits: 0, bytes: 0 };
}

public getLast30DaysStats(): { date: string, hits: number, bytes: number }[] {
const now = new Date();
const dateMap: { [key: string]: { hits: number, bytes: number } } = {};

// 填充最近30天的数据
for (let i = 0; i < 30; i++) {
const date = getLocalDateString(new Date(now.getTime() - i * 24 * 60 * 60 * 1000));
dateMap[date] = { hits: 0, bytes: 0 };
}

private maybeWriteToFile(): void {
if (this.dataUpdated) {
this.writeToFile();
this.dataUpdated = false; // 重置更新标志
}
// 更新映射中的数据
for (const entry of this.data) {
if (dateMap[entry.date]) {
dateMap[entry.date] = { hits: entry.hits, bytes: entry.bytes };
}
}

private writeToFile(): void {
// 使用 JSON 序列化存储数据
const fileContent = JSON.stringify(this.data);
fs.writeFileSync(this.filePath, fileContent, 'utf8');
}
return Object.keys(dateMap).sort().map(date => ({ date, hits: dateMap[date].hits, bytes: dateMap[date].bytes }));
}

private loadFromFile(): void {
// 直接读取 JSON 数据
const fileContent = fs.readFileSync(this.filePath, 'utf8');
this.data = JSON.parse(fileContent);
private maybeWriteToFile(): void {
if (this.dataUpdated) {
writeFileData(this.filePath, this.data);
this.dataUpdated = false;
}
}

public stopAutoSave(): void {
clearInterval(this.saveInterval);
this.maybeWriteToFile(); // 在停止时立即保存数据（如果有更新）
}
public stopAutoSave(): void {
clearInterval(this.saveInterval);
this.maybeWriteToFile();
}
}
Loading

0 comments on commit 12148ff

Please sign in to comment.