miduo_login/src/main/java/com/ljsd/controller/GetUserController.java

package com.ljsd.controller;

import com.alibaba.fastjson.JSONObject;
import com.ljsd.pojo.VerifyParams;
import com.ljsd.redis.RedisKey;
import com.ljsd.service.*;
import com.ljsd.util.AppConstans;
import com.ljsd.util.BaseGlobal;
import com.mongodb.BasicDBObject;
import com.mongodb.DBObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Random;

public class GetUserController extends HttpServlet {
    private final static String _COLLECTION_NAME = "user_info";
    private static final Logger LOGGER = LoggerFactory.getLogger(GetUserController.class);

    public static Map<String, VerifyService> serviceMap = new HashMap<>();
    public static void initHandler(){
        //初始化所有需要的验证方式
        serviceMap.put("MHT", new LdVerifyService("MHT"));
        serviceMap.put("XP", new XPVerifyService("XP"));
        serviceMap.put("CH",new CaohuaVerifyService("CH"));
        serviceMap.put("QUICK", new QuickVerifyService("QUICK"));
        serviceMap.put("QIANYOU", new QiYouVerifyService("huangjiayl&qypay"));
        serviceMap.put("YX", new YxylVerifyService("YX"));
        serviceMap.put("DY", new V3367VerifyService("DY"));
        serviceMap.put("QUICK2", new QuickVerifyService2("QUICK2"));
        serviceMap.put("QUICK3", new QuickVerifyService3("QUICK3"));
        serviceMap.put("YOUGU", new YouGuVerifyService("YG"));
        serviceMap.put("YGXQ", new YouguXiaoqiVerifyService("YGXQ"));
        serviceMap.put("YGDC", new YouguDiscountVerifyService("YGDC"));
        serviceMap.put("XQDC", new XiaoqiDiscountVerifyService("XQDC"));
        serviceMap.put("DUOYOU", new DuoYouVerifyService("DUOYOU"));
        serviceMap.put("AIWAN", new AiWanVerifyService("AIWAN"));
        serviceMap.put("HF", new HFVerifyService("HF"));
        serviceMap.put("FENGTI", new FengTiVerifyService("FENGTI"));
//        serviceMap.put("ANJIU", new AnJiuVerifyService("ANJIU"));
    }


    public GetUserController() {
        super();
    }

    @Override
    public void destroy() {
        super.destroy();
    }


