104 lines
2.5 KiB
Go
104 lines
2.5 KiB
Go
package common
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
qgen "github.com/Azareal/Gosora/query_gen"
|
|
)
|
|
|
|
var Likes LikeStore
|
|
|
|
type LikeStore interface {
|
|
BulkExists(ids []int, sentBy int, targetType string) ([]int, error)
|
|
BulkExistsFunc(ids []int, sentBy int, targetType string, f func(int) error) error
|
|
Delete(targetID int, targetType string) error
|
|
Count() (count int)
|
|
}
|
|
|
|
type DefaultLikeStore struct {
|
|
count *sql.Stmt
|
|
delete *sql.Stmt
|
|
singleExists *sql.Stmt
|
|
}
|
|
|
|
func NewDefaultLikeStore(acc *qgen.Accumulator) (*DefaultLikeStore, error) {
|
|
return &DefaultLikeStore{
|
|
count: acc.Count("likes").Prepare(),
|
|
delete: acc.Delete("likes").Where("targetItem=? AND targetType=?").Prepare(),
|
|
singleExists: acc.Select("likes").Columns("targetItem").Where("sentBy=? AND targetType=? AND targetItem=?").Prepare(),
|
|
}, acc.FirstError()
|
|
}
|
|
|
|
// TODO: Write a test for this
|
|
func (s *DefaultLikeStore) BulkExists(ids []int, sentBy int, targetType string) (eids []int, e error) {
|
|
if len(ids) == 0 {
|
|
return nil, nil
|
|
}
|
|
var rows *sql.Rows
|
|
if len(ids) == 1 {
|
|
rows, e = s.singleExists.Query(sentBy, targetType, ids[0])
|
|
} else {
|
|
rows, e = qgen.NewAcc().Select("likes").Columns("targetItem").Where("sentBy=? AND targetType=?").In("targetItem", ids).Query(sentBy, targetType)
|
|
}
|
|
if e == sql.ErrNoRows {
|
|
return nil, nil
|
|
} else if e != nil {
|
|
return nil, e
|
|
}
|
|
defer rows.Close()
|
|
|
|
var id int
|
|
for rows.Next() {
|
|
if e := rows.Scan(&id); e != nil {
|
|
return nil, e
|
|
}
|
|
eids = append(eids, id)
|
|
}
|
|
return eids, rows.Err()
|
|
}
|
|
|
|
// TODO: Write a test for this
|
|
func (s *DefaultLikeStore) BulkExistsFunc(ids []int, sentBy int, targetType string, f func(id int) error) (e error) {
|
|
if len(ids) == 0 {
|
|
return nil
|
|
}
|
|
var rows *sql.Rows
|
|
if len(ids) == 1 {
|
|
rows, e = s.singleExists.Query(sentBy, targetType, ids[0])
|
|
} else {
|
|
rows, e = qgen.NewAcc().Select("likes").Columns("targetItem").Where("sentBy=? AND targetType=?").In("targetItem", ids).Query(sentBy, targetType)
|
|
}
|
|
if e == sql.ErrNoRows {
|
|
return nil
|
|
} else if e != nil {
|
|
return e
|
|
}
|
|
defer rows.Close()
|
|
|
|
var id int
|
|
for rows.Next() {
|
|
if e := rows.Scan(&id); e != nil {
|
|
return e
|
|
}
|
|
if e := f(id); e != nil {
|
|
return e
|
|
}
|
|
}
|
|
return rows.Err()
|
|
}
|
|
|
|
func (s *DefaultLikeStore) Delete(targetID int, targetType string) error {
|
|
_, err := s.delete.Exec(targetID, targetType)
|
|
return err
|
|
}
|
|
|
|
// TODO: Write a test for this
|
|
// Count returns the total number of likes globally
|
|
func (s *DefaultLikeStore) Count() (count int) {
|
|
e := s.count.QueryRow().Scan(&count)
|
|
if e != nil {
|
|
LogError(e)
|
|
}
|
|
return count
|
|
}
|