Skip to content

Commit

Permalink
worker: Command to print resource-table env vars
Browse files Browse the repository at this point in the history
  • Loading branch information
magik6k committed Nov 29, 2021
1 parent ef3b51b commit 6cf54af
Show file tree
Hide file tree
Showing 3 changed files with 89 additions and 0 deletions.
1 change: 1 addition & 0 deletions cmd/lotus-seal-worker/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,7 @@ func main() {
storageCmd,
setCmd,
waitQuietCmd,
resourcesCmd,
tasksCmd,
}

Expand Down
72 changes: 72 additions & 0 deletions cmd/lotus-seal-worker/resources.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,72 @@
package main

import (
"fmt"
"os"
"sort"

"github.com/urfave/cli/v2"

"github.com/filecoin-project/lotus/extern/sector-storage/storiface"
)

var resourcesCmd = &cli.Command{
Name: "resources",
Usage: "Manage resource table overrides",
Flags: []cli.Flag{
&cli.BoolFlag{
Name: "all",
Usage: "print all resource envvars",
},
&cli.BoolFlag{
Name: "default",
Usage: "print all resource envvars",
},
},
Action: func(cctx *cli.Context) error {
def := map[string]string{}
set := map[string]string{}
all := map[string]string{}

_, err := storiface.ParseResourceEnv(func(key, d string) (string, bool) {
if d != "" {
all[key] = d
def[key] = d
}

s, ok := os.LookupEnv(key)
if ok {
all[key] = s
set[key] = s
}

return s, ok
})
if err != nil {
return err
}

printMap := func(m map[string]string) {
var arr []string
for k, v := range m {
arr = append(arr, fmt.Sprintf("%s=%s", k, v))
}
sort.Strings(arr)
for _, s := range arr {
fmt.Println(s)
}
}

if cctx.Bool("default") {
printMap(def)
} else {
if cctx.Bool("all") {
printMap(all)
} else {
printMap(set)
}
}

return nil
},
}
16 changes: 16 additions & 0 deletions documentation/en/cli-lotus-worker.md
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@ COMMANDS:
storage manage sector storage
set Manage worker settings
wait-quiet Block until all running tasks exit
resources Manage resource table overrides
tasks Manage task processing
help, h Shows a list of commands or help for one command
Expand Down Expand Up @@ -127,6 +128,21 @@ OPTIONS:
```

## lotus-worker resources
```
NAME:
lotus-worker resources - Manage resource table overrides
USAGE:
lotus-worker resources [command options] [arguments...]
OPTIONS:
--all print all resource envvars (default: false)
--default print all resource envvars (default: false)
--help, -h show help (default: false)
```

## lotus-worker tasks
```
NAME:
Expand Down

0 comments on commit 6cf54af

Please sign in to comment.