学习淘淘商城第八十八课（单点登录用户登录--Controller层及测试）

最新推荐文章于 2024-07-30 21:07:22 发布

在京奋斗者

最新推荐文章于 2024-07-30 21:07:22 发布

阅读量2.2k

点赞数 1

分类专栏：淘淘商城淘淘商城

本文链接：https://blog.csdn.net/u012453843/article/details/72934278

版权

淘淘商城同时被 2 个专栏收录

110 篇文章 81 订阅

订阅专栏

淘淘商城

110 篇文章 383 订阅

订阅专栏

首先看接口文档，如下图所示

我们需要把token保存到cookie当中，为了方便的操作cookie，我们特意封装了一个工具类，由于该工具类有可能被多个服务使用，因此我们最好把该工具类放到utils。如下图所示。

可以看到当前引入是有错误的，要解决这个问题就要在taotao-common工程添加对servlet-api的依赖，添加完依赖之后CookieUtils.java就不报错了，如下图所示。

添加的依赖代码如下：

<!-- servlet-api组件 -->
		<dependency>
			<groupId>javax.servlet</groupId>
			<artifactId>servlet-api</artifactId>
			<scope>provided</scope>
		</dependency>

CookieUtils.java类的所有代码如下：

package com.taotao.common.utils;

import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.net.URLEncoder;

import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;


/**
 * 
 * Cookie 工具类
 *
 */
public final class CookieUtils {

    /**
     * 得到Cookie的值, 不编码
     * 
     * @param request
     * @param cookieName
     * @return
     */
    public static String getCookieValue(HttpServletRequest request, String cookieName) {
        return getCookieValue(request, cookieName, false);
    }

    /**
     * 得到Cookie的值,
     * 
     * @param request
     * @param cookieName
     * @return
     */
    public static String getCookieValue(HttpServletRequest request, String cookieName, boolean isDecoder) {
        Cookie[] cookieList = request.getCookies();
        if (cookieList == null || cookieName == null) {
            return null;
        }
        String retValue = null;
        try {
            for (int i = 0; i < cookieList.length; i++) {
                if (cookieList[i].getName().equals(cookieName)) {
                    if (isDecoder) {
                        retValue = URLDecoder.decode(cookieList[i].getValue(), "UTF-8");
                    } else {
                        retValue = cookieList[i].getValue();
                    }
                    break;
                }
            }
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        }
        return retValue;
    }

    /**
     * 得到Cookie的值,
     * 
     * @param request
     * @param cookieName
     * @return
     */
    public static String getCookieValue(HttpServletRequest request, String cookieName, String encodeString) {
        Cookie[] cookieList = request.getCookies();
        if (cookieList == null || cookieName == null) {
            return null;
        }
        String retValue = null;
        try {
            for (int i = 0; i < cookieList.length; i++) {
                if (cookieList[i].getName().equals(cookieName)) {
                    retValue = URLDecoder.decode(cookieList[i].getValue(), encodeString);
                    break;
                }
            }
        } catch (UnsupportedEncodingException e) {
        	 e.printStackTrace();
        }
        return retValue;
    }

    /**
     * 设置Cookie的值 不设置生效时间默认浏览器关闭即失效,也不编码
     */
    public static void setCookie(HttpServletRequest request, HttpServletResponse response, String cookieName,
            String cookieValue) {
        setCookie(request, response, cookieName, cookieValue, -1);
    }

    /**
     * 设置Cookie的值 在指定时间内生效,但不编码
     */
    public static void setCookie(HttpServletRequest request, HttpServletResponse response, String cookieName,
            String cookieValue, int cookieMaxage) {
        setCookie(request, response, cookieName, cookieValue, cookieMaxage, false);
    }

    /**
     * 设置Cookie的值 不设置生效时间,但编码
     */
    public static void setCookie(HttpServletRequest request, HttpServletResponse response, String cookieName,
            String cookieValue, boolean isEncode) {
        setCookie(request, response, cookieName, cookieValue, -1, isEncode);
    }

