bbgo_origin/pkg/service/reward.go

156 lines
3.8 KiB
Go
Raw Normal View History

2021-02-23 01:37:06 +00:00
package service
import (
"context"
"fmt"
"strconv"
"strings"
2021-02-23 01:37:06 +00:00
"time"
"github.com/jmoiron/sqlx"
"github.com/c9s/bbgo/pkg/fixedpoint"
2021-02-23 01:37:06 +00:00
"github.com/c9s/bbgo/pkg/types"
)
type RewardService struct {
DB *sqlx.DB
}
func (s *RewardService) QueryLast(ex types.ExchangeName, limit int) ([]types.Reward, error) {
sql := "SELECT * FROM `rewards` WHERE `exchange` = :exchange ORDER BY `created_at` DESC LIMIT :limit"
rows, err := s.DB.NamedQuery(sql, map[string]interface{}{
"exchange": ex,
"limit": limit,
})
if err != nil {
return nil, err
}
defer rows.Close()
return s.scanRows(rows)
}
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,
2021-02-23 01:37:06 +00:00
})
2021-02-23 01:37:06 +00:00
if err != nil {
return nil, err
}
defer rows.Close()
return s.scanRows(rows)
}
2021-02-23 01:37:06 +00:00
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()
2021-02-23 01:37:06 +00:00
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) {
2021-02-23 01:37:06 +00:00
for rows.Next() {
var reward types.Reward
if err := rows.StructScan(&reward); err != nil {
return rewards, err
2021-02-23 01:37:06 +00:00
}
rewards = append(rewards, reward)
2021-02-23 01:37:06 +00:00
}
return rewards, rows.Err()
2021-02-23 01:37:06 +00:00
}
func (s *RewardService) Insert(reward types.Reward) error {
2021-02-23 01:37:06 +00:00
_, 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)
2021-02-23 01:37:06 +00:00
return err
}