1 files changed, 7 insertions, 6 deletions
diff --git a/users.go b/users.go
index 884fac1..740f94f 100644
--- a/users.go
+++ b/users.go
@@ -5,22 +5,23 @@ package main
 
 import (
 	"context"
+
+	"github.com/jackc/pgx/v5"
 )
 
 func add_user_ssh(ctx context.Context, pubkey string) (user_id int, err error) {
-	tx, err := database.Begin(ctx)
-	if err != nil {
+	var tx pgx.Tx
+
+	if tx, err = database.Begin(ctx); err != nil {
 		return
 	}
 	defer tx.Rollback(ctx)
 
-	err = tx.QueryRow(ctx, `INSERT INTO users (type) VALUES ('pubkey_only') RETURNING id`).Scan(&user_id)
-	if err != nil {
+	if err = tx.QueryRow(ctx, `INSERT INTO users (type) VALUES ('pubkey_only') RETURNING id`).Scan(&user_id); err != nil {
 		return
 	}
 
-	_, err = tx.Exec(ctx, `INSERT INTO ssh_public_keys (key_string, user_id) VALUES ($1, $2)`, pubkey, user_id)
-	if err != nil {
+	if _, err = tx.Exec(ctx, `INSERT INTO ssh_public_keys (key_string, user_id) VALUES ($1, $2)`, pubkey, user_id); err != nil {
 		return
 	}