58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
|
package infra
|
||
|
|
|
||
|
|
import (
|
||
|
|
"owl-blogs/app/repository"
|
||
|
|
|
||
|
|
"github.com/jmoiron/sqlx"
|
||
|
|
)
|
||
|
|
|
||
|
|
type sqlFollower struct {
|
||
|
|
Follwer string `db:"follower"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type DefaultFollowerRepo struct {
|
||
|
|
db *sqlx.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewFollowerRepository(db Database) repository.FollowerRepository {
|
||
|
|
sqlxdb := db.Get()
|
||
|
|
|
||
|
|
// Create tables if not exists
|
||
|
|
sqlxdb.MustExec(`
|
||
|
|
CREATE TABLE IF NOT EXISTS followers (
|
||
|
|
follower TEXT PRIMARY KEY
|
||
|
|
);
|
||
|
|
`)
|
||
|
|
|
||
|
|
return &DefaultFollowerRepo{
|
||
|
|
db: sqlxdb,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Add implements repository.FollowerRepository.
|
||
|
|
func (d *DefaultFollowerRepo) Add(follower string) error {
|
||
|
|
_, err := d.db.Exec("INSERT INTO followers (follower) VALUES (?) ON CONFLICT DO NOTHING", follower)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Remove implements repository.FollowerRepository.
|
||
|
|
func (d *DefaultFollowerRepo) Remove(follower string) error {
|
||
|
|
_, err := d.db.Exec("DELETE FROM followers WHERE follower = ?", follower)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// All implements repository.FollowerRepository.
|
||
|
|
func (d *DefaultFollowerRepo) All() ([]string, error) {
|
||
|
|
var followers []sqlFollower
|
||
|
|
err := d.db.Select(&followers, "SELECT * FROM followers")
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
result := []string{}
|
||
|
|
for _, follower := range followers {
|
||
|
|
result = append(result, follower.Follwer)
|
||
|
|
}
|
||
|
|
return result, nil
|
||
|
|
}
|