直接完成

dao/user.go

@@ -0,0 +1,51 @@
+package dao
+
+import (
+	"gorm.io/gorm"
+	"integrity-checkin/data/po"
+)
+
+type UserDAO struct {
+	Tx *gorm.DB
+}
+
+func (u UserDAO) FindUserByUsername(username string) *po.User {
+	var user po.User
+	err := u.Tx.Take(&user, "username=?", username).Error
+	if err == gorm.ErrRecordNotFound {
+		return nil
+	}
+	if err != nil {
+		panic(err)
+	}
+	return &user
+}
+func (u UserDAO) FindUserByID(id int) *po.User {
+	var user po.User
+	err := u.Tx.Take(&user, id).Error
+	if err == gorm.ErrRecordNotFound {
+		return nil
+	}
+	if err != nil {
+		panic(err)
+	}
+	return &user
+}
+func (u UserDAO) SaveUser(user *po.User) {
+	err := u.Tx.Save(user).Error
+	if err != nil {
+		panic(err)
+	}
+}
+func (u UserDAO) CreateUser(username string, realName string) *po.User {
+	user := po.User{
+		Username: username,
+		RealName: realName,
+		RoleID:   1,
+	}
+	err := u.Tx.Create(&user).Error
+	if err != nil {
+		panic(err)
+	}
+	return &user
+}