login.go 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. package user
  2. import (
  3. "ddpf/model/dbm"
  4. "github.com/tangs-drm/go-tool/util"
  5. "github.com/tangs-drm/go-tool/log"
  6. )
  7. // ExistUser 判断用户是否存在
  8. func ExistUser(account string) (bool, error) {
  9. if len(account) < 1 {
  10. return false, util.Error("uid is empty")
  11. }
  12. var sqlString string = "SELECT ID FROM USER WHERE ACCOUNT = ?;"
  13. rows, err := dbm.Db().Query(sqlString, account)
  14. if err != nil {
  15. log.Error("ExistUser find user by account(%v) error ->(%v)", account, err)
  16. return false, err
  17. }
  18. defer rows.Close()
  19. if !rows.Next() {
  20. return false, nil
  21. }
  22. return true, nil
  23. }
  24. /*
  25. CheckPassword
  26. arg
  27. account:用户账号
  28. password:用户密码
  29. desc
  30. 检查用户账号和密码是否匹配
  31. ret
  32. 0:返回用户信息
  33. 1:返回错误信息
  34. */
  35. func CheckPassword(account, password string) (*User, error) {
  36. var sqlString string = "SELECT ID FROM USER WHERE ACCOUNT = ? AND PASSWORD = ?;"
  37. rows, err := dbm.Db().Query(sqlString, account, password)
  38. if err != nil {
  39. log.Error("CheckPassword check account(%v) error ->(%v)", account, err)
  40. return nil, err
  41. }
  42. defer rows.Close()
  43. if !rows.Next() {
  44. return &User{}, nil
  45. }
  46. var user = &User{}
  47. err = rows.Scan(&user.Id)
  48. if err != nil {
  49. log.Error("CheckPassword check account(%v) scan value error ->(%v)", account, err)
  50. return nil, err
  51. }
  52. return user, nil
  53. }