    /**
     * openId 账号id
     * version
     * serverId
     * token
     * platform android、ios  3 跳过验证
     * admin
     * gid pid
     *
     *
     *---verty--
     *  openid 和token校验
     *
     * ---return--
     *
     * uid（openid + serverid + platid）

     *
     */
    @Override
    public void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        DBObject res = new BasicDBObject();
        String openId = request.getParameter("openId");
        if (openId == null || openId.isEmpty()) {
            returnErrorToFront(res,response, "openId is empty");
            return;
        }
        try {
            String banInfo = BaseGlobal.getInstance().redisApp.get(RedisKey.Ban_Open_Id, openId, String.class, -1);
            if (banInfo != null) {
                String[] split = banInfo.split("\\|");
                long currentTimeMillis = System.currentTimeMillis();
                if (Long.parseLong(split[0]) > currentTimeMillis) {
                    returnErrorToFront(res,response,banInfo);
                    LOGGER.info("doGet can not login Ban_Open_Id=>{} openId=>{} banInfo==null ? {}", RedisKey.Ban_Open_Id, openId, banInfo);
                    return;
                } else {
                    BaseGlobal.getInstance().redisApp.del(RedisKey.Ban_Open_Id, openId);
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        String version = request.getParameter("version");
        if (version == null)  {
            version="未知版本号";
        }
        String serverId = request.getParameter("serverId");
        if (serverId == null || serverId.isEmpty()) {
            returnErrorToFront(res,response,"serverId is empty");

            return;
        }
        String token = request.getParameter("token");
        if (token == null || token.isEmpty()) {
            returnErrorToFront(res,response,"token is empty");
            return;
        }
        //平台类型
        String platform = request.getParameter("platform");
        if (platform == null || platform.isEmpty()) {
            returnErrorToFront(res,response,"platform is empty");
            return;
        }
        //平台类型
        String admin = request.getParameter("admin");
        //gid
        String gid = request.getParameter("gid");
        //pid
        String pid = request.getParameter("pid");
        String channel = request.getParameter("channel");
        String sub_channel = request.getParameter("sub_channel");
        /*if (StringUtils.checkIsEmpty(admin)) {
            response.sendError(400, "platform is empety");
            return;
        }*/
        LOGGER.info("the opendId = {},token={},platform={} admin=>{} channel=>{} sub_channel={}",openId,token,platform,admin,channel,sub_channel);
        if(serviceMap == null){
            LOGGER.info("serviceMap is null");
        }
        try {
            //根据前端发过来的种类进行校验
            if(channel==null||channel.equals("")){
                channel="MHT";
            }
            boolean verify = false;
            if (!AppConstans.appsecret.equals(admin) && "MHT".equals(channel)
                    && !"3".equals(platform) && !"1000".equals(sub_channel) && !"20201222".equals(sub_channel)) {
                LdVerifyServiceStr instance = LdVerifyServiceStr.getInstance();
                VerifyParams params = new VerifyParams();
                params.setAppId(instance.properties.getProperty("appId_online"));
                params.setSecretKey(instance.properties.getProperty("secretKey_online"));
                VerifyParams formatParam = instance.getFormatParam(params,request, openId, token);
                String result = instance.verifyFormat(formatParam);
                if (result != null && !result.isEmpty()) {
                    JSONObject jsonObject = JSONObject.parseObject(result);
                    int state = jsonObject.getIntValue("code");
                    if(state != 200){
                        String msg = jsonObject.getString("message");
                        returnErrorToFront(res, response, msg);
                        return;
                    }
                    verify = true;
                }
            } else {
                // 内网或者 游心 开天，不需要验证
                if (AppConstans.appsecret.equals(admin) || "YX_YSL".equals(channel) || "KT".equals(channel)) {
                    verify = true;
                } else {
                    verify = serviceMap.get(channel).verify(response, request, admin, platform, pid, openId, token);
                }
            }
            if(!verify){
                returnErrorToFront(res,response,"校验失败，请重新登录");
                return;
            }
            response.setCharacterEncoding("UTF-8");
            response.setContentType("application/json; charset=utf-8");
            DBObject dbObject = new BasicDBObject();
            dbObject.put("openId", openId);
            dbObject.put("serverId", serverId);
            dbObject.put("platform", platform);

            DBObject serverDBObject = new BasicDBObject();
            serverDBObject.put("server_id", serverId);
            List<DBObject> server_info = BaseGlobal.getInstance().mongoDBPool.find("server_info", serverDBObject);
            if(server_info.size()==0){
                returnErrorToFront(res,response,"服务器信息获取错误");
                return;
            }
            int uid = 0;
            //加分布式锁
            List<DBObject> userInfos = BaseGlobal.getInstance().mongoDBPool.find(_COLLECTION_NAME, dbObject);
            if (userInfos.size() == 0) {
                try{
                boolean getLock=false;
                    while(!getLock){
                        getLock = BaseGlobal.getInstance().redisApp.tryGetDistributedLock(RedisKey.REGISTER_LOCK, "", "get", 800);
                        if(getLock){
                            userInfos = BaseGlobal.getInstance().mongoDBPool.find(_COLLECTION_NAME, dbObject);
                            //双重校验
                            if (userInfos.size() == 0) {
                                DBObject next = server_info.iterator().next();
                                Object register_state = next.get("register_state");
                                if(register_state!=null&& register_state.equals("0")){
                                    returnErrorToFront(res,response,"该服务器已停止注册，请选择其他服务器游戏");
                                    return;
                                }
                                uid = BaseGlobal.getInstance().mongoDBPool.inc("uid") + 10000000;
                                dbObject.put("uid", uid);
                                dbObject.put("_id", uid);
                                userInfos.add(dbObject);
                                if (gid != null && !gid.isEmpty()) {
                                    dbObject.put("gid", gid);
                                }
                                if (pid != null && !pid.isEmpty()) {
                                    dbObject.put("pid", pid);
                                }
                                BaseGlobal.getInstance().mongoDBPool.save(_COLLECTION_NAME, dbObject);
                            }
                        }
                    }
                }finally {
                    BaseGlobal.getInstance().redisApp.releaseDistributedLock(RedisKey.REGISTER_LOCK,"","get");
                }
            }
            Random random = new Random();
            int utoken = Math.abs(random.nextInt());
            uid = (int) userInfos.get(0).get("uid");
            res.put("uid", uid);
            res.put("token", utoken);
            res.put("errorCode", 0);
            BaseGlobal.getInstance().redisApp.set(RedisKey.TOKEN, String.valueOf(uid), utoken, -1, false);
            BaseGlobal.getInstance().redisApp.set(RedisKey.PIDGIDTEMP, String.valueOf(uid), pid+"#"+gid, -1, false);
            PrintWriter out = response.getWriter();
            out.print(res);
            out.flush();
            out.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    private void returnErrorToFront(DBObject res,HttpServletResponse response,String errorReason) throws IOException {
        res.put("errorCode", -1);
        res.put("reason",errorReason);
        PrintWriter out = response.getWriter();
        out.print(res);
        out.flush();
        out.close();
    }


    @Override
    public void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        this.doGet(request, response);

    }

}