From 3bd2d90e3c6011201dc0239a8c7bb9663024ff2e Mon Sep 17 00:00:00 2001
From: edwin <edwin@maicoin.com>
Date: Tue, 20 Feb 2024 15:14:09 +0800
Subject: [PATCH] pkg/exchange: allow char in place order

---
 pkg/exchange/okex/exchange.go      | 18 ++++++++++-------
 pkg/exchange/okex/exchange_test.go | 31 ++++++++++++++++++++++++++++++
 2 files changed, 42 insertions(+), 7 deletions(-)
 create mode 100644 pkg/exchange/okex/exchange_test.go

diff --git a/pkg/exchange/okex/exchange.go b/pkg/exchange/okex/exchange.go
index b905cb3eb..dc398bfd0 100644
--- a/pkg/exchange/okex/exchange.go
+++ b/pkg/exchange/okex/exchange.go
@@ -3,6 +3,7 @@ package okex
 import (
 	"context"
 	"fmt"
+	"regexp"
 	"strconv"
 	"time"
 
@@ -33,6 +34,9 @@ var (
 	queryClosedOrderRateLimiter = rate.NewLimiter(rate.Every(100*time.Millisecond), 10)
 	queryTradeLimiter           = rate.NewLimiter(rate.Every(100*time.Millisecond), 10)
 	queryKLineLimiter           = rate.NewLimiter(rate.Every(100*time.Millisecond), 20)
+
+	// clientOrderIdRegex combine of case-sensitive alphanumerics, all numbers, or all letters of up to 32 characters.
+	clientOrderIdRegex = regexp.MustCompile("^[a-zA-Z0-9]{0,32}$")
 )
 
 const (
@@ -241,11 +245,12 @@ func (e *Exchange) SubmitOrder(ctx context.Context, order types.SubmitOrder) (*t
 		return nil, fmt.Errorf("place order rate limiter wait error: %w", err)
 	}
 
-	_, err = strconv.ParseInt(order.ClientOrderID, 10, 64)
-	if err != nil {
-		return nil, fmt.Errorf("client order id should be numberic: %s, err: %w", order.ClientOrderID, err)
+	if len(order.ClientOrderID) > 0 {
+		if ok := clientOrderIdRegex.MatchString(order.ClientOrderID); !ok {
+			return nil, fmt.Errorf("client order id should be case-sensitive alphanumerics, all numbers, or all letters of up to 32 characters: %s", order.ClientOrderID)
+		}
+		orderReq.ClientOrderID(order.ClientOrderID)
 	}
-	orderReq.ClientOrderID(order.ClientOrderID)
 
 	orders, err := orderReq.Do(ctx)
 	if err != nil {
@@ -357,9 +362,8 @@ func (e *Exchange) CancelOrders(ctx context.Context, orders ...types.Order) erro
 		req.InstrumentID(toLocalSymbol(order.Symbol))
 		req.OrderID(strconv.FormatUint(order.OrderID, 10))
 		if len(order.ClientOrderID) > 0 {
-			_, err := strconv.ParseInt(order.ClientOrderID, 10, 64)
-			if err != nil {
-				return fmt.Errorf("client order id should be numberic: %s, err: %w", order.ClientOrderID, err)
+			if ok := clientOrderIdRegex.MatchString(order.ClientOrderID); !ok {
+				return fmt.Errorf("client order id should be case-sensitive alphanumerics, all numbers, or all letters of up to 32 characters: %s", order.ClientOrderID)
 			}
 			req.ClientOrderID(order.ClientOrderID)
 		}
diff --git a/pkg/exchange/okex/exchange_test.go b/pkg/exchange/okex/exchange_test.go
new file mode 100644
index 000000000..0282a1c25
--- /dev/null
+++ b/pkg/exchange/okex/exchange_test.go
@@ -0,0 +1,31 @@
+package okex
+
+import (
+	"strings"
+	"testing"
+
+	"github.com/google/uuid"
+	"github.com/stretchr/testify/assert"
+)
+
+func Test_clientOrderIdRegex(t *testing.T) {
+	t.Run("empty client order id", func(t *testing.T) {
+		assert.True(t, clientOrderIdRegex.MatchString(""))
+	})
+
+	t.Run("mixed of digit and char", func(t *testing.T) {
+		assert.True(t, clientOrderIdRegex.MatchString("1s2f3g4h5j"))
+	})
+
+	t.Run("mixed of 16 chars and 16 digit", func(t *testing.T) {
+		assert.True(t, clientOrderIdRegex.MatchString(strings.Repeat("s", 16)+strings.Repeat("1", 16)))
+	})
+
+	t.Run("out of maximum length", func(t *testing.T) {
+		assert.False(t, clientOrderIdRegex.MatchString(strings.Repeat("s", 33)))
+	})
+
+	t.Run("invalid char: `-`", func(t *testing.T) {
+		assert.False(t, clientOrderIdRegex.MatchString(uuid.NewString()))
+	})
+}