spider_creater/user.go

package main

import qu "jygit.jydev.jianyu360.cn/data_processing/common_utils"

type UserInfo struct {
	ID       string   `json:"_id"`
	Username string   `json:"s_name"`
	Fullname string   `json:"s_fullname"`
	Email    string   `json:"s_email"`
	Auth     int      `json:"i_auth"`
	Scope    int      `json:"i_scope"`
	Identity int      `json:"i_identity"`
	Ids      []string `json:"ids"`
}

var (
	User *UserInfo //user对象
)

// ServerActionUserLogin 登录
func (a *App) ServerActionUserLogin(param map[string]interface{}) map[string]interface{} {
	qu.Debug("param---", param)
	User = &UserInfo{}
	//User = map[string]interface{}{}
	getResult(param, User, "login")
	qu.Debug("user:", *User)
	if User.ID == "" {
		return map[string]interface{}{
			"msg":  "登录失败",
			"err":  0,
			"data": nil,
		}
	}
	return map[string]interface{}{
		"msg":  "",
		"err":  1,
		"data": User,
	}
}

// ServerActionCheckLogin 检查是否登录
func (a *App) ServerActionCheckLogin() map[string]interface{} {
	if User != nil && User.ID != "" {
		return map[string]interface{}{
			"msg":  "",
			"err":  1,
			"data": User,
		}
	}
	return map[string]interface{}{
		"msg":  "",
		"err":  0,
		"data": nil,
	}
}

// ServerActionUserLogout 退出登录
func (a *App) ServerActionUserLogout() map[string]interface{} {
	User = &UserInfo{}
	return map[string]interface{}{
		"msg":  "退出成功",
		"err":  1,
		"data": nil,
	}
}

// ServerActionGetModifyUsers 获取爬虫开发人员列表
func (a *App) ServerActionGetModifyUsers() *Result {
	data := &Result{}
	if User != nil && User.Auth > 1 { //禁止开发查询
		getResult(nil, data, "getmodifyusers")
		if len(User.Ids) > 0 && User.Identity == 0 { //外包审核员或管理员只能查看外包开发人员信息
			resultUsers := []map[string]interface{}{}
			if data.Data != nil {
				for _, user := range data.Data.List {
					userid := qu.ObjToString(user["_id"])
					for _, id := range User.Ids {
						if userid == id {
							resultUsers = append(resultUsers, user)
							break
						}
					}
				}
				data.Data.List = resultUsers
				data.Data.Total = len(resultUsers)
			}
		}
	} else {
		data.Msg = "查询开发列表失败，权限不够！"
	}
	return data
}