add database models for twitch channel

This commit is contained in:
lyx0 2022-08-08 23:58:38 +02:00
parent 7ab1fe8e73
commit 54d1b93b2d
2 changed files with 65 additions and 0 deletions

58
internal/data/channel.go Normal file
View file

@ -0,0 +1,58 @@
package data
import (
"database/sql"
"errors"
"time"
)
type Channel struct {
ID int `json:"id"`
JoinedAt time.Time `json:"-"`
Login string `json:"login"`
TwitchID string `json:"twitchid"`
Announce bool `json:"announce"`
}
type ChannelModel struct {
DB *sql.DB
}
func (c ChannelModel) Insert(channel *Channel) error {
query := `
INSERT INTO channels (login, twitchid, announce)
VALUES ($1, $2, $3)
RETURNING id, created_at`
args := []interface{}{channel.Login, channel.TwitchID, channel.Announce}
return c.DB.QueryRow(query, args...).Scan(&channel.ID, &channel.JoinedAt)
}
func (c ChannelModel) Get(login string) (*Channel, error) {
query := `
SELECT id, joined_at, login, twitchid, announce
FROM channels
WHERE login = $1`
var channel Channel
err := c.DB.QueryRow(query, login).Scan(
&channel.ID,
&channel.JoinedAt,
&channel.Login,
&channel.TwitchID,
&channel.Announce,
)
if err != nil {
switch {
case errors.Is(err, sql.ErrNoRows):
return nil, ErrRecordNotFound
default:
return nil, err
}
}
return &channel, nil
}

7
internal/data/models.go Normal file
View file

@ -0,0 +1,7 @@
package data
import "errors"
var (
ErrRecordNotFound = errors.New("record not found")
)