init repo

internal/storage/dao/user.go

@@ -0,0 +1,53 @@
+package dao
+
+import (
+	"errors"
+
+	"gorm.io/gorm"
+)
+
+type User struct {
+	BaseModel
+	Username      string `gorm:"column:username" json:"username"`
+	Phone         string `gorm:"column:phone" json:"phone"`
+	Password      string `gorm:"column:password" json:"password"`
+	WechatOpenID  string `gorm:"column:wechat_open_id" json:"wechat_open_id"`
+	WechatUnionID string `gorm:"column:wechat_union_id" json:"wechat_union_id"`
+}
+
+func (u *User) TableName() string {
+	return "users"
+}
+
+type UserDao struct {
+}
+
+func NewUserDao() *UserDao {
+	return &UserDao{}
+}
+
+func (dao *UserDao) Create(tx *gorm.DB, user *User) error {
+	return tx.Create(user).Error
+}
+
+func (dao *UserDao) GetByPhone(tx *gorm.DB, phone string) (*User, error) {
+	var user User
+	if err := tx.Where("phone = ?", phone).First(&user).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, nil
+		}
+		return nil, err
+	}
+	return &user, nil
+}
+
+func (dao *UserDao) GetByID(tx *gorm.DB, id int64) (*User, error) {
+	var user User
+	if err := tx.Where("id = ?", id).First(&user).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, nil
+		}
+		return nil, err
+	}
+	return &user, nil
+}