36 lines
569 B
Go
36 lines
569 B
Go
package repo
|
|
|
|
import (
|
|
"gorm.io/gorm"
|
|
"os/user"
|
|
"work_cation/models"
|
|
"work_cation/pkg/utils"
|
|
)
|
|
|
|
type User struct{}
|
|
|
|
var UserRepo *User
|
|
|
|
func (*User) GetUserInfo(db *gorm.DB) *models.Users {
|
|
var users models.Users
|
|
db.Find(&users)
|
|
ip := utils.Get192Ip()
|
|
if users.Ip != ip {
|
|
current, err := user.Current()
|
|
if err != nil {
|
|
current = &user.User{Name: ip}
|
|
}
|
|
users = models.Users{
|
|
Ip: ip,
|
|
Name: current.Username,
|
|
Avatar: "",
|
|
Cover: "",
|
|
Email: "",
|
|
Phone: "",
|
|
Address: "友嘉",
|
|
}
|
|
db.Create(&users)
|
|
}
|
|
return &users
|
|
}
|