sojuctl: use background context
This commit is contained in:
parent
3941f67380
commit
0c5aa8cbff
@ -49,6 +49,8 @@ func main() {
|
|||||||
log.Fatalf("failed to open database: %v", err)
|
log.Fatalf("failed to open database: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
switch cmd := flag.Arg(0); cmd {
|
switch cmd := flag.Arg(0); cmd {
|
||||||
case "create-user":
|
case "create-user":
|
||||||
username := flag.Arg(1)
|
username := flag.Arg(1)
|
||||||
@ -76,7 +78,7 @@ func main() {
|
|||||||
Password: string(hashed),
|
Password: string(hashed),
|
||||||
Admin: *admin,
|
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)
|
log.Fatalf("failed to create user: %v", err)
|
||||||
}
|
}
|
||||||
case "change-password":
|
case "change-password":
|
||||||
@ -86,7 +88,7 @@ func main() {
|
|||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
|
|
||||||
user, err := db.GetUser(context.TODO(), username)
|
user, err := db.GetUser(ctx, username)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatalf("failed to get user: %v", err)
|
log.Fatalf("failed to get user: %v", err)
|
||||||
}
|
}
|
||||||
@ -102,7 +104,7 @@ func main() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
user.Password = string(hashed)
|
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)
|
log.Fatalf("failed to update password: %v", err)
|
||||||
}
|
}
|
||||||
default:
|
default:
|
||||||
|
Loading…
Reference in New Issue
Block a user