package service import ( "XT_New/models" "XT_New/utils" "encoding/json" "github.com/jinzhu/gorm" "strconv" "time" ) func IsSuperAdmin(mobile string) bool { var count int readUserDb.Model(&models.AdminUser{}).Where("mobile = ? AND is_super_admin = 1 AND status = 1", mobile).Count(&count) return count == 1 } func GetValidAdminUserByMobileReturnErr(mobile string) (*models.AdminUser, error) { var user models.AdminUser err := readUserDb.Where("mobile = ? AND status = 1", mobile).First(&user).Error if err != nil { if err == gorm.ErrRecordNotFound { return nil, nil } else { return nil, err } } return &user, nil } func DidAdminUserCreateOrg(adminUserID int64) (bool, error) { var count int err := readUserDb.Model(&models.Org{}).Where("creator = ? AND status <> 0", adminUserID).Count(&count).Error return count > 0, err } func DidAdminUserOrgCreateApp(adminUserID int, appType int) (bool, error) { var count int db := readUserDb.Model(&models.OrgApp{}).Where("creator = ? AND status = 1 AND open_status = 1", adminUserID) if appType != 0 { db = db.Where("app_type = ?", appType) } err := db.Count(&count).Error return count > 0, err } func GetAdminUserLastLoginLog(adminUserID int, appType int) (*models.AdminUserLoginLog, error) { var record models.AdminUserLoginLog var err error if appType > 0 { err = readUserDb.Last(&record, "admin_user_id = ? AND operate_type <> 2 AND app_type = ?", adminUserID, appType).Error } else { err = readUserDb.Last(&record, "admin_user_id = ? AND operate_type <> 2", adminUserID).Error } if err == gorm.ErrRecordNotFound { return nil, nil } return &record, err } func GetAdminUserPrioritizedAppType(adminUserID int) (int8, error) { rows, err := readUserDb.Raw("SELECT app.app_type FROM sgj_user_org_app AS app, sgj_user_admin_role AS r WHERE r.admin_user_id = ? AND r.app_id = app.id AND app.status = 1 AND app.open_status = 1 AND r.status = 1 ORDER BY app.org_id ASC, app.app_type ASC LIMIT 1;", adminUserID).Rows() defer rows.Close() if err != nil { return 0, err } else { if rows.Next() { var appType int8 rows.Scan(&appType) return appType, nil } else { return 0, nil } } } func GetAdminUserAllOrgWithUID(adminUserID int, isSuperAdmin bool) ([]*models.Org, error) { if isSuperAdmin { var org models.Org err := readUserDb.Preload("OrgGallery", "status = 1").Where("creator = ? AND status <> 0", adminUserID).First(&org).Error if err != nil { return nil, err } return []*models.Org{&org}, nil } else { rows, err := readUserDb.Raw("SELECT org.*, COUNT(DISTINCT org.id) FROM sgj_user_admin_role AS u_r, sgj_user_org AS org WHERE u_r.admin_user_id = ? AND u_r.org_id = org.id AND u_r.status = 1 AND org.status <> 0 GROUP BY org.id ORDER BY org.id;", adminUserID).Rows() defer rows.Close() if err != nil { return nil, err } else { orgs := make([]*models.Org, 0) for rows.Next() { var org models.Org if scanErr := readUserDb.Preload("OrgGallery", "status = 1").ScanRows(rows, &org); scanErr != nil { return nil, scanErr } else { orgs = append(orgs, &org) } } return orgs, nil } } } func GetAdminUserAllOrgApp(adminUserID int, orgID int) ([]*models.OrgApp, error) { rows, err := readUserDb.Raw("SELECT app.* from sgj_user_admin_role AS u_r, sgj_user_org_app AS app WHERE u_r.admin_user_id = ? AND u_r.org_id = ? AND u_r.app_id = app.id AND u_r.status = 1 AND app.status = 1 ORDER BY app_type ASC;", adminUserID, orgID).Rows() defer rows.Close() if err != nil { return nil, err } else { apps := make([]*models.OrgApp, 0) for rows.Next() { var app models.OrgApp if scanErr := readUserDb.ScanRows(rows, &app); scanErr != nil { return nil, scanErr } else { apps = append(apps, &app) } } return apps, nil } } func InsertLoginLog(loginLog *models.AdminUserLoginLog) error { tx := writeUserDb.Begin() if err := tx.Create(loginLog).Error; err != nil { tx.Rollback() return err } return tx.Commit().Error } //func GetAppRole(adminUserID int, orgID int, appID int) (*models.App_Role, error) { // var appRole models.App_Role // err := readUserDb.Where("admin_user_id = ? AND org_id = ? AND app_id = ? AND status = 1", adminUserID, orgID, appID).First(&appRole).Error // if err != nil { // return nil, err // } // return &appRole, nil //} // 获取最近被创建的血透系统管理员角色(AppType为3) func GetLastXTAdminRole(adminUserID int64, appType int) (*models.App_Role, error) { var appRole models.App_Role // select a_r.* from sgj_user_admin_role as a_r // join sgj_user_admin as a on a.id = a_r.admin_user_id // join sgj_user_org_app as app on app.id = a_r.app_id and app.status = 1 and app.app_type = 3 // where a_r.status = 1 and a_r.admin_user_id = 380 err := readUserDb.Table("sgj_user_admin_role"). Select("sgj_user_admin_role.*"). Joins("join sgj_user_admin as a on a.id = sgj_user_admin_role.admin_user_id"). Joins("join sgj_user_org_app as app on app.id = sgj_user_admin_role.app_id and app.status = 1 and app.app_type = ?", appType). Where("sgj_user_admin_role.status = 1 and sgj_user_admin_role.admin_user_id = ?", adminUserID). Order("sgj_user_admin_role.id desc"). First(&appRole). Error if err != nil { if err == gorm.ErrRecordNotFound { return nil, nil } else { return nil, err } } return &appRole, nil } func GetOrgById(orgID int64) (*models.Org, error) { var org models.Org err := readUserDb.Model(&models.Org{}).Where("id = ?", orgID).First(&org).Error if err != nil { if err == gorm.ErrRecordNotFound { return nil, nil } else { return nil, err } } return &org, nil } func GetAdminUserPassWord(admin_user_id int64) (models.SgjUserAdmin, error) { userAdmin := models.SgjUserAdmin{} err = XTReadDB().Where("id = ?", admin_user_id).Find(&userAdmin).Error return userAdmin, err } func GetAppById(appID int64) (*models.OrgApp, error) { var app models.OrgApp err := readUserDb.Model(&models.OrgApp{}).Where("id = ?", appID).First(&app).Error if err != nil { if err == gorm.ErrRecordNotFound { return nil, nil } else { return nil, err } } return &app, nil } func GetOrgServeSubscibe(orgID int64) (*models.ServeSubscibe, error) { var model models.ServeSubscibe err := readUserDb.Model(&models.ServeSubscibe{}).Where("org_id = ? AND status = 1", orgID).First(&model).Error if err != nil { if err == gorm.ErrRecordNotFound { return nil, nil } else { return nil, err } } return &model, nil } func GetAppRoleById(id int64) (*models.App_Role, error) { var model models.App_Role err := readUserDb.Model(&models.App_Role{}).Where("id = ? AND status = 1", id).First(&model).Error if err != nil { if err == gorm.ErrRecordNotFound { return nil, nil } else { return nil, err } } return &model, nil } func IsMobileRegister(mobile string) bool { var count int err := readUserDb. Model(&models.AdminUser{}). Where("mobile = ?", mobile). Count(&count). Error if err != nil { utils.ErrorLog("判断手机号是否被注册时失败: %v", err) return true } return count > 0 } func FindAppRoleById(id int64) (*models.App_Role, error) { var model models.App_Role redis := RedisClient() defer redis.Close() // cur_date := time.Now().Format("2006-01-02") key := "user_admin_role:" + strconv.FormatInt(id, 10) role_str, _ := redis.Get(key).Result() if len(role_str) == 0 { //没有到缓存数据,从数据库中获取数据,进行缓存到redis err := readUserDb.Model(&models.App_Role{}).Where("id = ? ", id).First(&model).Error if err != nil { if err == gorm.ErrRecordNotFound { return nil, nil } else { return nil, err } } else { if model.Id > 0 { //缓存数据 role_json, err := json.Marshal(model) if err == nil { redis.Set(key, role_json, time.Second*60*60*18) } } return &model, nil } } else { //缓存数据了数据,将redis缓存的json字符串转为map if err := json.Unmarshal([]byte(role_str), &model); err == nil { } else { } return &model, nil } }