mirror of https://github.com/sunface/rust-course
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
1.8 KiB
66 lines
1.8 KiB
package models
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/imdotdev/im.dev/server/pkg/db"
|
|
)
|
|
|
|
type User struct {
|
|
ID string `json:"id"`
|
|
Username string `json:"username"`
|
|
Nickname string `json:"nickname"`
|
|
Avatar string `json:"avatar"`
|
|
Email string `json:"email"`
|
|
Role RoleType `json:"role"`
|
|
|
|
Tagline string `json:"tagline"`
|
|
Cover string `json:"cover"`
|
|
Location string `json:"location"`
|
|
AvailFor string `json:"availFor"`
|
|
About string `json:"about"`
|
|
RawSkills []*Tag `json:"rawSkills"`
|
|
Skills []string `json:"skills"`
|
|
|
|
Website string `json:"website"`
|
|
Twitter string `json:"twitter"`
|
|
Github string `json:"github"`
|
|
Zhihu string `json:"zhihu"`
|
|
Weibo string `json:"weibo"`
|
|
Facebook string `json:"facebook"`
|
|
Stackoverflow string `json:"stackoverflow"`
|
|
|
|
LastSeenAt time.Time `json:"lastSeenAt,omitempty"`
|
|
Created time.Time `json:"created"`
|
|
}
|
|
|
|
func (user *User) Query(id string, username string, email string) error {
|
|
err := db.Conn.QueryRow(`SELECT id,username,role,nickname,email,avatar,last_seen_at,created FROM user WHERE id=? or username=? or email=?`,
|
|
id, username, email).Scan(&user.ID, &user.Username, &user.Role, &user.Nickname, &user.Email, &user.Avatar, &user.LastSeenAt, &user.Created)
|
|
|
|
if user.Avatar == "" {
|
|
user.Avatar = DefaultAvatar
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
type UserSimple struct {
|
|
ID string `json:"id"`
|
|
Username string `json:"username"`
|
|
Nickname string `json:"nickname"`
|
|
Avatar string `json:"avatar"`
|
|
}
|
|
|
|
func (user *UserSimple) Query() error {
|
|
err := db.Conn.QueryRow(`SELECT id,username,nickname,avatar FROM user WHERE id=?`, user.ID).Scan(
|
|
&user.ID, &user.Username, &user.Nickname, &user.Avatar,
|
|
)
|
|
|
|
if user.Avatar == "" {
|
|
user.Avatar = DefaultAvatar
|
|
}
|
|
|
|
return err
|
|
}
|