mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 15:09:42 +00:00
cc05f7e6fd
* portfolio: Fix CryptoID balance issue and assertify tests * portfolio: Expand context usage, enhance tests and a few other minor improvements * portfolio: Further improvements and enhance common.IsValidCryptoAddress * config, portfolio: Use v6.DefaultConfig, switch to context.WithCancel
73 lines
2.6 KiB
Go
73 lines
2.6 KiB
Go
package versions
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/buger/jsonparser"
|
|
)
|
|
|
|
// Version5 implements ConfigVersion
|
|
type Version5 struct{}
|
|
|
|
func init() {
|
|
Manager.registerVersion(5, &Version5{})
|
|
}
|
|
|
|
// defaultConfig contains the stateless V5 representation of orderbookManager
|
|
// Note: Do not be tempted to use a constant for Duration. Whilst defaults are still written to config, we need to manage default upgrades discretely.
|
|
var defaultFuturesTrackingSeekDuration = strconv.FormatInt(int64(time.Hour)*24*365, 10)
|
|
|
|
var defaultConfigV5 = []byte(`{
|
|
"enabled": true,
|
|
"verbose": false,
|
|
"activelyTrackFuturesPositions": true,
|
|
"futuresTrackingSeekDuration": ` + defaultFuturesTrackingSeekDuration + `,
|
|
"cancelOrdersOnShutdown": false,
|
|
"respectOrderHistoryLimits": true
|
|
}`)
|
|
|
|
// UpgradeConfig handles upgrading config for OrderManager:
|
|
// * Sets OrderManager config to defaults if it doesn't exist or enabled is null
|
|
// * Sets respectOrderHistoryLimits to true if it doesn't exist or is null
|
|
// * Sets futuresTrackingSeekDuration to positive if it's negative
|
|
func (v *Version5) UpgradeConfig(_ context.Context, e []byte) ([]byte, error) {
|
|
_, valueType, _, err := jsonparser.Get(e, "orderManager", "enabled")
|
|
switch {
|
|
case errors.Is(err, jsonparser.KeyPathNotFoundError), valueType == jsonparser.Null:
|
|
return jsonparser.Set(e, defaultConfigV5, "orderManager")
|
|
case err != nil:
|
|
return e, err
|
|
}
|
|
|
|
_, valueType, _, err = jsonparser.Get(e, "orderManager", "respectOrderHistoryLimits")
|
|
if errors.Is(err, jsonparser.KeyPathNotFoundError) || valueType == jsonparser.Null {
|
|
if e, err = jsonparser.Set(e, []byte(`true`), "orderManager", "respectOrderHistoryLimits"); err != nil {
|
|
return e, err
|
|
}
|
|
}
|
|
|
|
if i, err := jsonparser.GetInt(e, "orderManager", "futuresTrackingSeekDuration"); err != nil {
|
|
if e, err = jsonparser.Set(e, []byte(defaultFuturesTrackingSeekDuration), "orderManager", "futuresTrackingSeekDuration"); err != nil {
|
|
return e, err
|
|
}
|
|
} else if i < 0 {
|
|
if e, err = jsonparser.Set(e, []byte(strconv.FormatInt(-i, 10)), "orderManager", "futuresTrackingSeekDuration"); err != nil {
|
|
return e, err
|
|
}
|
|
}
|
|
return e, nil
|
|
}
|
|
|
|
// DowngradeConfig just reverses the futuresTrackingSeekDuration to negative, and leaves everything else alone
|
|
func (v *Version5) DowngradeConfig(_ context.Context, e []byte) ([]byte, error) {
|
|
if i, err := jsonparser.GetInt(e, "orderManager", "futuresTrackingSeekDuration"); err == nil && i > 0 {
|
|
if e, err = jsonparser.Set(e, []byte(strconv.FormatInt(-i, 10)), "orderManager", "futuresTrackingSeekDuration"); err != nil {
|
|
return e, err
|
|
}
|
|
}
|
|
return e, nil
|
|
}
|