embee/woomy-v2
1
1
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

move some logic into db file

Browse source
This commit is contained in:
Emily 2020-10-21 15:01:36 +11:00
parent 7b9f5bfa33
commit e2fba99044
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
bot/util/database.js
View file

@ -15,18 +15,24 @@ class Database {
async getGuild (id) { async getGuild (id) {
const res = await this.pool.query('SELECT * FROM guilds WHERE guild_id = $1;', [id]); const res = await this.pool.query('SELECT * FROM guilds WHERE guild_id = $1;', [id]);
return res.rows[0]; let data = res.rows[0];
if (!data) data = await this.createGuild(id);
return data;
} }
async getMember (guild_id, user_id) { async getMember (guild_id, user_id) {
const key = guild_id + ':' + user_id; const key = guild_id + ':' + user_id;
const res = await this.pool.query('SELECT * FROM members WHERE member_id = $1;', [key]); const res = await this.pool.query('SELECT * FROM members WHERE member_id = $1;', [key]);
return res.rows[0]; let data = res.rows[0];
if (!data) data = await this.createMember(guild_id, user_id);
return data;
} }
async getUser (id) { async getUser (id) {
const res = await this.pool.query('SELECT * FROM users WHERE user_id = $1;', [id]); const res = await this.pool.query('SELECT * FROM users WHERE user_id = $1;', [id]);
return res.rows[0]; let data = res.rows[0];
if (!data) data = await this.createUser(id);
return data;
} }
async getGuildField (id, column) { async getGuildField (id, column) {