mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-02 03:48:37 +08:00
110 lines
2.8 KiB
Go
110 lines
2.8 KiB
Go
// Licensed to the LF AI & Data foundation under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you under the Apache License, Version 2.0 (the
|
|
// "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package datacoord
|
|
|
|
import (
|
|
"go.opentelemetry.io/otel/trace"
|
|
|
|
"github.com/milvus-io/milvus/internal/proto/datapb"
|
|
)
|
|
|
|
type CompactionTask interface {
|
|
Process() bool
|
|
BuildCompactionRequest() (*datapb.CompactionPlan, error)
|
|
GetSlotUsage() int64
|
|
GetLabel() string
|
|
|
|
SetTask(*datapb.CompactionTask)
|
|
GetTaskProto() *datapb.CompactionTask
|
|
SetPlan(plan *datapb.CompactionPlan)
|
|
ShadowClone(opts ...compactionTaskOpt) *datapb.CompactionTask
|
|
|
|
SetNodeID(UniqueID) error
|
|
NeedReAssignNodeID() bool
|
|
GetSpan() trace.Span
|
|
SetSpan(trace.Span)
|
|
SaveTaskMeta() error
|
|
}
|
|
|
|
type compactionTaskOpt func(task *datapb.CompactionTask)
|
|
|
|
func setNodeID(nodeID int64) compactionTaskOpt {
|
|
return func(task *datapb.CompactionTask) {
|
|
task.NodeID = nodeID
|
|
}
|
|
}
|
|
|
|
func setFailReason(reason string) compactionTaskOpt {
|
|
return func(task *datapb.CompactionTask) {
|
|
task.FailReason = reason
|
|
}
|
|
}
|
|
|
|
func setEndTime(endTime int64) compactionTaskOpt {
|
|
return func(task *datapb.CompactionTask) {
|
|
task.EndTime = endTime
|
|
}
|
|
}
|
|
|
|
func setTimeoutInSeconds(dur int32) compactionTaskOpt {
|
|
return func(task *datapb.CompactionTask) {
|
|
task.TimeoutInSeconds = dur
|
|
}
|
|
}
|
|
|
|
func setResultSegments(segments []int64) compactionTaskOpt {
|
|
return func(task *datapb.CompactionTask) {
|
|
task.ResultSegments = segments
|
|
}
|
|
}
|
|
|
|
func setTmpSegments(segments []int64) compactionTaskOpt {
|
|
return func(task *datapb.CompactionTask) {
|
|
task.TmpSegments = segments
|
|
}
|
|
}
|
|
|
|
func setState(state datapb.CompactionTaskState) compactionTaskOpt {
|
|
return func(task *datapb.CompactionTask) {
|
|
task.State = state
|
|
}
|
|
}
|
|
|
|
func setStartTime(startTime int64) compactionTaskOpt {
|
|
return func(task *datapb.CompactionTask) {
|
|
task.StartTime = startTime
|
|
}
|
|
}
|
|
|
|
func setRetryTimes(retryTimes int32) compactionTaskOpt {
|
|
return func(task *datapb.CompactionTask) {
|
|
task.RetryTimes = retryTimes
|
|
}
|
|
}
|
|
|
|
func setLastStateStartTime(lastStateStartTime int64) compactionTaskOpt {
|
|
return func(task *datapb.CompactionTask) {
|
|
task.LastStateStartTime = lastStateStartTime
|
|
}
|
|
}
|
|
|
|
func setAnalyzeTaskID(id int64) compactionTaskOpt {
|
|
return func(task *datapb.CompactionTask) {
|
|
task.AnalyzeTaskID = id
|
|
}
|
|
}
|