bbgo/pkg/service/reward.go

200 lines
5.2 KiB
Go

package service
import (
"context"
"fmt"
"strconv"
"strings"
"time"
sq "github.com/Masterminds/squirrel"
"github.com/jmoiron/sqlx"
exchange2 "git.qtrade.icu/lychiyu/bbgo/pkg/exchange"
"git.qtrade.icu/lychiyu/bbgo/pkg/exchange/batch"
"git.qtrade.icu/lychiyu/bbgo/pkg/fixedpoint"
"git.qtrade.icu/lychiyu/bbgo/pkg/types"
)
// RewardService collects the reward records from the exchange,
// currently it's only available for MAX exchange.
// TODO: add summary query for calculating the reward amounts
// CREATE VIEW reward_summary_by_years AS SELECT YEAR(created_at) as year, reward_type, currency, SUM(quantity) FROM rewards WHERE reward_type != 'airdrop' GROUP BY YEAR(created_at), reward_type, currency ORDER BY year DESC;
type RewardService struct {
DB *sqlx.DB
}
func (s *RewardService) Sync(ctx context.Context, exchange types.Exchange, startTime time.Time) error {
api, ok := exchange.(types.ExchangeRewardService)
if !ok {
return ErrExchangeRewardServiceNotImplemented
}
isMargin, isFutures, _, _ := exchange2.GetSessionAttributes(exchange)
if isMargin || isFutures {
return nil
}
tasks := []SyncTask{
{
Type: types.Reward{},
Select: SelectLastRewards(exchange.Name(), 100),
BatchQuery: func(ctx context.Context, startTime, endTime time.Time) (interface{}, chan error) {
query := &batch.RewardBatchQuery{
Service: api,
}
return query.Query(ctx, startTime, endTime)
},
Time: func(obj interface{}) time.Time {
return obj.(types.Reward).CreatedAt.Time()
},
ID: func(obj interface{}) string {
reward := obj.(types.Reward)
return string(reward.Type) + "_" + reward.UUID
},
LogInsert: true,
},
}
for _, sel := range tasks {
if err := sel.execute(ctx, s.DB, startTime); err != nil {
return err
}
}
return nil
}
type CurrencyPositionMap map[string]fixedpoint.Value
func (s *RewardService) AggregateUnspentCurrencyPosition(ctx context.Context, ex types.ExchangeName, since time.Time) (CurrencyPositionMap, error) {
m := make(CurrencyPositionMap)
rewards, err := s.QueryUnspentSince(ctx, ex, since)
if err != nil {
return nil, err
}
for _, reward := range rewards {
m[reward.Currency] = m[reward.Currency].Add(reward.Quantity)
}
return m, nil
}
func (s *RewardService) QueryUnspentSince(ctx context.Context, ex types.ExchangeName, since time.Time, rewardTypes ...types.RewardType) ([]types.Reward, error) {
sql := "SELECT * FROM rewards WHERE created_at >= :since AND exchange = :exchange AND spent IS FALSE "
if len(rewardTypes) == 0 {
sql += " AND `reward_type` NOT IN ('airdrop') "
} else {
var args []string
for _, n := range rewardTypes {
args = append(args, strconv.Quote(string(n)))
}
sql += " AND `reward_type` IN (" + strings.Join(args, ", ") + ") "
}
sql += " ORDER BY created_at ASC"
rows, err := s.DB.NamedQueryContext(ctx, sql, map[string]interface{}{
"exchange": ex,
"since": since,
})
if err != nil {
return nil, err
}
defer rows.Close()
return s.scanRows(rows)
}
func (s *RewardService) QueryUnspent(ctx context.Context, ex types.ExchangeName, rewardTypes ...types.RewardType) ([]types.Reward, error) {
sql := "SELECT * FROM rewards WHERE exchange = :exchange AND spent IS FALSE "
if len(rewardTypes) == 0 {
sql += " AND `reward_type` NOT IN ('airdrop') "
} else {
var args []string
for _, n := range rewardTypes {
args = append(args, strconv.Quote(string(n)))
}
sql += " AND `reward_type` IN (" + strings.Join(args, ", ") + ") "
}
sql += " ORDER BY created_at ASC"
rows, err := s.DB.NamedQueryContext(ctx, sql, map[string]interface{}{
"exchange": ex,
})
if err != nil {
return nil, err
}
defer rows.Close()
return s.scanRows(rows)
}
func (s *RewardService) MarkCurrencyAsSpent(ctx context.Context, currency string) error {
result, err := s.DB.NamedExecContext(ctx, "UPDATE `rewards` SET `spent` = TRUE WHERE `currency` = :currency AND `spent` IS FALSE", map[string]interface{}{
"currency": currency,
})
if err != nil {
return err
}
_, err = result.RowsAffected()
return err
}
func (s *RewardService) MarkAsSpent(ctx context.Context, uuid string) error {
result, err := s.DB.NamedExecContext(ctx, "UPDATE `rewards` SET `spent` = TRUE WHERE `uuid` = :uuid", map[string]interface{}{
"uuid": uuid,
})
if err != nil {
return err
}
cnt, err := result.RowsAffected()
if err != nil {
return err
}
if cnt == 0 {
return fmt.Errorf("reward uuid:%s not found", uuid)
}
return nil
}
func (s *RewardService) scanRows(rows *sqlx.Rows) (rewards []types.Reward, err error) {
for rows.Next() {
var reward types.Reward
if err := rows.StructScan(&reward); err != nil {
return rewards, err
}
rewards = append(rewards, reward)
}
return rewards, rows.Err()
}
func (s *RewardService) Insert(reward types.Reward) error {
_, err := s.DB.NamedExec(`
INSERT INTO rewards (exchange, uuid, reward_type, currency, quantity, state, note, created_at)
VALUES (:exchange, :uuid, :reward_type, :currency, :quantity, :state, :note, :created_at)`,
reward)
return err
}
func SelectLastRewards(ex types.ExchangeName, limit uint64) sq.SelectBuilder {
return sq.Select("*").
From("rewards").
Where(sq.And{
sq.Eq{"exchange": ex},
}).
OrderBy("created_at DESC").
Limit(limit)
}