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

sojuctl: use background context

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) 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: