com.github.dennisit.vplus.data.incid.impl.PrimaryIdRedisRepository Maven / Gradle / Ivy
/*--------------------------------------------------------------------------
* Copyright (c) 2009-2020, www.wuyushuo.com All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* Neither the name of the wuyushuo developer nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission.
* Author: [email protected] (tencent qq: 2094998377)
*--------------------------------------------------------------------------
*/
package com.github.dennisit.vplus.data.incid.impl;
import com.github.dennisit.vplus.data.incid.PrimaryIdRepository;
import org.springframework.dao.DataAccessException;
import org.springframework.data.redis.connection.RedisConnection;
import org.springframework.data.redis.core.RedisCallback;
import org.springframework.data.redis.core.RedisTemplate;
public class PrimaryIdRedisRepository implements PrimaryIdRepository {
/**
* redis template
*/
private RedisTemplate redisTemplate;
/**
* @param channel channel type
* @return 自增ID
*/
public long incrementId(final String channel) {
return redisTemplate.execute(new RedisCallback() {
@Override
public Long doInRedis(RedisConnection redisConnection) throws DataAccessException {
byte[] key = String.valueOf(channel).getBytes();
Long id = redisConnection.incr(key);
if (null != id) {
return id;
}
return 0L;
}
});
}
public void setRedisTemplate(RedisTemplate redisTemplate) {
this.redisTemplate = redisTemplate;
}
public RedisTemplate getRedisTemplate() {
return redisTemplate;
}
}