gosora/common/promotions.go

package common

import (
    "database/sql"
    //"log"
    "time"

    qgen "github.com/Azareal/Gosora/query_gen"
)

var GroupPromotions GroupPromotionStore

type GroupPromotion struct {
    ID     int
    From   int
    To     int
    TwoWay bool

    Level         int
    Posts         int
    MinTime       int
    RegisteredFor int
}

type GroupPromotionStore interface {
    GetByGroup(gid int) (gps []*GroupPromotion, err error)
    Get(id int) (*GroupPromotion, error)
    PromoteIfEligible(u *User, level, posts int, registeredAt time.Time) error
    Delete(id int) error
    Create(from, to int, twoWay bool, level, posts, registeredFor int) (int, error)
}

type DefaultGroupPromotionStore struct {
    getByGroup *sql.Stmt
    get        *sql.Stmt
    delete     *sql.Stmt
    create     *sql.Stmt

    getByUser     *sql.Stmt
    getByUserMins *sql.Stmt
    updateUser    *sql.Stmt
    updateGeneric *sql.Stmt
}

func NewDefaultGroupPromotionStore(acc *qgen.Accumulator) (*DefaultGroupPromotionStore, error) {
    ugp := "users_groups_promotions"
    prs := &DefaultGroupPromotionStore{
        getByGroup: acc.Select(ugp).Columns("pid, from_gid, to_gid, two_way, level, posts, minTime, registeredFor").Where("from_gid=? OR to_gid=?").Prepare(),
        get:        acc.Select(ugp).Columns("from_gid, to_gid, two_way, level, posts, minTime, registeredFor").Where("pid=?").Prepare(),
        delete:     acc.Delete(ugp).Where("pid=?").Prepare(),
        create:     acc.Insert(ugp).Columns("from_gid, to_gid, two_way, level, posts, minTime, registeredFor").Fields("?,?,?,?,?,?,?").Prepare(),

        getByUserMins: acc.Select(ugp).Columns("pid, to_gid, two_way, level, posts, minTime, registeredFor").Where("from_gid=? AND level<=? AND posts<=? AND registeredFor<=?").Orderby("level DESC").Limit("1").Prepare(),
        getByUser:     acc.Select(ugp).Columns("pid, to_gid, two_way, level, posts, minTime, registeredFor").Where("from_gid=? AND level<=? AND posts<=?").Orderby("level DESC").Limit("1").Prepare(),
        updateUser:    acc.Update("users").Set("group=?").Where("group=? AND uid=?").Prepare(),
        updateGeneric: acc.Update("users").Set("group=?").Where("group=? AND level>=? AND posts>=?").Prepare(),
    }
    Tasks.FifteenMin.Add(prs.Tick)
    return prs, acc.FirstError()
}

func (s *DefaultGroupPromotionStore) Tick() error {
    return nil
}

func (s *DefaultGroupPromotionStore) GetByGroup(gid int) (gps []*GroupPromotion, err error) {
    rows, err := s.getByGroup.Query(gid, gid)
    if err != nil {
        return nil, err
    }
    defer rows.Close()

    for rows.Next() {
        g := &GroupPromotion{}
        err := rows.Scan(&g.ID, &g.From, &g.To, &g.TwoWay, &g.Level, &g.Posts, &g.MinTime, &g.RegisteredFor)
        if err != nil {
            return nil, err
        }
        gps = append(gps, g)
    }
    return gps, rows.Err()
}

// TODO: Cache the group promotions to avoid hitting the database as much
func (s *DefaultGroupPromotionStore) Get(id int) (*GroupPromotion, error) {
    /*g, err := s.cache.Get(id)
    if err == nil {
        return u, nil
    }*/

    g := &GroupPromotion{ID: id}
    err := s.get.QueryRow(id).Scan(&g.From, &g.To, &g.TwoWay, &g.Level, &g.Posts, &g.MinTime, &g.RegisteredFor)
    if err == nil {
        //s.cache.Set(u)
    }
    return g, err
}

// TODO: Optimise this to avoid the query
func (s *DefaultGroupPromotionStore) PromoteIfEligible(u *User, level, posts int, registeredAt time.Time) error {
    mins := time.Since(registeredAt).Minutes()
    g := &GroupPromotion{From: u.Group}
    //log.Printf("pre getByUserMins: %+v\n", u)
    err := s.getByUserMins.QueryRow(u.Group, level, posts, mins).Scan(&g.ID, &g.To, &g.TwoWay, &g.Level, &g.Posts, &g.MinTime, &g.RegisteredFor)
    if err == sql.ErrNoRows {
        //log.Print("no matches found")
        return nil
    } else if err != nil {
        return err
    }
    //log.Printf("g: %+v\n", g)
    if g.RegisteredFor == 0 {
        _, err = s.updateGeneric.Exec(g.To, g.From, g.Level, g.Posts)
    } else {
        _, err = s.updateUser.Exec(g.To, g.From, u.ID)
    }
    return err
}

func (s *DefaultGroupPromotionStore) Delete(id int) error {
    _, err := s.delete.Exec(id)
    return err
}

func (s *DefaultGroupPromotionStore) Create(from, to int, twoWay bool, level, posts, registeredFor int) (int, error) {
    res, err := s.create.Exec(from, to, twoWay, level, posts, 0, registeredFor)
    if err != nil {
        return 0, err
    }
    lastID, err := res.LastInsertId()
    return int(lastID), err
}