    /**
     * 设置Cookie的值 在指定时间内生效, 编码参数
     */
    public static void setCookie(HttpServletRequest request, HttpServletResponse response, String cookieName,
            String cookieValue, int cookieMaxage, boolean isEncode) {
        doSetCookie(request, response, cookieName, cookieValue, cookieMaxage, isEncode);
    }

    /**
     * 设置Cookie的值 在指定时间内生效, 编码参数(指定编码)
     */
    public static void setCookie(HttpServletRequest request, HttpServletResponse response, String cookieName,
            String cookieValue, int cookieMaxage, String encodeString) {
        doSetCookie(request, response, cookieName, cookieValue, cookieMaxage, encodeString);
    }

    /**
     * 删除Cookie带cookie域名
     */
    public static void deleteCookie(HttpServletRequest request, HttpServletResponse response,
            String cookieName) {
        doSetCookie(request, response, cookieName, "", -1, false);
    }

    /**
     * 设置Cookie的值，并使其在指定时间内生效
     * 
     * @param cookieMaxage cookie生效的最大秒数
     */
    private static final void doSetCookie(HttpServletRequest request, HttpServletResponse response,
            String cookieName, String cookieValue, int cookieMaxage, boolean isEncode) {
        try {
            if (cookieValue == null) {
                cookieValue = "";
            } else if (isEncode) {
                cookieValue = URLEncoder.encode(cookieValue, "utf-8");
            }
            Cookie cookie = new Cookie(cookieName, cookieValue);
            if (cookieMaxage > 0)
                cookie.setMaxAge(cookieMaxage);
            if (null != request) {// 设置域名的cookie
            	String domainName = getDomainName(request);
            	System.out.println(domainName);
                if (!"localhost".equals(domainName)) {
                	cookie.setDomain(domainName);
                }
            }
            cookie.setPath("/");
            response.addCookie(cookie);
        } catch (Exception e) {
        	 e.printStackTrace();
        }
    }

    /**
     * 设置Cookie的值，并使其在指定时间内生效
     * 
     * @param cookieMaxage cookie生效的最大秒数
     */
    private static final void doSetCookie(HttpServletRequest request, HttpServletResponse response,
            String cookieName, String cookieValue, int cookieMaxage, String encodeString) {
        try {
            if (cookieValue == null) {
                cookieValue = "";
            } else {
                cookieValue = URLEncoder.encode(cookieValue, encodeString);
            }
            Cookie cookie = new Cookie(cookieName, cookieValue);
            if (cookieMaxage > 0)
                cookie.setMaxAge(cookieMaxage);
            if (null != request) {// 设置域名的cookie
            	String domainName = getDomainName(request);
            	System.out.println(domainName);
                if (!"localhost".equals(domainName)) {
                	cookie.setDomain(domainName);
                }
            }
            cookie.setPath("/");
            response.addCookie(cookie);
        } catch (Exception e) {
        	 e.printStackTrace();
        }
    }

