This commit is contained in:
2023-05-11 21:28:34 +08:00
parent 2871fb556c
commit cf0d32bb32
5 changed files with 80 additions and 22 deletions

View File

@@ -1,34 +1,51 @@
package models
import (
"crypto/md5"
"fmt"
"log"
"main/configs"
"main/utils"
"time"
)
type User struct {
ID int `json:"id"`
Name string `json:"name"`
Email string `json:"email"`
Password string `json:"password"`
Slat string `json:"slat"`
CreatedAt string `json:"created_at"`
UpdatedAt string `json:"updated_at"`
password string
slat string
}
func (user *User) Create() error {
func (user *User) Create(name, email, password string) error {
if name == "" || email == "" || password == "" {
return fmt.Errorf("name, email and password can not be empty")
}
user.slat = utils.RandomString(16)
user.password = fmt.Sprintf("%x", md5.Sum([]byte(password+user.slat)))
user.Name = name
user.Email = email
user.CreatedAt = time.Now().Format("2006-01-02 15:04:05")
user.UpdatedAt = user.CreatedAt
db, err := configs.GetDB()
if err != nil {
log.Println(err)
return err
}
defer db.Close()
stmt, err := db.Prepare("INSERT INTO users(name, email, created_at, updated_at) values(?, ?, ?, ?)")
fmt.Println(user)
stmt, err := db.Prepare("INSERT INTO users(name, email, password, slat, created_at, updated_at) values(?, ?, ?, ?, ?, ?)")
if err != nil {
log.Println(err)
return err
}
defer stmt.Close()
result, err := stmt.Exec(user.Name, user.Email, user.CreatedAt, user.UpdatedAt)
result, err := stmt.Exec(user.Name, user.Email, user.password, user.slat, user.CreatedAt, user.UpdatedAt)
if err != nil {
log.Println(err)
return err