bbgo: fix active book order removal

pkg/bbgo/active_book.go

@@ -102,17 +102,9 @@ func (b *LocalActiveOrderBook) orderUpdateHandler(order types.Order) {
 			b.EmitFilled(order)
 		}
 
-	case types.OrderStatusPartiallyFilled:
+	case types.OrderStatusPartiallyFilled, types.OrderStatusNew:
 		b.Update(order)
 
-	case types.OrderStatusNew:
-		if order.Quantity == 0 {
-			log.Debugf("[LocalActiveOrderBook] order status %s, removing order %s", order.Status, order)
-			b.Remove(order)
-		} else {
-			b.Update(order)
-		}
-
 	case types.OrderStatusCanceled, types.OrderStatusRejected:
 		log.Debugf("[LocalActiveOrderBook] order status %s, removing order %s", order.Status, order)
 		b.Remove(order)

pkg/types/ordermap.go

@@ -97,15 +97,14 @@ func (m *SyncOrderMap) Backup() (orders []SubmitOrder) {
 }
 
 func (m *SyncOrderMap) Remove(orderID uint64) (exists bool) {
-	exists = m.Exists(orderID)
+	m.Lock()
+	defer m.Unlock()
+
+	exists = m.orders.Exists(orderID)
 	if exists {
-		m.Lock()
 		m.orders.Remove(orderID)
-		m.Unlock()
 	} else {
-		m.Lock()
 		m.pendingRemoval[orderID] = time.Now()
-		m.Unlock()
 	}
 
 	return exists