This commit is contained in:
syuilo 2018-08-18 20:22:56 +09:00
parent 3231f12b93
commit 69e2576387
2 changed files with 231 additions and 0 deletions

88
src/models/chart.ts Normal file
View File

@ -0,0 +1,88 @@
import * as mongo from 'mongodb';
import db from '../db/mongodb';
const Chart = db.get<IChart>('chart');
Chart.createIndex('date', { unique: true });
export default Chart;
export interface IChart {
_id: mongo.ObjectID;
date: Date;
users: {
local: {
/**
*
*/
total: number;
/**
*
*/
diff: number;
};
remote: {
/**
*
*/
total: number;
/**
*
*/
diff: number;
};
};
notes: {
local: {
/**
* 稿
*/
total: number;
diffs: {
/**
* 稿
*/
normal: number;
/**
* 稿
*/
reply: number;
/**
* Renoteの投稿数の前日比
*/
renote: number;
};
};
remote: {
/**
* 稿
*/
total: number;
diffs: {
/**
* 稿
*/
normal: number;
/**
* 稿
*/
reply: number;
/**
* Renoteの投稿数の前日比
*/
renote: number;
};
};
};
}

View File

@ -0,0 +1,143 @@
import { INote } from '../models/note';
import Chart, { IChart } from '../models/chart';
import { isLocalUser } from '../models/user';
async function getTodayStats(): Promise<IChart> {
const now = new Date();
const y = now.getFullYear();
const m = now.getMonth();
const d = now.getDate();
const today = new Date(y, m, d);
// 今日の統計
const todayStats = await Chart.findOne({
date: today
});
// 日付が変わってから、初めてのチャート更新なら
if (todayStats == null) {
// 最も最近の統計を持ってくる
// * 昨日何もチャートを更新するような出来事がなかった場合は、
// 統計がそもそも作られずドキュメントが存在しないということがあり得るため、
// 「昨日の」と決め打ちせずに「もっとも最近の」とします
const mostRecentStats = await Chart.findOne({}, {
sort: {
createdAt: -1
}
});
// 統計が存在しなかったら
// * Misskeyインスタンスを建てて初めてのチャート更新時など
if (mostRecentStats == null) {
// 空の統計を作成
const stats = await Chart.insert({
date: today,
users: {
local: {
total: 0,
diff: 0
},
remote: {
total: 0,
diff: 0
}
},
notes: {
local: {
total: 0,
diffs: {
normal: 0,
reply: 0,
renote: 0
}
},
remote: {
total: 0,
diffs: {
normal: 0,
reply: 0,
renote: 0
}
}
}
});
return stats;
} else {
// 今日の統計を初期挿入
const stats = await Chart.insert({
date: today,
users: {
local: {
total: mostRecentStats.users.local.total,
diff: 0
},
remote: {
total: mostRecentStats.users.remote.total,
diff: 0
}
},
notes: {
local: {
total: mostRecentStats.notes.local.total,
diffs: {
normal: 0,
reply: 0,
renote: 0
}
},
remote: {
total: mostRecentStats.notes.remote.total,
diffs: {
normal: 0,
reply: 0,
renote: 0
}
}
}
});
return stats;
}
} else {
return todayStats;
}
}
async function update(inc: any) {
const stats = await getTodayStats();
await Chart.findOneAndUpdate({
_id: stats._id
}, {
$inc: inc
});
}
export async function incNote(note: INote) {
const inc = {} as any;
if (isLocalUser(note._user)) {
inc['notes.local.total'] = 1;
if (note.replyId != null) {
inc['notes.local.diffs.reply'] = 1;
} else if (note.renoteId != null) {
inc['notes.local.diffs.renote'] = 1;
} else {
inc['notes.local.diffs.normal'] = 1;
}
} else {
inc['notes.remote.total'] = 1;
if (note.replyId != null) {
inc['notes.remote.diffs.reply'] = 1;
} else if (note.renoteId != null) {
inc['notes.remote.diffs.renote'] = 1;
} else {
inc['notes.remote.diffs.normal'] = 1;
}
}
await update(inc);
}