package common import ( "database/sql" qgen "github.com/Azareal/Gosora/query_gen" ) var ActivityMatches ActivityStreamMatches type ActivityStreamMatches interface { Add(watcher, asid int) error Delete(watcher, asid int) error DeleteAndCountChanged(watcher, asid int) (int, error) CountAsid(asid int) int } type DefaultActivityStreamMatches struct { add *sql.Stmt delete *sql.Stmt countAsid *sql.Stmt } func NewDefaultActivityStreamMatches(acc *qgen.Accumulator) (*DefaultActivityStreamMatches, error) { asm := "activity_stream_matches" return &DefaultActivityStreamMatches{ add: acc.Insert(asm).Columns("watcher,asid").Fields("?,?").Prepare(), delete: acc.Delete(asm).Where("watcher=? AND asid=?").Prepare(), countAsid: acc.Count(asm).Where("asid=?").Prepare(), }, acc.FirstError() } func (s *DefaultActivityStreamMatches) Add(watcher, asid int) error { _, e := s.add.Exec(watcher, asid) return e } func (s *DefaultActivityStreamMatches) Delete(watcher, asid int) error { _, e := s.delete.Exec(watcher, asid) return e } func (s *DefaultActivityStreamMatches) DeleteAndCountChanged(watcher, asid int) (int, error) { res, e := s.delete.Exec(watcher, asid) if e != nil { return 0, e } c64, e := res.RowsAffected() return int(c64), e } func (s *DefaultActivityStreamMatches) CountAsid(asid int) int { return Countf(s.countAsid, asid) }