zz_backend/zhongzheng-api/src/main/java/com/zhongzheng/controller/wx/WxLoginController.java

package com.zhongzheng.controller.wx;

import com.github.xiaoymin.knife4j.annotations.ApiSupport;
import com.zhongzheng.common.constant.Constants;
import com.zhongzheng.common.core.domain.AjaxResult;
import com.zhongzheng.common.core.domain.entity.SysMenu;
import com.zhongzheng.common.core.domain.entity.SysUser;
import com.zhongzheng.common.core.domain.model.LoginBody;
import com.zhongzheng.common.core.domain.model.LoginUser;
import com.zhongzheng.common.utils.ServletUtils;
import com.zhongzheng.framework.web.service.*;
import com.zhongzheng.modules.system.service.ISysMenuService;
import com.zhongzheng.modules.user.entity.ClientLoginUser;
import com.zhongzheng.modules.user.service.IUserService;
import com.zhongzheng.modules.wx.bo.WxLoginBody;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

@Api(tags ="微信登录用户操作管理")
@ApiSupport(order = 2)
@RestController
public class WxLoginController
{
    @Autowired
    private WxLoginService wxLoginService;

    @Autowired
    private WxTokenService wxTokenService;
    /**
     * 登录方法
     *
     * @param loginBody 登录信息
     * @return 结果
     */
    @ApiOperation("登录")
    @PostMapping("/login")
    public AjaxResult login(@RequestBody WxLoginBody loginBody)
    {
        String union = "cba";
        String token = wxLoginService.login(loginBody);
        Map<String,Object> map = new HashMap<>();
        map.put(Constants.TOKEN, token);
        map.put("union_id", union);
        return AjaxResult.success(map);
    }

    /**
     * 获取用户信息
     *
     * @return 用户信息
     */
    @ApiOperation("登录用户信息")
    @GetMapping("getInfo")
    public AjaxResult getInfo()
    {
        ClientLoginUser loginUser = wxTokenService.getLoginUser(ServletUtils.getRequest());
        Map<String,Object> map = new HashMap<>();
        map.put("union_id", loginUser.getUser().getUnionId());
        return AjaxResult.success(map);
    }

}