109 lines
3.5 KiB
TypeScript
109 lines
3.5 KiB
TypeScript
import {MessageInsert, messagesTable} from "./schema";
|
|
import {DatabaseManager} from "./database-manager";
|
|
import {StoredMessage} from "../model/stored-message";
|
|
import {and, eq} from "drizzle-orm";
|
|
import {inArray} from "drizzle-orm/sql/expressions/conditions";
|
|
import {Dao} from "../base/dao";
|
|
import {buildExcludedSet} from "../util/utils";
|
|
|
|
export class MessageDao extends Dao<StoredMessage> {
|
|
|
|
private tag: string = "MessageDao";
|
|
|
|
override async getAll(): Promise<StoredMessage[]> {
|
|
const then = Date.now();
|
|
|
|
const messages = await DatabaseManager.db.select().from(messagesTable);
|
|
|
|
const now = Date.now();
|
|
const diff = now - then;
|
|
console.log(`${this.tag}: getAll()`, `took ${diff}ms; size: ${messages.length}`);
|
|
|
|
return this.mapFrom(messages);
|
|
}
|
|
|
|
override async getById(params: { chatId: number, id: number }): Promise<StoredMessage | null> {
|
|
const then = Date.now();
|
|
|
|
const messages =
|
|
await DatabaseManager.db.select()
|
|
.from(messagesTable)
|
|
.where(
|
|
and(
|
|
eq(messagesTable.chatId, params.chatId),
|
|
eq(messagesTable.id, params.id)
|
|
)
|
|
);
|
|
|
|
const now = Date.now();
|
|
const diff = now - then;
|
|
console.log(`${this.tag}: getById(${params.chatId}, ${params.id})`, `took ${diff}ms; size: ${messages.length}`);
|
|
|
|
const m = messages[0];
|
|
if (!m) return null;
|
|
return this.mapFrom([m])[0];
|
|
}
|
|
|
|
override async getByIds(params: { chatId: number, ids: number[] }): Promise<StoredMessage[]> {
|
|
const then = Date.now();
|
|
|
|
const messages =
|
|
await DatabaseManager.db.select()
|
|
.from(messagesTable)
|
|
.where(
|
|
and(
|
|
eq(messagesTable.chatId, params.chatId),
|
|
inArray(messagesTable.id, params.ids)
|
|
)
|
|
);
|
|
|
|
const now = Date.now();
|
|
const diff = now - then;
|
|
console.log(`${this.tag}: getByIds(${params.chatId}, ${params.ids})`, `took ${diff}ms; size: ${messages.length}`);
|
|
|
|
return this.mapFrom(messages);
|
|
}
|
|
|
|
async insert(values: MessageInsert[]): Promise<true> {
|
|
const then = Date.now();
|
|
const r = await DatabaseManager.db
|
|
.insert(messagesTable)
|
|
.values(values)
|
|
.onConflictDoUpdate({
|
|
target: messagesTable.id,
|
|
set: buildExcludedSet(messagesTable, ["id"])
|
|
});
|
|
|
|
const now = Date.now();
|
|
const diff = now - then;
|
|
console.log(`${this.tag}: insert(size: ${values.length})`, `took ${diff}ms'; inserted: ${r.rowsAffected}`);
|
|
return true;
|
|
}
|
|
|
|
mapStoredTo(messages: StoredMessage[]): MessageInsert[] {
|
|
return messages.map(msg => {
|
|
return {
|
|
chatId: msg.chatId,
|
|
id: msg.id,
|
|
replyToMessageId: msg.replyToMessageId,
|
|
fromId: msg.fromId,
|
|
text: msg.text,
|
|
date: msg.date,
|
|
};
|
|
});
|
|
}
|
|
|
|
mapFrom(messages: MessageInsert[]): StoredMessage[] {
|
|
return messages.map(m => {
|
|
return {
|
|
chatId: m.chatId,
|
|
id: m.id,
|
|
replyToMessageId: m.replyToMessageId,
|
|
fromId: m.fromId,
|
|
text: m.text,
|
|
date: m.date,
|
|
photoMaxSizeFilePath: m.photoMaxSizeFilePath
|
|
};
|
|
});
|
|
}
|
|
} |