summaryrefslogtreecommitdiff
path: root/golang/pkg/database
diff options
context:
space:
mode:
Diffstat (limited to 'golang/pkg/database')
-rw-r--r--golang/pkg/database/tokens.go11
1 files changed, 4 insertions, 7 deletions
diff --git a/golang/pkg/database/tokens.go b/golang/pkg/database/tokens.go
index 16bda2b..0787b5c 100644
--- a/golang/pkg/database/tokens.go
+++ b/golang/pkg/database/tokens.go
@@ -1,14 +1,11 @@
package database
-func (db DB) AddToken(token string) error {
+func (db *DB) AddToken(token string) error {
_, err := db.db.ExecContext(db.ctx, `INSERT INTO tokens(data) VALUES (?);`, token)
return err
}
-func (db DB) GetToken() (string, error) {
- var token string
- if err := db.db.QueryRowContext(db.ctx, `SELECT data FROM tokens;`).Scan(&token); err != nil {
- return "", err
- }
- return token, nil
+func (db *DB) GetToken() (token string, err error) {
+ err = db.db.QueryRowContext(db.ctx, `SELECT data FROM tokens;`).Scan(&token)
+ return token, err
}