|
|
@@ -42,16 +42,15 @@ type InspectionDao struct {
|
|
|
|
|
|
func (insdao *InspectionDao) getUserFromDB(role int, uid int64) (ChatUser, error) {
|
|
|
var member ChatUser
|
|
|
- var sql string
|
|
|
+ var tableName string
|
|
|
if role == 1 {
|
|
|
- sql = "SELECT realname FROM `go_user` WHERE chat_uid = ?"
|
|
|
+ tableName = "go_user"
|
|
|
} else {
|
|
|
- sql = "SELECT nickname AS realname, avatar FROM `members` WHERE chat_uid=? "
|
|
|
+ tableName = "members"
|
|
|
}
|
|
|
- query := GetTiNodeDbInstance().Debug().Raw(sql, uid).Select("realname")
|
|
|
- err := query.Find(&member).Error
|
|
|
- if err == gorm.ErrRecordNotFound {
|
|
|
- err = nil
|
|
|
+ err := GetTiNodeDbInstance().Debug().Table(tableName).Where("chat_uid=?",uid).Find(&member).Error
|
|
|
+ if err !=nil {
|
|
|
+ return member, err
|
|
|
}
|
|
|
return member, nil
|
|
|
}
|
