/
server.go
45 lines (38 loc) · 1.12 KB
/
server.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
package db
// CreateServerIfNotExists returns true if the server exists
func (db *DB) CreateServerIfNotExists(guildID string) (exists bool, err error) {
ctx, cancel := db.Context()
defer cancel()
err = db.QueryRow(ctx, "select exists (select from public.servers where id = $1)", guildID).Scan(&exists)
if err != nil {
return exists, err
}
if !exists {
ctx, cancel := db.Context()
defer cancel()
Debug("Creating server %v", guildID)
commandTag, err := db.Exec(ctx, "insert into public.servers (id, prefixes) values ($1, $2)", guildID, db.Config.Bot.Prefixes)
if err != nil {
return exists, err
}
if commandTag.RowsAffected() != 1 {
return exists, ErrorNoRowsAffected
}
return exists, err
}
return exists, nil
}
// DeleteServer deletes a server's database entry
func (db *DB) DeleteServer(guildID string) (err error) {
ctx, cancel := db.Context()
defer cancel()
Debug("Deleting server %v", guildID)
commandTag, err := db.Exec(ctx, "delete from public.servers where id = $1", guildID)
if err != nil {
return err
}
if commandTag.RowsAffected() != 1 {
return ErrorNoRowsAffected
}
return err
}