1/trading-go/model/user.go

package model

import (
	"fmt"
	"trading-go/common"
	"trading-go/response"
	"trading-go/util"
)

type User struct {
	Uid    uint   `db:"uid"`
	Vid    string `db:"vid"`
	Name   string `db:"name"`
	Avatar string `db:"avatar"`
	Phone  string `db:"phone"`
}

type UserJson struct {
	Vid    string `json:"vid"`
	Name   string `json:"name"`
	Avatar string `json:"avatar"`
	Phone  string `json:"phone"`
}

func (u User) Register() error {
	db := common.DB
	sqlStr := "INSERT INTO users(uid, vid,name, avatar, phone) values (:uid, :vid, :name, :avatar, :phone)"
	exec, err := db.NamedExec(sqlStr, u)
	if err != nil {
		return err
	}
	affected, err := exec.RowsAffected()
	if err != nil {
		return err
	}
	if affected != 1 {
		return util.InsertFailError
	}
	return nil
}

func (u User) Login() (err error, nu User) {
	db := common.DB
	sqlStr := "SELECT * FROM users WHERE vid = ?"
	err = db.Get(&nu, sqlStr, u.Vid)
	return
}

// SPhone 查找电话号码
func (u User) SPhone() error {
	var nu User
	db := common.DB
	fmt.Println(u.Phone)
	sqlStr := "SELECT * FROM users WHERE phone = ?"
	err := db.Get(&nu, sqlStr, u.Phone)
	if err != nil && err.Error() == "sql: no rows in result set" {
		return nil
	} else if err != nil {
		return err
	}
	return util.PhoneBeUsed
}

// Modify 修改信息
func (u User) Modify() error {
	db := common.DB
	sqlStr := "UPDATE users set name = ?,avatar = ?, phone = ? WHERE uid = ?"
	exec, err := db.Exec(sqlStr, u.Name, u.Avatar, u.Phone, u.Uid)
	if err != nil {
		return err
	}
	affected, err := exec.RowsAffected()
	if err != nil {
		return err
	}
	if affected != 1 {
		return util.UpdateFailError
	}
	return nil
}

// Info 获取用户信息
func (u User) Info() (nu User, err error) {
	db := common.DB
	sqlStr := "SELECT * FROM users WHERE uid = ?"
	err = db.Get(&nu, sqlStr, u.Uid)
	return
}

func (u User) GetPaged(page, pageSize int) (rsp response.PageResponse, err error) {
	var users []User
	db := common.DB
	sqlStr := "SELECT * FROM users LIMIT ?,?"
	rsp.Page = page
	rsp.PageSize = pageSize
	err = db.Select(&users, sqlStr, (page-1)*pageSize, pageSize)
	rsp.Data = users
	return
}

func (u User) GetCnt() (cnt int, err error) {
	db := common.DB
	sqlStr := "SELECT COUNT(1) FROM users"
	err = db.Get(&cnt, sqlStr)
	return
}