gosora/common/activity_stream_matches.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)
}