123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156 |
- package mysql
- import (
- "context"
- "strings"
- "github.com/pkg/errors"
- "github.com/usememos/memos/store"
- )
- func (d *DB) CreateUser(ctx context.Context, create *store.User) (*store.User, error) {
- fields := []string{"`username`", "`role`", "`email`", "`nickname`", "`password_hash`", "`avatar_url`"}
- placeholder := []string{"?", "?", "?", "?", "?", "?"}
- args := []any{create.Username, create.Role, create.Email, create.Nickname, create.PasswordHash, create.AvatarURL}
- stmt := "INSERT INTO user (" + strings.Join(fields, ", ") + ") VALUES (" + strings.Join(placeholder, ", ") + ")"
- result, err := d.db.ExecContext(ctx, stmt, args...)
- if err != nil {
- return nil, err
- }
- id, err := result.LastInsertId()
- if err != nil {
- return nil, err
- }
- id32 := int32(id)
- list, err := d.ListUsers(ctx, &store.FindUser{ID: &id32})
- if err != nil {
- return nil, err
- }
- if len(list) != 1 {
- return nil, errors.Wrapf(nil, "unexpected user count: %d", len(list))
- }
- return list[0], nil
- }
- func (d *DB) UpdateUser(ctx context.Context, update *store.UpdateUser) (*store.User, error) {
- set, args := []string{}, []any{}
- if v := update.UpdatedTs; v != nil {
- set, args = append(set, "`updated_ts` = FROM_UNIXTIME(?)"), append(args, *v)
- }
- if v := update.RowStatus; v != nil {
- set, args = append(set, "`row_status` = ?"), append(args, *v)
- }
- if v := update.Username; v != nil {
- set, args = append(set, "`username` = ?"), append(args, *v)
- }
- if v := update.Email; v != nil {
- set, args = append(set, "`email` = ?"), append(args, *v)
- }
- if v := update.Nickname; v != nil {
- set, args = append(set, "`nickname` = ?"), append(args, *v)
- }
- if v := update.AvatarURL; v != nil {
- set, args = append(set, "`avatar_url` = ?"), append(args, *v)
- }
- if v := update.PasswordHash; v != nil {
- set, args = append(set, "`password_hash` = ?"), append(args, *v)
- }
- args = append(args, update.ID)
- query := "UPDATE `user` SET " + strings.Join(set, ", ") + " WHERE `id` = ?"
- if _, err := d.db.ExecContext(ctx, query, args...); err != nil {
- return nil, err
- }
- user, err := d.GetUser(ctx, &store.FindUser{ID: &update.ID})
- if err != nil {
- return nil, err
- }
- return user, nil
- }
- func (d *DB) ListUsers(ctx context.Context, find *store.FindUser) ([]*store.User, error) {
- where, args := []string{"1 = 1"}, []any{}
- if v := find.ID; v != nil {
- where, args = append(where, "`id` = ?"), append(args, *v)
- }
- if v := find.Username; v != nil {
- where, args = append(where, "`username` = ?"), append(args, *v)
- }
- if v := find.Role; v != nil {
- where, args = append(where, "`role` = ?"), append(args, *v)
- }
- if v := find.Email; v != nil {
- where, args = append(where, "`email` = ?"), append(args, *v)
- }
- if v := find.Nickname; v != nil {
- where, args = append(where, "`nickname` = ?"), append(args, *v)
- }
- query := "SELECT `id`, `username`, `role`, `email`, `nickname`, `password_hash`, `avatar_url`, UNIX_TIMESTAMP(`created_ts`), UNIX_TIMESTAMP(`updated_ts`), `row_status` FROM `user` WHERE " + strings.Join(where, " AND ") + " ORDER BY `created_ts` DESC, `row_status` DESC"
- rows, err := d.db.QueryContext(ctx, query, args...)
- if err != nil {
- return nil, err
- }
- defer rows.Close()
- list := make([]*store.User, 0)
- for rows.Next() {
- var user store.User
- if err := rows.Scan(
- &user.ID,
- &user.Username,
- &user.Role,
- &user.Email,
- &user.Nickname,
- &user.PasswordHash,
- &user.AvatarURL,
- &user.CreatedTs,
- &user.UpdatedTs,
- &user.RowStatus,
- ); err != nil {
- return nil, err
- }
- list = append(list, &user)
- }
- if err := rows.Err(); err != nil {
- return nil, err
- }
- return list, nil
- }
- func (d *DB) GetUser(ctx context.Context, find *store.FindUser) (*store.User, error) {
- list, err := d.ListUsers(ctx, find)
- if err != nil {
- return nil, err
- }
- if len(list) != 1 {
- return nil, errors.Wrapf(nil, "unexpected user count: %d", len(list))
- }
- return list[0], nil
- }
- func (d *DB) DeleteUser(ctx context.Context, delete *store.DeleteUser) error {
- result, err := d.db.ExecContext(ctx, "DELETE FROM `user` WHERE `id` = ?", delete.ID)
- if err != nil {
- return err
- }
- if _, err := result.RowsAffected(); err != nil {
- return err
- }
- if err := d.Vacuum(ctx); err != nil {
- // Prevent linter warning.
- return err
- }
- return nil
- }
|