diff options
Diffstat (limited to '')
-rw-r--r-- | golang/pkg/database/tokens.go | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/golang/pkg/database/tokens.go b/golang/pkg/database/tokens.go new file mode 100644 index 0000000..16bda2b --- /dev/null +++ b/golang/pkg/database/tokens.go @@ -0,0 +1,14 @@ +package database + +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 +} |