Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
M
mogo
Project
Project
Details
Activity
Releases
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Odysseus
mogo
Commits
9b398efe
Commit
9b398efe
authored
Jun 05, 2024
by
vicotor
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
remove not used func
parent
116fa5b6
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
31 additions
and
30 deletions
+31
-30
workerInstalledInfo.go
operator/workerInstalledInfo.go
+21
-20
workerinfo.go
operator/workerinfo.go
+10
-10
No files found.
operator/workerInstalledInfo.go
View file @
9b398efe
...
...
@@ -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
...
...
operator/workerinfo.go
View file @
9b398efe
...
...
@@ -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
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment