sad/soju
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

sojuctl: use background context

Browse Source
This commit is contained in:
Simon Ser 2021-11-17 11:54:28 +01:00
parent 3941f67380
commit 0c5aa8cbff
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
cmd/sojuctl/main.go
View File

@ -49,6 +49,8 @@ func main() {
log.Fatalf("failed to open database: %v", err)
}
ctx := context.Background()
switch cmd := flag.Arg(0); cmd {
case "create-user":
username := flag.Arg(1)
@ -76,7 +78,7 @@ func main() {
Password: string(hashed),
Admin: *admin,
}
if err := db.StoreUser(context.TODO(), &user); err != nil {
if err := db.StoreUser(ctx, &user); err != nil {
log.Fatalf("failed to create user: %v", err)
}
case "change-password":
@ -86,7 +88,7 @@ func main() {
os.Exit(1)
}
user, err := db.GetUser(context.TODO(), username)
user, err := db.GetUser(ctx, username)
if err != nil {
log.Fatalf("failed to get user: %v", err)
}
@ -102,7 +104,7 @@ func main() {
}
user.Password = string(hashed)
if err := db.StoreUser(context.TODO(), user); err != nil {
if err := db.StoreUser(ctx, user); err != nil {
log.Fatalf("failed to update password: %v", err)
}
default: