Spring Cloud -- 使用redis+token实现登录

1.整合redis:

1) 添加redis依赖并在要调用这个方法的工程配置redis:

spring: 
  redis:
    host: localhost

        <dependency>
            <groupId>org.springframework.data</groupId>
            <artifactId>spring-data-redis</artifactId>
            <version>RELEASE</version>
        </dependency>

2) 新建BaseRedisService:

package com.id0304.common.redis;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;

import java.util.concurrent.TimeUnit;
//详细的redis操作都可以在这里封装,这里写了几个作为参考
@Component
public class BaseRedisService {

    @Autowired
    private StringRedisTemplate stringRedisTemplate;
    
    /**
     * 添加redis信息
     * @param key
     */
    public void setString(String key, String value) {
        set(key, value, null);
    }

	//传入键值对和过期时间
    public void setString(String key, String value, Long timeOut) {
        set(key, value, timeOut);
    }

	//传入object对象
    private void set(String key, Object value, Long timeOut) {
        if (value != null) {
            if (value instanceof String) {
                String setValue = (String) value;
                stringRedisTemplate.opsForValue().set(key, setValue);
            }
            //设置有效期
            if (timeOut != null) {
                stringRedisTemplate.expire(key, timeOut, TimeUnit.SECONDS);
            }
        }
    }

    /**
     * 使用key查找redis信息
     * @param key
     */
    public String get(String key){
        return stringRedisTemplate.opsForValue().get(key);
    }

    /**
     * 使用key删除redis信息
     * @param key
     */
    public void delete(String key){
        stringRedisTemplate.delete(key);
    }

}

3) 使用redis:

@Autowired
    private BaseRedisService baseRedisService;

2.创建TokenUtils:

package com.id0304.utils;

import org.springframework.stereotype.Component;

import java.util.UUID;
//简单示例
public class TokenUtils {
    public static String getToken(){
        return UUID.randomUUID().toString();
    }
}

3.实现登录与查询

1) UserService:

package com.id0304.api.service.impl;

import com.id0304.api.service.UserService;
import com.id0304.common.api.BaseApiService;
import com.id0304.common.redis.BaseRedisService;
import com.id0304.entity.UserEntity;
import com.id0304.manage.UserServiceManage;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;

import java.util.Map;

@Slf4j
@RestController
public class UserServiceImpl extends BaseApiService implements UserService{
    @Autowired
    private UserServiceManage userServiceManage;

    @Autowired
    private BaseRedisService baseRedisService;
	
	//登录并存入redis
    @Override
    public Map<String, Object> login(@RequestBody UserEntity userEntity) {
        return userServiceManage.login(userEntity);
    }

    //使用token查找用户信息
    @Override
    public Map<String, Object> getUserByToken(@RequestParam("token") String token) {
        if(StringUtils.isEmpty(token)){
            return setParamError("token不能为空");
        }
        return userServiceManage.getUser(token);
    }

}

2) UserServiceManage:

package com.id0304.manage.impl;

import com.alibaba.fastjson.JSONObject;
import com.id0304.common.api.BaseApiService;
import com.id0304.common.redis.BaseRedisService;
import com.id0304.constants.Constants;
import com.id0304.constants.DBTableName;
import com.id0304.constants.MQInterfaceType;
import com.id0304.dao.UserDao;
import com.id0304.entity.UserEntity;
import com.id0304.manage.UserServiceManage;
import com.id0304.mq.producer.RegisterMailboxProducer;
import com.id0304.utils.DateUtils;
import com.id0304.utils.Md5Util;
import com.id0304.utils.TokenUtils;
import lombok.extern.slf4j.Slf4j;
import org.apache.activemq.command.ActiveMQQueue;
import org.apache.commons.lang.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import javax.jms.Destination;
import java.util.Map;


@Service
@Slf4j
public class UserServiceManageImpl extends BaseApiService implements UserServiceManage {

    @Autowired
    private UserDao userDao;

    @Autowired
    private BaseRedisService baseRedisService;

    //实现登录业务处理
    @Override
    public Map<String, Object> login(UserEntity userEntity) {
        String phone = userEntity.getPhone();
        String password = userEntity.getPassword();
        //往数据库里查询数据
        UserEntity user = userDao.getUserByPhoneAndPwd(phone, MD5PasswordSalt(phone, password));
        if (user == null) {
            return setResultError("用户名或密码错误");
        }
        
        //生成对应的token
        String token = TokenUtils.getToken();
        Long id = user.getId();
        //存放在redis中,key为自定义token令牌,value为用户id
        baseRedisService.setString(token, id+"", Constants.USER_TOKEN_TERMVALIDITY);
        //返回token
        return setResultSuccessData(token);
    }

    //根据令牌查找user
    @Override
    public Map<String, Object> getUser(String token) {
        //从redis中根据key查找到用户id
        String id = baseRedisService.get(token);
        if(StringUtils.isEmpty(id)){
            return setResultError("用户已过期!");
        }
        UserEntity userEntity = userDao.getUserById(Long.parseLong(id));
        //不要把用户密码返回给客户端
        userEntity.setPassword(null);
        return setResultSuccessData(userEntity);
    }
}

3) UserDao:

package com.id0304.dao;

import com.id0304.common.mybatis.BaseDao;
import com.id0304.entity.UserEntity;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.Select;
import org.springframework.stereotype.Component;

@Component
@Mapper
public interface UserDao extends BaseDao{
    //根据手机号和密码查询用户,首次登录
    @Select("select id,username,password,phone,email,created,updated from mb_user where phone = #{phone} and password = #{password}")
    public UserEntity getUserByPhoneAndPwd(@Param("phone") String phone,@Param("password") String password);

    //根据id查询用户,从redis取出id再去查
    @Select("select id,username,password,phone,email,created,updated from mb_user where id = #{id}")
    public UserEntity getUserById(@Param("id") Long id);

}