gosora/extend/guilds/lib/guild_store.go
2020-02-05 12:48:35 +10:00

48 lines
1.4 KiB
Go

package guilds
import (
"database/sql"
qgen "github.com/Azareal/Gosora/query_gen"
)
var Gstore GuildStore
type GuildStore interface {
Get(id int) (g *Guild, err error)
Create(name, desc string, active bool, privacy, uid, fid int) (int, error)
}
type SQLGuildStore struct {
get *sql.Stmt
create *sql.Stmt
}
func NewSQLGuildStore() (*SQLGuildStore, error) {
acc := qgen.NewAcc()
return &SQLGuildStore{
get: acc.Select("guilds").Columns("name, desc, active, privacy, joinable, owner, memberCount, mainForum, backdrop, createdAt, lastUpdateTime").Where("guildID=?").Prepare(),
create: acc.Insert("guilds").Columns("name, desc, active, privacy, joinable, owner, memberCount, mainForum, backdrop, createdAt, lastUpdateTime").Fields("?,?,?,?,1,?,1,?,'',UTC_TIMESTAMP(),UTC_TIMESTAMP()").Prepare(),
}, acc.FirstError()
}
func (s *SQLGuildStore) Close() {
_ = s.get.Close()
_ = s.create.Close()
}
func (s *SQLGuildStore) Get(id int) (g *Guild, err error) {
g = &Guild{ID: id}
err = s.get.QueryRow(id).Scan(&g.Name, &g.Desc, &g.Active, &g.Privacy, &g.Joinable, &g.Owner, &g.MemberCount, &g.MainForumID, &g.Backdrop, &g.CreatedAt, &g.LastUpdateTime)
return g, err
}
func (s *SQLGuildStore) Create(name, desc string, active bool, privacy, uid, fid int) (int, error) {
res, err := s.create.Exec(name, desc, active, privacy, uid, fid)
if err != nil {
return 0, err
}
lastID, err := res.LastInsertId()
return int(lastID), err
}