okex: load broker id from env

c9s · Dec 16, 2024 · 49d1188 · 49d1188
1 parent 24148a3
commit 49d1188
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/pkg/exchange/okex/const.go b/pkg/exchange/okex/const.go
@@ -0,0 +1,3 @@
+package okex
+
+const defaultBrokerId string = "812a557d1116BCDE"
diff --git a/pkg/exchange/okex/exchange.go b/pkg/exchange/okex/exchange.go
@@ -3,6 +3,7 @@ package okex
 import (
 	"context"
 	"fmt"
+	"os"
 	"regexp"
 	"strconv"
 	"time"
@@ -92,6 +93,11 @@ func New(key, secret, passphrase string, opts ...Option) *Exchange {
 		passphrase:  passphrase,
 		client:      client,
 		timeNowFunc: time.Now,
+		brokerId:    defaultBrokerId,
+	}
+
+	if str, ok := os.LookupEnv("OKEX_BROKER_ID"); ok {
+		ex.brokerId = str
 	}
 
 	for _, o := range opts {
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		package okex

		const defaultBrokerId string = "812a557d1116BCDE"