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
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
// sort by wait time
findOptions := options.Find()
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)
if err != nil {
return nil, err
}
defer cursor.Close(ctx)
var workers []*WorkerInstalledInfo
if err = cursor.All(ctx, &workers); err != nil {
return nil, err
}
return workers, nil
}
//
//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
// // sort by gpu_mem_free.
// findOptions := options.Find()
// 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)
// if err != nil {
// return nil, err
// }
// defer cursor.Close(ctx)
//
// var workers []*WorkerInstalledInfo
// if err = cursor.All(ctx, &workers); err != nil {
// return nil, err
// }
// return workers, nil
//}
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
......
......@@ -57,11 +57,11 @@ func (d *WorkerInfoOperator) DeleteByWorkerId(ctx context.Context, workerid stri
return int(res.DeletedCount), err
}
func (d *WorkerInfoOperator) UpdateModel(ctx context.Context, workerid string, models *types.ModelInfo) error {
update := bson.M{"$set": bson.M{"model_infos": models}}
_, err := d.col.UpdateOne(ctx, bson.M{"worker_id": workerid}, update)
return err
}
//func (d *WorkerInfoOperator) UpdateModel(ctx context.Context, workerid string, models *types.ModelInfo) error {
// update := bson.M{"$set": bson.M{"model_infos": models}}
// _, err := d.col.UpdateOne(ctx, bson.M{"worker_id": workerid}, update)
// return err
//}
func (d *WorkerInfoOperator) UpdateBenefitAddr(ctx context.Context, workerid string, benefitAddr string) error {
update := bson.M{"$set": bson.M{"node_info.benefit_addr": benefitAddr}}
......@@ -140,11 +140,11 @@ func (d *WorkerInfoOperator) UpdateHardware(ctx context.Context, workerid string
return err
}
func (d *WorkerInfoOperator) UpdateNodeInfo(ctx context.Context, workerid string, nodeInfo *types.NodeInfo) error {
update := bson.M{"$set": bson.M{"node_info": nodeInfo}}
_, err := d.col.UpdateOne(ctx, bson.M{"worker_id": workerid}, update)
return err
}
//func (d *WorkerInfoOperator) UpdateNodeInfo(ctx context.Context, workerid string, nodeInfo *types.NodeInfo) error {
// update := bson.M{"$set": bson.M{"node_info": nodeInfo}}
// _, err := d.col.UpdateOne(ctx, bson.M{"worker_id": workerid}, update)
// return err
//}
func (d *WorkerInfoOperator) UpdateWorker(ctx context.Context, worker *WorkerInfo) error {
_, 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