bbgo: call clear to reset map

pkg/bbgo/activeorderbook.go

@@ -24,14 +24,14 @@ const DefaultOrderCancelTimeout = 5 * time.Second
 //go:generate callbackgen -type ActiveOrderBook
 type ActiveOrderBook struct {
 	Symbol string
-	orders *types.SyncOrderMap
+
+	orders              *types.SyncOrderMap
+	pendingOrderUpdates *types.SyncOrderMap
 
 	newCallbacks      []func(o types.Order)
 	filledCallbacks   []func(o types.Order)
 	canceledCallbacks []func(o types.Order)
 
-	pendingOrderUpdates *types.SyncOrderMap
-
 	// sig is the order update signal
 	// this signal will be emitted when a new order is added or removed.
 	C sigchan.Chan
@@ -291,13 +291,21 @@ func (b *ActiveOrderBook) GracefulCancel(ctx context.Context, ex types.Exchange,
 			}
 
 		}
+	}
 
+	if cancelAll {
+		b.clear()
 	}
 
 	log.Debugf("[ActiveOrderBook] all %s orders are cancelled successfully in %s", b.Symbol, time.Since(startTime))
 	return nil
 }
 
+func (b *ActiveOrderBook) clear() {
+	b.orders.Clear()
+	b.pendingOrderUpdates.Clear()
+}
+
 func (b *ActiveOrderBook) orderUpdateHandler(order types.Order) {
 	if oldOrder, ok := b.Get(order.OrderID); ok {
 		order.Tag = oldOrder.Tag

pkg/types/ordermap.go

@@ -55,6 +55,10 @@ func (m OrderMap) Remove(orderID uint64) {
 	delete(m, orderID)
 }
 
+func (m *OrderMap) Clear() {
+	*m = make(OrderMap, 10)
+}
+
 func (m OrderMap) IDs() (ids []uint64) {
 	for id := range m {
 		ids = append(ids, id)
@@ -123,6 +127,13 @@ func (m *SyncOrderMap) Backup() (orders []SubmitOrder) {
 	return orders
 }
 
+func (m *SyncOrderMap) Clear() {
+	m.Lock()
+	m.orders.Clear()
+	m.pendingRemoval = make(map[uint64]time.Time, 10)
+	m.Unlock()
+}
+
 func (m *SyncOrderMap) Remove(orderID uint64) (exists bool) {
 	m.Lock()
 	defer m.Unlock()