mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-31 07:26:44 +00:00
a5b638bfb7
* GHA, tests: Add additional checks for common issues These checks include: - Ensuring that all testify funcs use their formatted variants (e.g., `assert.Equalf(t, expected, actual)` instead of `assert.Equal(t, expected, actual)`). - Replacing `%s` with %q - Enforcing consistent usage of should/must wording for testify assert/require messages * Add support for checking backticked string format specifiers and fix issues * tests: Fix error comparisons * tests: Replace errors.Is(err, nil) usage with testify and automate check * refactor: Rename ExtractPort to ExtractPortOrDefault * tests: Replace assert with require for error handling in multiple test files * tests: Replace assert with require for error handling and improve assertions in data tests * tests: Fix typo in assertion message for StreamVol test * OKX: Fix GetOpenInterestAndVolumeStrike test with instrument selection and improved assertions * OKX: Revert intentional error check * Improve error message for expiry time check in GetOpenInterestAndVolumeStrike test
77 lines
2.6 KiB
Go
77 lines
2.6 KiB
Go
package v4
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/buger/jsonparser"
|
|
)
|
|
|
|
// Version is an Exchange upgrade to move currencyPairs.assetTypes to currencyPairs.pairs.*.assetEnabled
|
|
type Version struct{}
|
|
|
|
// Exchanges returns all exchanges: "*"
|
|
func (*Version) Exchanges() []string { return []string{"*"} }
|
|
|
|
// UpgradeExchange sets AssetEnabled: true for all assets listed in assetTypes, and false for any with no field
|
|
func (*Version) UpgradeExchange(_ context.Context, e []byte) ([]byte, error) {
|
|
toEnable := map[string]bool{}
|
|
|
|
assetTypesFn := func(asset []byte, valueType jsonparser.ValueType, _ int, _ error) {
|
|
if valueType == jsonparser.String {
|
|
toEnable[string(asset)] = true
|
|
}
|
|
}
|
|
_, err := jsonparser.ArrayEach(e, assetTypesFn, "currencyPairs", "assetTypes")
|
|
if err != nil && !errors.Is(err, jsonparser.KeyPathNotFoundError) {
|
|
return e, fmt.Errorf("error upgrading assetTypes: %w", err)
|
|
}
|
|
|
|
assetEnabledFn := func(assetBytes, v []byte, _ jsonparser.ValueType, _ int) (err error) {
|
|
asset := string(assetBytes)
|
|
if toEnable[asset] {
|
|
e, err = jsonparser.Set(e, []byte(`true`), "currencyPairs", "pairs", asset, "assetEnabled")
|
|
} else {
|
|
var vT jsonparser.ValueType
|
|
_, vT, _, err = jsonparser.Get(v, "assetEnabled")
|
|
switch {
|
|
case vT == jsonparser.Null, errors.Is(err, jsonparser.KeyPathNotFoundError):
|
|
e, err = jsonparser.Set(e, []byte(`false`), "currencyPairs", "pairs", asset, "assetEnabled")
|
|
case err == nil && vT != jsonparser.Boolean:
|
|
err = fmt.Errorf("assetEnabled: %w (%q)", jsonparser.UnknownValueTypeError, vT)
|
|
}
|
|
}
|
|
if err != nil {
|
|
err = fmt.Errorf("%w for asset %q", err, asset)
|
|
}
|
|
return err
|
|
}
|
|
if err = jsonparser.ObjectEach(bytes.Clone(e), assetEnabledFn, "currencyPairs", "pairs"); err != nil {
|
|
return e, fmt.Errorf("error upgrading currencyPairs.pairs: %w", err)
|
|
}
|
|
e = jsonparser.Delete(e, "currencyPairs", "assetTypes")
|
|
return e, err
|
|
}
|
|
|
|
// DowngradeExchange moves AssetEnabled assets into AssetType field
|
|
func (*Version) DowngradeExchange(_ context.Context, e []byte) ([]byte, error) {
|
|
assetTypes := []string{}
|
|
|
|
assetEnabledFn := func(asset, v []byte, _ jsonparser.ValueType, _ int) error {
|
|
if b, err := jsonparser.GetBoolean(v, "assetEnabled"); err == nil {
|
|
if b {
|
|
assetTypes = append(assetTypes, fmt.Sprintf("%q", asset))
|
|
}
|
|
e = jsonparser.Delete(e, "currencyPairs", "pairs", string(asset), "assetEnabled")
|
|
}
|
|
return nil
|
|
}
|
|
if err := jsonparser.ObjectEach(bytes.Clone(e), assetEnabledFn, "currencyPairs", "pairs"); err != nil {
|
|
return e, err
|
|
}
|
|
return jsonparser.Set(e, []byte(`[`+strings.Join(assetTypes, ",")+`]`), "currencyPairs", "assetTypes")
|
|
}
|