Java筆記-模擬QQ三方登錄（單點登錄2.0）

本次例子模擬這樣的情況：

一共有2個spring boot應用：

一個是某公司的web，第二個是QQ服務端。

某公司的應用爲127.0.0.1:8082

QQ服務端爲127.0.0.1:8081

演示如下：

點擊使用QQ登錄：

在彈出的界面輸入數據點擊登錄後：

關鍵代碼如下：

某公司web端：

MyController.java

package cn.it1995.login.controller;

import cn.it1995.login.util.CookieUtil;
import cn.it1995.login.util.Result;
import com.alibaba.fastjson.JSON;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.ResponseEntity;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.client.RestTemplate;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;


@RestController
public class MyController {

    @Autowired
    private RestTemplate restTemplate;

    private static final String USER_KEY="user_key";

    private ConcurrentMap<String, Object> user = new ConcurrentHashMap<>();

    @GetMapping("/getUser")
    public Object getUser(HttpServletRequest request, HttpServletResponse response){

        String loginCookie = CookieUtil.getLoginCookie(request, response);
        Object o = user.get(loginCookie);
        return Result.success(o);
    }

    @PostMapping("/loginByQQ")
    public Object loginByQQ(String token, HttpServletResponse response, HttpServletRequest request){

        MultiValueMap<String, Object> paramMap = new LinkedMultiValueMap();
        paramMap.add("token", token);
        ResponseEntity<Object> objectResponseEntity = restTemplate.postForEntity("http://127.0.0.7:8081/getLoginInfo", paramMap, Object.class);
        Object body = objectResponseEntity.getBody();
        String uuid = CookieUtil.setLoginCookie(request, response);

        //json標準化
        String newJson = body.toString().replace("=", ":");
        System.out.println(newJson);

        Map map = JSON.parseObject(newJson, Map.class);
        Map data = JSON.parseObject(map.get("data").toString(), Map.class);
        user.put(uuid, data);
        return Result.success();
    }
}

QQ服務端：

UserController.java

package cn.it1995.qqServer.controller;

import cn.it1995.qqServer.util.JwtUtil;
import cn.it1995.qqServer.util.Result;
import com.alibaba.fastjson.JSON;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.*;

import java.util.HashMap;
import java.util.Map;
import java.util.UUID;


@Controller
public class UserController {

    @ResponseBody
    @RequestMapping("/login")
    public Object login(@RequestParam("username") String username,
                        @RequestParam("password") String password){

        Map<String, Object> map = new HashMap<>();
        map.put("username", username);
        map.put("password", password);
        String jwt = JwtUtil.createJWT(UUID.randomUUID().toString(), JSON.toJSONString(map), 3600 * 24);
        return jwt;
    }

    @ResponseBody
    @RequestMapping("/getLoginInfo")
    public Object getLoginInfo(String token){

        String subject = JwtUtil.parseJWT(token).getSubject();
        return Result.success().data(subject);
    }

}

項目打包下載地址：

https://github.com/fengfanchen/Java/tree/master/SSODemo