55 lines
1.4 KiB
Go
55 lines
1.4 KiB
Go
|
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)
|
||
|
|
}
|