Commit 9b398efe authored by vicotor's avatar vicotor

remove not used func

parent 116fa5b6
...@@ -55,26 +55,27 @@ func (d *WorkerInstalledOperator) DeleteByWorkerId(ctx context.Context, workerid ...@@ -55,26 +55,27 @@ func (d *WorkerInstalledOperator) DeleteByWorkerId(ctx context.Context, workerid
return int(res.DeletedCount), err return int(res.DeletedCount), err
} }
func (d *WorkerInstalledOperator) FindWorkerByModelId(ctx context.Context, modelId int, limit int) ([]*WorkerInstalledInfo, error) { //
// find all worker that at least one installed model's mode_id is equal modelId //func (d *WorkerInstalledOperator) FindWorkerByModelId(ctx context.Context, modelId int, limit int) ([]*WorkerInstalledInfo, error) {
// sort by wait time // // find all worker that at least one installed model's mode_id is equal modelId
findOptions := options.Find() // // sort by gpu_mem_free.
findOptions.SetLimit(int64(limit)) // findOptions := options.Find()
findOptions.SetSort(bson.D{{"gpu_free", 1}}) // findOptions.SetLimit(int64(limit))
// findOptions.SetSort(bson.D{{"gpu_free", -1}})
selector := bson.M{"model_id": modelId} //
cursor, err := d.col.Find(ctx, selector, findOptions) // selector := bson.M{"model_id": modelId}
if err != nil { // cursor, err := d.col.Find(ctx, selector, findOptions)
return nil, err // if err != nil {
} // return nil, err
defer cursor.Close(ctx) // }
// defer cursor.Close(ctx)
var workers []*WorkerInstalledInfo //
if err = cursor.All(ctx, &workers); err != nil { // var workers []*WorkerInstalledInfo
return nil, err // if err = cursor.All(ctx, &workers); err != nil {
} // return nil, err
return workers, nil // }
} // return workers, nil
//}
func (d *WorkerInstalledOperator) FindWorkerByModelIdAndGpuMem(ctx context.Context, modelId int, mem int64, limit int) ([]*WorkerInstalledInfo, error) { func (d *WorkerInstalledOperator) FindWorkerByModelIdAndGpuMem(ctx context.Context, modelId int, mem int64, limit int) ([]*WorkerInstalledInfo, error) {
// find all worker that at least one installed model's mode_id is equal modelId // find all worker that at least one installed model's mode_id is equal modelId
......
...@@ -57,11 +57,11 @@ func (d *WorkerInfoOperator) DeleteByWorkerId(ctx context.Context, workerid stri ...@@ -57,11 +57,11 @@ func (d *WorkerInfoOperator) DeleteByWorkerId(ctx context.Context, workerid stri
return int(res.DeletedCount), err return int(res.DeletedCount), err
} }
func (d *WorkerInfoOperator) UpdateModel(ctx context.Context, workerid string, models *types.ModelInfo) error { //func (d *WorkerInfoOperator) UpdateModel(ctx context.Context, workerid string, models *types.ModelInfo) error {
update := bson.M{"$set": bson.M{"model_infos": models}} // update := bson.M{"$set": bson.M{"model_infos": models}}
_, err := d.col.UpdateOne(ctx, bson.M{"worker_id": workerid}, update) // _, err := d.col.UpdateOne(ctx, bson.M{"worker_id": workerid}, update)
return err // return err
} //}
func (d *WorkerInfoOperator) UpdateBenefitAddr(ctx context.Context, workerid string, benefitAddr string) error { func (d *WorkerInfoOperator) UpdateBenefitAddr(ctx context.Context, workerid string, benefitAddr string) error {
update := bson.M{"$set": bson.M{"node_info.benefit_addr": benefitAddr}} update := bson.M{"$set": bson.M{"node_info.benefit_addr": benefitAddr}}
...@@ -140,11 +140,11 @@ func (d *WorkerInfoOperator) UpdateHardware(ctx context.Context, workerid string ...@@ -140,11 +140,11 @@ func (d *WorkerInfoOperator) UpdateHardware(ctx context.Context, workerid string
return err return err
} }
func (d *WorkerInfoOperator) UpdateNodeInfo(ctx context.Context, workerid string, nodeInfo *types.NodeInfo) error { //func (d *WorkerInfoOperator) UpdateNodeInfo(ctx context.Context, workerid string, nodeInfo *types.NodeInfo) error {
update := bson.M{"$set": bson.M{"node_info": nodeInfo}} // update := bson.M{"$set": bson.M{"node_info": nodeInfo}}
_, err := d.col.UpdateOne(ctx, bson.M{"worker_id": workerid}, update) // _, err := d.col.UpdateOne(ctx, bson.M{"worker_id": workerid}, update)
return err // return err
} //}
func (d *WorkerInfoOperator) UpdateWorker(ctx context.Context, worker *WorkerInfo) error { func (d *WorkerInfoOperator) UpdateWorker(ctx context.Context, worker *WorkerInfo) error {
_, err := d.col.ReplaceOne(ctx, bson.M{"worker_id": worker.WorkerId}, worker) _, err := d.col.ReplaceOne(ctx, bson.M{"worker_id": worker.WorkerId}, worker)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment