Fix segments not removed from flushingCache (#10738)

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>

internal/datacoord/policy_test.go

@@ -481,7 +481,6 @@ func TestAvgAssignRegisterPolicy(t *testing.T) {
 					memkv.NewMemoryKV(),
 					map[int64]*NodeChannelInfo{
 						1: {1, []*channel{{"ch1", 1}, {"ch2", 1}}},
-						2: {2, []*channel{{"ch3", 1}, {"ch4", 1}}},
 					},
 				},
 				3,

internal/datanode/data_sync_service.go

@@ -212,8 +212,8 @@ func (dsService *dataSyncService) initNodes(vchanInfo *datapb.VchannelInfo) erro
 		if rsp.ErrorCode != commonpb.ErrorCode_Success {
 			return fmt.Errorf("data service save bin log path failed, reason = %s", rsp.Reason)
 		}
+		dsService.flushingSegCache.Remove(req.GetSegmentID())
 		return nil
-
 	})
 
 	c := &nodeConfig{