urlooker/modules/web/model/strategy.go

package model

import (
	"github.com/710leo/urlooker/dataobj"
	. "github.com/710leo/urlooker/modules/web/store"
)

type Strategy dataobj.Strategy

func GetAllStrategyCount(mine int, query, username string) (int64, error) {
	if mine == 1 {
		if query != "" {
			return Orm.Where("url LIKE ? AND creator = ?", "%"+query+"%", username).Count(new(Strategy))
		} else {
			num, err := Orm.Where("creator = ?", username).Count(new(Strategy))
			return num, err
		}
	} else {
		if query != "" {
			return Orm.Where("url LIKE ?", "%"+query+"%").Count(new(Strategy))
		} else {
			num, err := Orm.Count(new(Strategy))
			return num, err
		}
	}

}

func GetAllStrategy(mine, limit, offset int, query, username string) ([]*Strategy, error) {
	items := make([]*Strategy, 0)

	var err error
	if mine == 1 {
		if query != "" {
			err = Orm.Where("url LIKE ? AND creator = ?", "%"+query+"%", username).Limit(limit, offset).OrderBy("id").Find(&items)
		} else {
			err = Orm.Where("creator = ?", username).Limit(limit, offset).OrderBy("id").Find(&items)
		}
	} else {
		if query != "" {
			err = Orm.Where("url LIKE ?", "%"+query+"%").Limit(limit, offset).OrderBy("id").Find(&items)
		} else {
			err = Orm.Limit(limit, offset).Find(&items)
		}
	}
	return items, err
}

func GetAllStrategyByCron() ([]*Strategy, error) {
	strategies := make([]*Strategy, 0)
	err := Orm.Where("enable = 1").Find(&strategies)
	return strategies, err
}

func GetStrategyById(sid int64) (*Strategy, error) {
	strategy := new(Strategy)
	_, err := Orm.Where("id=?", sid).Get(strategy)

	return strategy, err
}

func (this *Strategy) Add() (int64, error) {
	_, err := Orm.Insert(this)
	return this.Id, err
}

func (this *Strategy) Update() error {
	_, err := Orm.Where("id=?", this.Id).Cols("times", "max_step", "expect_code", "timeout", "url", "idc", "enable", "ip", "keywords", "note", "data", "tag", "endpoint", "header", "post_data", "method", "teams").Update(this)
	return err
}

func (this *Strategy) Delete() error {
	_, err := Orm.Where("id=?", this.Id).Delete(new(Strategy))
	return err
}