springboot学习笔记-3 整合redis&mongodb

@Entity
@Table(name="user")
public class User implements Serializable {
    @Id
    @GeneratedValue(strategy=GenerationType.AUTO)
    private Long id;
    private String name;
    @DateTimeFormat(pattern="yyyy-MM-dd HH:mm:ss")
    private Date createDate;
    
    @JsonBackReference        //防止json的重复引用问题
    private Department department;
    private Set<Role> roles;
    public Long getId() {
        return id;
    }
    public void setId(Long id) {
        this.id = id;
    }
    public String getName() {
        return name;
    }
    public void setName(String name) {
        this.name = name;
    }
    public Date getCreateDate() {
        return createDate;
    }
    public void setCreateDate(Date createDate) {
        this.createDate = createDate;
    }
    public Department getDepartment() {
        return department;
    }
    public void setDepartment(Department department) {
        this.department = department;
    }
    public Set<Role> getRoles() {
        return roles;
    }
    public void setRoles(Set<Role> roles) {
        this.roles = roles;
    }
    @Override
    public String toString() {
        return "User [id=" + id + ", name=" + name + ", createDate=" + createDate + ", department=" + department
                + ", roles=" + roles + "]";
    }
    
}

@ConfigurationProperties("application.properties")
@Configuration
public class RedisConfig {
    @Value("${spring.redis.hostName}")
    private String hostName;
    @Value("${spring.redis.port}")
    private Integer port;

    @Bean
    public RedisConnectionFactory redisConnectionFactory() {
        JedisConnectionFactory cf = new JedisConnectionFactory();  
        cf.setHostName(hostName);  
        cf.setPort(port); 
        cf.afterPropertiesSet();  
        return cf;  
    }
    
    @Bean
    public RedisTemplate<String, String> redisTemplate(RedisConnectionFactory factory) {
        StringRedisTemplate template=new StringRedisTemplate(factory);
        Jackson2JsonRedisSerializer jackson2JsonRedisSerializer=new Jackson2JsonRedisSerializer(Object.class);
        ObjectMapper om=new ObjectMapper();
        om.setVisibility(PropertyAccessor.ALL,JsonAutoDetect.Visibility.ANY);
        om.enableDefaultTyping(ObjectMapper.DefaultTyping.NON_FINAL);
        jackson2JsonRedisSerializer.setObjectMapper(om);
        template.setValueSerializer(jackson2JsonRedisSerializer);
        template.afterPropertiesSet();
        return template;
    }
}

@Repository
public class UserRedis {
    @Autowired
    private RedisTemplate<String, String> redisTemplate;
    
    public void add(String key,User user) {
        Gson gson=new Gson();
        redisTemplate.opsForValue().set(key,gson.toJson(user));
    }
    public void add(String key,List<User> users) {
        Gson gson=new Gson();
        redisTemplate.opsForValue().set(key,gson.toJson(users));
    }
    public User get(String key ) {
        Gson gson=new Gson();
        User user=null;
        String userStr=redisTemplate.opsForValue().get(key);
        if(!StringUtils.isEmpty(userStr))
            user=gson.fromJson(userStr, User.class);
        return user;
    }
    public List<User> getList(String key) {
        Gson gson=new Gson();
        List<User> users=null;
        String listJson=redisTemplate.opsForValue().get(key);
        if(!StringUtils.isEmpty(listJson)) {
            users=gson.fromJson(listJson,new TypeToken<List<User>>(){}.getType());
        }
        return users;
    }
    public void delete(String key) {
        redisTemplate.opsForValue().getOperations().delete(key);
    }
}

@Controller
public class UserController {
    @Autowired
    UserRedis userRedis;
    
    @RequestMapping("/user/testRedisSave")
    public String testRedis() {
        Department department=new Department();
        department.setName("开发部");
        Role role=new Role();
        role.setName("admin");
        User user=new User();
        user.setName("hlhdidi");
        user.setCreateDate(new Date());
        user.setDepartment(department);
        Set<Role> roles=new HashSet<>();
        roles.add(role);
        user.setRoles(roles);
        userRedis.delete(this.getClass().getName()+":username:"+user.getName());
        userRedis.add(this.getClass().getName()+":username:"+user.getName(), user);
        return null;
    }
    @RequestMapping("/user/testRedisGet")
    public String testRedis2() {
        User user=userRedis.get(this.getClass().getName()+":username:hlhdidi");
        System.out.println(user);
        return null;
    }
}

@Configuration
@EnableMongoRepositories(basePackages={"com.hlhdidi.springboot.mongo"})//MongoRepository的扫描包
@PropertySource("classpath:mongo.properties")//注入配置文件属性
public class MongoConfig extends AbstractMongoConfiguration{

    @Autowired
    private Environment env;
    
    @Override
    protected String getDatabaseName() {
        return env.getRequiredProperty("mongo.name");
    }

    @Override
    @Bean
    public Mongo mongo() throws Exception {
        ServerAddress serverAddress=new ServerAddress(env.getRequiredProperty("mongo.host"));
        List<MongoCredential> credentials=new ArrayList<>();
        return new MongoClient(serverAddress, credentials);
    }

}

@Document(collection="user")//配置collection的名称,如果没有将会自动建立对应的Collection
public class SysUser {
    @Id
    private String userId;
    @NotNull @Indexed(unique=true)
    private String username;
    @NotNull
    private String password;
    @NotNull
    private String name;
    @NotNull
    private String email;
    @NotNull
    private Date registrationDate=new Date();
    private Set<String> roles=new HashSet<>();
    public SysUser(){}
    @PersistenceConstructor
    public SysUser(String userId, String username, String password, String name, String email, Date registrationDate,
            Set<String> roles) {
        super();
        this.userId = userId;
        this.username = username;
        this.password = password;
        this.name = name;
        this.email = email;
        this.registrationDate = registrationDate;
        this.roles = roles;
    }
    public String getUserId() {
        return userId;
    }
    public void setUserId(String userId) {
        this.userId = userId;
    }
    public String getUsername() {
        return username;
    }
    public void setUsername(String username) {
        this.username = username;
    }
    public String getPassword() {
        return password;
    }
    public void setPassword(String password) {
        this.password = password;
    }
    public String getName() {
        return name;
    }
    public void setName(String name) {
        this.name = name;
    }
    public String getEmail() {
        return email;
    }
    public void setEmail(String email) {
        this.email = email;
    }
    public Date getRegistrationDate() {
        return registrationDate;
    }
    public void setRegistrationDate(Date registrationDate) {
        this.registrationDate = registrationDate;
    }
    public Set<String> getRoles() {
        return roles;
    }
    public void setRoles(Set<String> roles) {
        this.roles = roles;
    }
    @Override
    public String toString() {
        return "SysUser [userId=" + userId + ", username=" + username + ", password=" + password + ", name=" + name
                + ", email=" + email + ", registrationDate=" + registrationDate + ", roles=" + roles + "]";
    }
    
}

@Repository
public interface SysUserRepository extends MongoRepository<SysUser, String>{

}

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(classes={MongoConfig.class})
@FixMethodOrder
public class MongoTest {
    @Autowired
    SysUserRepository repository;
    
    @Before
    public void setup() {
        Set<String> roles=new HashSet<>();
        roles.add("manage");
        SysUser sysUser=new SysUser("1", "hlhdidi", "123", "xiaohulong", "email@com.cn", new Date(), roles);
        repository.save(sysUser);
    }
    @Test
    public void findAll() {
        List<SysUser> users=repository.findAll();
        for(SysUser user:users) {
            System.out.println(user);
        }
    }
}