    /**
     * 得到cookie的域名
     */
    private static final String getDomainName(HttpServletRequest request) {
        String domainName = null;
        //举个例子好理解，比如用户要登录，那么request.getRequestURL().toString()得到的值
        //便是http://localhost:8088/user/login
        String serverName = request.getRequestURL().toString();
        if (serverName == null || serverName.equals("")) {
            domainName = "";
        } else {
            serverName = serverName.toLowerCase();//将请求全部转为小写字母
            //将http://去掉，只剩下了localhost:8088/user/login
            serverName = serverName.substring(7);
            //获取第一个"/"的索引号
            final int end = serverName.indexOf("/");
            //substring包含开始0，不包括end，因此我们取到的是localhost:8088
            serverName = serverName.substring(0, end);
            //以"."进行分割，由于"localhost:8088"中没有"."因此得到的字符串数组空的
            final String[] domains = serverName.split("\\.");
            //获取数组的长度，现在是0
            int len = domains.length;
            if (len > 3) {
                // www.xxx.com.cn
                domainName = "." + domains[len - 3] + "." + domains[len - 2] + "." + domains[len - 1];
            } else if (len <= 3 && len > 1) {
                // xxx.com or xxx.cn
                domainName = "." + domains[len - 2] + "." + domains[len - 1];
            } else {
            	//由于数组的长度是0，因此走这个分支，直接把localhost:8088赋值给domainName
                domainName = serverName;
            }
        }
        //domainName的值是localhost:8088，符合不为null并且包含":"
        if (domainName != null && domainName.indexOf(":") > 0) {
        	//按":"分组，得到的数组是["localhost","8088"]
            String[] ary = domainName.split("\\:");
            //取第一个，就是"localhost"
            domainName = ary[0];
        }
        //返回localhost
        return domainName;
    }

}

下面我们在UserController类添加login接口，添加完接口后整个Controller类代码如下

package com.taotao.sso.controller;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.ResponseBody;

import com.taotao.common.pojo.TaotaoResult;
import com.taotao.common.utils.CookieUtils;
import com.taotao.pojo.TbUser;
import com.taotao.sso.service.UserService;

/**
 * 用户处理Controller
 * @author xxxxxxxx
 *
 */
@Controller
public class UserController {
    @Autowired
	private UserService userService;
    
    @Value("${TOKEN_KEY}")
    private String TOKEN_KEY;
    
    @RequestMapping("/user/check/{param}/{type}")
    @ResponseBody
    public TaotaoResult checkUserData(@PathVariable String param,@PathVariable Integer type){
    	TaotaoResult result = userService.checkUserData(param, type);
    	return result;
    }
    
    @RequestMapping(value = "/user/register",method = RequestMethod.POST)
    @ResponseBody
    public TaotaoResult register(TbUser tbUser){
    	TaotaoResult register = userService.register(tbUser);
    	return register;
    }
    
    @RequestMapping(value = "/user/login",method = RequestMethod.POST)
    @ResponseBody
    public TaotaoResult login(String username,String password,
    		HttpServletRequest request,HttpServletResponse response){
    	TaotaoResult result = userService.login(username, password);
    	//把token写入到cookie
    	CookieUtils.setCookie(request, response, TOKEN_KEY, result.getData().toString());
    	return result;
    }
}

login接口中用到了配置文件，配置文件

下面我们来测试，由于在taotao-common中添加了工具类，因此需要重新打包taotao-common工程到本地maven仓库，由于在taotao-sso-interface添加了接口，因此我们需要把taotao-sso工程重新打包到本地maven仓库。

在启动工程前要保证zookeeper和redis服务器处于启动状态！！！

redis服务器的IP及端口要与jedis配置文件一致，如下图所示。

打包完后，我们启动taotao-sso工程和taotao-sso-web工程，启动之后，我们还是使用RestClient3.5来测试。我们修改下URL地址将/user/register修改为/user/login，然后点击下面那个圈住的图标。

可以看到弹出一个小框，我们点击"是"

登录我们只用用户名和密码，不用电话和邮箱，因此我们把电话和邮箱两个参数去掉，选中email和phone两行，右键点击"Remove Selected"。

去掉email和phone之后，双击value那一个框进入编辑状态，修改username的值为"zhangsan6"，如下图，点击"Generate"。

点击执行按钮，可以看到返回的结果状态码是200，data中存放的是token的值。说明登录成功了，同时也说明我们的登录接口正确。

登录成功了，我们利用redis-desktop-manager工具来查看token是否已经被保存起来了（关于redis-server的使用大家可以参考http://blog.csdn.net/u012453843/article/details/70258026这篇博客进行学习），如下图所示，发现已经被保存起来了。