short url web page v1

2023-12-25 17:04:22 +08:00
parent f27023c2ac
commit c1f4731669
45 changed files with 1637 additions and 93 deletions
--- a/db/db.go
+++ b/db/db.go
@@ -0,0 +1,31 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.24.0
+
+package db
+
+import (
+	"context"
+	"database/sql"
+)
+
+type DBTX interface {
+	ExecContext(context.Context, string, ...interface{}) (sql.Result, error)
+	PrepareContext(context.Context, string) (*sql.Stmt, error)
+	QueryContext(context.Context, string, ...interface{}) (*sql.Rows, error)
+	QueryRowContext(context.Context, string, ...interface{}) *sql.Row
+}
+
+func New(db DBTX) *Queries {
+	return &Queries{db: db}
+}
+
+type Queries struct {
+	db DBTX
+}
+
+func (q *Queries) WithTx(tx *sql.Tx) *Queries {
+	return &Queries{
+		db: tx,
+	}
+}
--- a/db/models.go
+++ b/db/models.go
@@ -0,0 +1,27 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.24.0
+
+package db
+
+import (
+	"time"
+)
+
+type User struct {
+	ID             string    `json:"id"`
+	Username       string    `json:"username"`
+	HashedPassword string    `json:"hashed_password"`
+	Email          string    `json:"email"`
+	CreatedAt      time.Time `json:"created_at"`
+}
+
+type UserRelateUrl struct {
+	ID        int64     `json:"id"`
+	UserID    string    `json:"user_id"`
+	ShortUrl  string    `json:"short_url"`
+	OriginUrl string    `json:"origin_url"`
+	Status    int32     `json:"status"`
+	ExpireAt  time.Time `json:"expire_at"`
+	CreatedAt time.Time `json:"created_at"`
+}
--- a/db/querier.go
+++ b/db/querier.go
@@ -0,0 +1,23 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.24.0
+
+package db
+
+import (
+	"context"
+)
+
+type Querier interface {
+	CreateUser(ctx context.Context, arg *CreateUserParams) (*User, error)
+	CreateUserUrl(ctx context.Context, arg *CreateUserUrlParams) (*UserRelateUrl, error)
+	DeleteUser(ctx context.Context, id string) error
+	GetUser(ctx context.Context, id string) (*User, error)
+	GetUserByEmail(ctx context.Context, email string) (*User, error)
+	GetUserByName(ctx context.Context, username string) (*User, error)
+	ListUrlByUser(ctx context.Context, userID string) ([]*UserRelateUrl, error)
+	ListUsers(ctx context.Context, arg *ListUsersParams) ([]*User, error)
+	UpdateUser(ctx context.Context, arg *UpdateUserParams) (*User, error)
+}
+
+var _ Querier = (*Queries)(nil)
--- a/db/query/user.sql
+++ b/db/query/user.sql
@@ -0,0 +1,36 @@
+-- name: CreateUser :one
+INSERT INTO users (
+  id, username, hashed_password, email
+) VALUES (
+  $1, $2, $3, $4
+)
+RETURNING *;
+
+-- name: DeleteUser :exec
+DELETE FROM users
+WHERE id = $1;
+
+-- name: UpdateUser :one
+UPDATE users
+SET hashed_password = $2,
+    email = $3
+WHERE id = $1
+RETURNING *;
+
+-- name: GetUser :one
+SELECT * FROM users
+WHERE id = $1 LIMIT 1;
+
+-- name: GetUserByName :one
+SELECT * FROM users
+WHERE username = $1 LIMIT 1;
+
+-- name: GetUserByEmail :one
+SELECT * FROM users
+WHERE email = $1 LIMIT 1;
+
+-- name: ListUsers :many
+SELECT * FROM users
+ORDER BY id
+LIMIT $1
+OFFSET $2;
--- a/db/query/user_relate_url.sql
+++ b/db/query/user_relate_url.sql
@@ -0,0 +1,13 @@
+-- name: CreateUserUrl :one
+INSERT INTO user_relate_url (
+  user_id, short_url, origin_url, status, expire_at
+) VALUES (
+  $1, $2, $3, 0, $4
+)
+RETURNING *;
+
+-- name: ListUrlByUser :many
+SELECT *
+FROM user_relate_url
+WHERE user_id = $1
+ORDER BY id DESC;
--- a/db/schema/000001_init_schema.down.sql
+++ b/db/schema/000001_init_schema.down.sql
@@ -0,0 +1,2 @@
+DROP TABLE "user_relate_url";
+DROP TABLE "users";
--- a/db/schema/000001_init_schema.up.sql
+++ b/db/schema/000001_init_schema.up.sql
@@ -0,0 +1,27 @@
+CREATE TABLE "users" (
+  "id" varchar NOT NULL PRIMARY KEY,
+  "username" varchar NOT NULL,
+  "hashed_password" varchar NOT NULL,
+  "email" varchar NOT NULL,
+  "created_at" timestamptz NOT NULL DEFAULT (now())
+);
+
+ALTER TABLE "users" ADD CONSTRAINT "username_key" UNIQUE ("username");
+ALTER TABLE "users" ADD CONSTRAINT "email_key" UNIQUE ("email");
+CREATE INDEX ON "users" ("username");
+CREATE INDEX ON "users" ("email");
+
+
+CREATE TABLE "user_relate_url" (
+    "id" bigserial NOT NULL PRIMARY KEY,
+    "user_id" varchar NOT NULL,
+    "short_url" varchar NOT NULL,
+    "origin_url" varchar NOT NULL,
+    "status" int NOT NULL DEFAULT 0,
+    "expire_at" timestamptz NOT NULL,
+    "created_at" timestamptz NOT NULL DEFAULT (now())
+);
+
+ALTER TABLE "user_relate_url" ADD CONSTRAINT "short_url_key" UNIQUE ("short_url");
+CREATE INDEX ON "user_relate_url" ("user_id");
+CREATE INDEX ON "user_relate_url" ("short_url");
--- a/db/store.go
+++ b/db/store.go
@@ -0,0 +1,61 @@
+package db
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+
+	"github.com/lib/pq"
+)
+
+type Store interface {
+	Querier
+	ExecTx(ctx context.Context, fn func(*Queries) error) error
+	IsUniqueViolation(err error) bool
+	IsForeignKeyViolation(err error) bool
+	IsNoRows(err error) bool
+}
+
+type SQLStore struct {
+	db *sql.DB
+	*Queries
+}
+
+func NewStore(db *sql.DB) Store {
+	return &SQLStore{
+		db:      db,
+		Queries: New(db),
+	}
+}
+
+func (store *SQLStore) ExecTx(ctx context.Context, fn func(*Queries) error) error {
+	tx, err := store.db.BeginTx(ctx, nil)
+	if err != nil {
+		return err
+	}
+
+	q := New(tx)
+	err = fn(q)
+	if err != nil {
+		if rbErr := tx.Rollback(); rbErr != nil {
+			return fmt.Errorf("tx err: %v, rb err: %v", err, rbErr)
+		}
+		return err
+	}
+
+	return tx.Commit()
+}
+
+func (store *SQLStore) IsUniqueViolation(err error) bool {
+	pqErr, ok := err.(*pq.Error)
+	return ok && pqErr.Code == "23505"
+}
+
+func (store *SQLStore) IsForeignKeyViolation(err error) bool {
+	pqErr, ok := err.(*pq.Error)
+	return ok && pqErr.Code == "23503"
+}
+
+func (store *SQLStore) IsNoRows(err error) bool {
+	return err == sql.ErrNoRows
+}
--- a/db/user.sql.go
+++ b/db/user.sql.go
@@ -0,0 +1,176 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.24.0
+// source: user.sql
+
+package db
+
+import (
+	"context"
+)
+
+const createUser = `-- name: CreateUser :one
+INSERT INTO users (
+  id, username, hashed_password, email
+) VALUES (
+  $1, $2, $3, $4
+)
+RETURNING id, username, hashed_password, email, created_at
+`
+
+type CreateUserParams struct {
+	ID             string `json:"id"`
+	Username       string `json:"username"`
+	HashedPassword string `json:"hashed_password"`
+	Email          string `json:"email"`
+}
+
+func (q *Queries) CreateUser(ctx context.Context, arg *CreateUserParams) (*User, error) {
+	row := q.db.QueryRowContext(ctx, createUser,
+		arg.ID,
+		arg.Username,
+		arg.HashedPassword,
+		arg.Email,
+	)
+	var i User
+	err := row.Scan(
+		&i.ID,
+		&i.Username,
+		&i.HashedPassword,
+		&i.Email,
+		&i.CreatedAt,
+	)
+	return &i, err
+}
+
+const deleteUser = `-- name: DeleteUser :exec
+DELETE FROM users
+WHERE id = $1
+`
+
+func (q *Queries) DeleteUser(ctx context.Context, id string) error {
+	_, err := q.db.ExecContext(ctx, deleteUser, id)
+	return err
+}
+
+const getUser = `-- name: GetUser :one
+SELECT id, username, hashed_password, email, created_at FROM users
+WHERE id = $1 LIMIT 1
+`
+
+func (q *Queries) GetUser(ctx context.Context, id string) (*User, error) {
+	row := q.db.QueryRowContext(ctx, getUser, id)
+	var i User
+	err := row.Scan(
+		&i.ID,
+		&i.Username,
+		&i.HashedPassword,
+		&i.Email,
+		&i.CreatedAt,
+	)
+	return &i, err
+}
+
+const getUserByEmail = `-- name: GetUserByEmail :one
+SELECT id, username, hashed_password, email, created_at FROM users
+WHERE email = $1 LIMIT 1
+`
+
+func (q *Queries) GetUserByEmail(ctx context.Context, email string) (*User, error) {
+	row := q.db.QueryRowContext(ctx, getUserByEmail, email)
+	var i User
+	err := row.Scan(
+		&i.ID,
+		&i.Username,
+		&i.HashedPassword,
+		&i.Email,
+		&i.CreatedAt,
+	)
+	return &i, err
+}
+
+const getUserByName = `-- name: GetUserByName :one
+SELECT id, username, hashed_password, email, created_at FROM users
+WHERE username = $1 LIMIT 1
+`
+
+func (q *Queries) GetUserByName(ctx context.Context, username string) (*User, error) {
+	row := q.db.QueryRowContext(ctx, getUserByName, username)
+	var i User
+	err := row.Scan(
+		&i.ID,
+		&i.Username,
+		&i.HashedPassword,
+		&i.Email,
+		&i.CreatedAt,
+	)
+	return &i, err
+}
+
+const listUsers = `-- name: ListUsers :many
+SELECT id, username, hashed_password, email, created_at FROM users
+ORDER BY id
+LIMIT $1
+OFFSET $2
+`
+
+type ListUsersParams struct {
+	Limit  int32 `json:"limit"`
+	Offset int32 `json:"offset"`
+}
+
+func (q *Queries) ListUsers(ctx context.Context, arg *ListUsersParams) ([]*User, error) {
+	rows, err := q.db.QueryContext(ctx, listUsers, arg.Limit, arg.Offset)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	items := []*User{}
+	for rows.Next() {
+		var i User
+		if err := rows.Scan(
+			&i.ID,
+			&i.Username,
+			&i.HashedPassword,
+			&i.Email,
+			&i.CreatedAt,
+		); err != nil {
+			return nil, err
+		}
+		items = append(items, &i)
+	}
+	if err := rows.Close(); err != nil {
+		return nil, err
+	}
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+	return items, nil
+}
+
+const updateUser = `-- name: UpdateUser :one
+UPDATE users
+SET hashed_password = $2,
+    email = $3
+WHERE id = $1
+RETURNING id, username, hashed_password, email, created_at
+`
+
+type UpdateUserParams struct {
+	ID             string `json:"id"`
+	HashedPassword string `json:"hashed_password"`
+	Email          string `json:"email"`
+}
+
+func (q *Queries) UpdateUser(ctx context.Context, arg *UpdateUserParams) (*User, error) {
+	row := q.db.QueryRowContext(ctx, updateUser, arg.ID, arg.HashedPassword, arg.Email)
+	var i User
+	err := row.Scan(
+		&i.ID,
+		&i.Username,
+		&i.HashedPassword,
+		&i.Email,
+		&i.CreatedAt,
+	)
+	return &i, err
+}
--- a/db/user_relate_url.sql.go
+++ b/db/user_relate_url.sql.go
@@ -0,0 +1,85 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.24.0
+// source: user_relate_url.sql
+
+package db
+
+import (
+	"context"
+	"time"
+)
+
+const createUserUrl = `-- name: CreateUserUrl :one
+INSERT INTO user_relate_url (
+  user_id, short_url, origin_url, status, expire_at
+) VALUES (
+  $1, $2, $3, 0, $4
+)
+RETURNING id, user_id, short_url, origin_url, status, expire_at, created_at
+`
+
+type CreateUserUrlParams struct {
+	UserID    string    `json:"user_id"`
+	ShortUrl  string    `json:"short_url"`
+	OriginUrl string    `json:"origin_url"`
+	ExpireAt  time.Time `json:"expire_at"`
+}
+
+func (q *Queries) CreateUserUrl(ctx context.Context, arg *CreateUserUrlParams) (*UserRelateUrl, error) {
+	row := q.db.QueryRowContext(ctx, createUserUrl,
+		arg.UserID,
+		arg.ShortUrl,
+		arg.OriginUrl,
+		arg.ExpireAt,
+	)
+	var i UserRelateUrl
+	err := row.Scan(
+		&i.ID,
+		&i.UserID,
+		&i.ShortUrl,
+		&i.OriginUrl,
+		&i.Status,
+		&i.ExpireAt,
+		&i.CreatedAt,
+	)
+	return &i, err
+}
+
+const listUrlByUser = `-- name: ListUrlByUser :many
+SELECT id, user_id, short_url, origin_url, status, expire_at, created_at
+FROM user_relate_url
+WHERE user_id = $1
+ORDER BY id DESC
+`
+
+func (q *Queries) ListUrlByUser(ctx context.Context, userID string) ([]*UserRelateUrl, error) {
+	rows, err := q.db.QueryContext(ctx, listUrlByUser, userID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	items := []*UserRelateUrl{}
+	for rows.Next() {
+		var i UserRelateUrl
+		if err := rows.Scan(
+			&i.ID,
+			&i.UserID,
+			&i.ShortUrl,
+			&i.OriginUrl,
+			&i.Status,
+			&i.ExpireAt,
+			&i.CreatedAt,
+		); err != nil {
+			return nil, err
+		}
+		items = append(items, &i)
+	}
+	if err := rows.Close(); err != nil {
+		return nil, err
+	}
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+	return items, nil
+}