@@ -180,19 +180,14 @@ func (mgr *segmentManager) UpdateSegmentVersion(segmentType SegmentType, segment
180
180
mgr .mu .Lock ()
181
181
defer mgr .mu .Unlock ()
182
182
183
- targetMap := mgr .growingSegments
184
- switch segmentType {
185
- case SegmentTypeGrowing :
186
- targetMap = mgr .growingSegments
187
- case SegmentTypeSealed :
188
- targetMap = mgr .sealedSegments
189
- default :
190
- panic ("unexpected segment type" )
183
+ segment , ok := mgr .sealedSegments [segmentID ]
184
+ if ! ok {
185
+ segment , ok = mgr .growingSegments [segmentID ]
191
186
}
192
187
193
- segment , ok := targetMap [segmentID ]
194
188
if ! ok {
195
189
log .Warn ("segment not exist, skip segment version change" ,
190
+ zap .Any ("type" , segmentType ),
196
191
zap .Int64 ("segmentID" , segmentID ),
197
192
zap .Int64 ("newVersion" , newVersion ),
198
193
)
@@ -202,13 +197,13 @@ func (mgr *segmentManager) UpdateSegmentVersion(segmentType SegmentType, segment
202
197
if segment .Version () >= newVersion {
203
198
log .Warn ("Invalid segment version changed, skip it" ,
204
199
zap .Int64 ("segmentID" , segment .ID ()),
200
+ zap .Any ("type" , segmentType ),
205
201
zap .Int64 ("oldVersion" , segment .Version ()),
206
202
zap .Int64 ("newVersion" , newVersion ))
207
203
return
208
204
}
209
205
210
206
segment .UpdateVersion (newVersion )
211
- targetMap [segmentID ] = segment
212
207
}
213
208
214
209
func (mgr * segmentManager ) Get (segmentID UniqueID ) Segment {
0 commit comments