Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

add init table function and used in main #7

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

add init table function and used in main #7

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 21 additions & 0 deletions main.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,22 @@ import (
"github.com/jackc/pgx/v4"
)

func initTable(ctx context.Context, tx pgx.Tx) error {
// Dropping existing table if it exists
log.Println("Drop existing accounts table if necessary.")
if _, err := tx.Exec(ctx, "DROP TABLE IF EXISTS accounts"); err != nil {
return err
}

// Create the accounts table
log.Println("Creating accounts table.")
if _, err := tx.Exec(ctx,
"CREATE TABLE accounts (id UUID PRIMARY KEY DEFAULT gen_random_uuid(), balance INT8)"); err != nil {
return err
}
return nil
}

func insertRows(ctx context.Context, tx pgx.Tx, accts [4]uuid.UUID) error {
// Insert four rows into the "accounts" table.
log.Println("Creating new rows...")
Expand Down Expand Up @@ -85,6 +101,11 @@ func main() {
}
defer conn.Close(context.Background())

// Set up table
err = crdbpgx.ExecuteTx(context.Background(), conn, pgx.TxOptions{}, func(tx pgx.Tx) error {
return initTable(context.Background(), tx)
})

// Insert initial rows
var accounts [4]uuid.UUID
for i := 0; i < len(accounts); i++ {
Expand Down