Commit 1d2cac90 authored by luxq's avatar luxq

add delete func

parent 13aa8d27
...@@ -32,6 +32,11 @@ func (d *WorkerInfoOperator) InsertWorker(ctx context.Context, worker *WorkerInf ...@@ -32,6 +32,11 @@ func (d *WorkerInfoOperator) InsertWorker(ctx context.Context, worker *WorkerInf
return d.col.InsertOne(ctx, worker) return d.col.InsertOne(ctx, worker)
} }
func (d *WorkerInfoOperator) DeleteByWorkerId(ctx context.Context, workerid string) (int, error) {
res, err := d.col.DeleteOne(ctx, bson.M{"worker_id": workerid})
return int(res.DeletedCount), err
}
func (d *WorkerInfoOperator) UpdateModel(ctx context.Context, id string, models *types.ModelInfo) error { func (d *WorkerInfoOperator) UpdateModel(ctx context.Context, id 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{"_id": id}, update) _, err := d.col.UpdateOne(ctx, bson.M{"_id": id}, update)
......
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