package common import ( "database/sql" qgen "github.com/Azareal/Gosora/query_gen" ) var Activity ActivityStream type ActivityStream interface { Add(a Alert) (int, error) Get(id int) (Alert, error) Delete(id int) error DeleteByParams(event string, targetID int, targetType string) error DeleteByParamsExtra(event string, targetID int, targetType, extra string) error AidsByParamsExtra(event string, elementID int, elementType, extra string) (aids []int, err error) Count() (count int) } type DefaultActivityStream struct { add *sql.Stmt get *sql.Stmt delete *sql.Stmt deleteByParams *sql.Stmt deleteByParamsExtra *sql.Stmt aidsByParamsExtra *sql.Stmt count *sql.Stmt } func NewDefaultActivityStream(acc *qgen.Accumulator) (*DefaultActivityStream, error) { as := "activity_stream" return &DefaultActivityStream{ add: acc.Insert(as).Columns("actor,targetUser,event,elementType,elementID,createdAt,extra").Fields("?,?,?,?,?,UTC_TIMESTAMP(),?").Prepare(), get: acc.Select(as).Columns("actor,targetUser,event,elementType,elementID,createdAt,extra").Where("asid=?").Prepare(), delete: acc.Delete(as).Where("asid=?").Prepare(), deleteByParams: acc.Delete(as).Where("event=? AND elementID=? AND elementType=?").Prepare(), deleteByParamsExtra: acc.Delete(as).Where("event=? AND elementID=? AND elementType=? AND extra=?").Prepare(), aidsByParamsExtra: acc.Select(as).Columns("asid").Where("event=? AND elementID=? AND elementType=? AND extra=?").Prepare(), count: acc.Count(as).Prepare(), }, acc.FirstError() } func (s *DefaultActivityStream) Add(a Alert) (int, error) { res, err := s.add.Exec(a.ActorID, a.TargetUserID, a.Event, a.ElementType, a.ElementID, a.Extra) if err != nil { return 0, err } lastID, err := res.LastInsertId() return int(lastID), err } func (s *DefaultActivityStream) Get(id int) (Alert, error) { a := Alert{ASID: id} err := s.get.QueryRow(id).Scan(&a.ActorID, &a.TargetUserID, &a.Event, &a.ElementType, &a.ElementID, &a.CreatedAt, &a.Extra) return a, err } func (s *DefaultActivityStream) Delete(id int) error { _, err := s.delete.Exec(id) return err } func (s *DefaultActivityStream) DeleteByParams(event string, elementID int, elementType string) error { _, err := s.deleteByParams.Exec(event, elementID, elementType) return err } func (s *DefaultActivityStream) DeleteByParamsExtra(event string, elementID int, elementType, extra string) error { _, err := s.deleteByParamsExtra.Exec(event, elementID, elementType, extra) return err } func (s *DefaultActivityStream) AidsByParamsExtra(event string, elementID int, elementType, extra string) (aids []int, err error) { rows, err := s.aidsByParamsExtra.Query(event, elementID, elementType, extra) if err != nil { return nil, err } defer rows.Close() for rows.Next() { var aid int if err := rows.Scan(&aid); err != nil { return nil, err } aids = append(aids, aid) } return aids, rows.Err() } // TODO: Write a test for this // Count returns the total number of activity stream items func (s *DefaultActivityStream) Count() (count int) { err := s.count.QueryRow().Scan(&count) if err != nil { LogError(err) } return count }