基于redis实现队列

最新推荐文章于 2024-08-13 22:45:00 发布

流离岁月

最新推荐文章于 2024-08-13 22:45:00 发布

阅读量3.5k

点赞数

分类专栏： redis 文章标签： redis java 数据库

本文链接：https://blog.csdn.net/liupantao/article/details/121670195

版权

redis 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

该博客介绍了如何利用Redis实现数据的先入先出（FIFO）策略，通过`lpush`和`lpop`命令进行队列操作。同时，展示了在并发环境下使用`setNX`命令进行加锁，确保队列操作的正确性。此外，还提供了Redis队列模板类的详细代码实现，包括批量插入、获取指定长度的元素、获取队列长度等功能。

摘要由CSDN通过智能技术生成

基于redis实现队列

一、说明
基于redis实现队列，实现数据先入先出策略。
在这里插入图片描述

二、基本命令

lpush key val (数据存在list头部，队列尾部)

查看值
lpop key （取list第一个元素，list尾部,队列头部）

在这里插入图片描述

setNX key value (命令在指定的 key 不存在时，为 key 设置指定的值。) 用来加锁

在这里插入图片描述

三、代码


/**
 * Redis队列 实现类
 * 
 */
public class RedisQueueTemplate {

	private Log logger = LogFactory.getLog(RedisQueueTemplate.class);

	private StringRedisTemplate redisTemplate;

	private static final String QUESUE_LOCK_KEY_SUFFIX = "_gsoft_lock";

	public RedisQueueTemplate(StringRedisTemplate redisTemplate) {
		if (redisTemplate == null) {
			throw new RuntimeException("redisTemplate can not be null ");
		}
		this.redisTemplate = redisTemplate;
	}

	public long rPush(final String queueName, final String value) {
		return redisTemplate.execute(new RedisCallback<Long>() {
			@Override
			public Long doInRedis(RedisConnection connection) throws DataAccessException {
				return connection.rPush(redisTemplate.getStringSerializer().serialize((queueName)),
						redisTemplate.getStringSerializer().serialize(value));
			}
		});
	}

	public String lPop(final String queueName) {
		return redisTemplate.execute(new RedisCallback<String>() {
			@Override
			public String doInRedis(RedisConnection connection) throws DataAccessException {
				byte[] value = connection.lPop(redisTemplate.getStringSerializer().serialize((queueName)));
				return redisTemplate.getStringSerializer().deserialize(value);
			}
		});
	}

	public long rPush(final String queueName, final List<String> values) {
		if (values != null && values.size() > 0) {
			return redisTemplate.execute(new RedisCallback<Long>() {
				@Override
				public Long doInRedis(RedisConnection connection) throws DataAccessException {
					List<byte[]> valuesByte = new ArrayList<>();
					for (String value : values) {
						if (value != null) {
							valuesByte.add(redisTemplate.getStringSerializer().serialize(value));
						}
					}
					return connection.rPush(redisTemplate.getStringSerializer().serialize((queueName)),
							valuesByte.toArray(new byte[valuesByte.size()][]));
				}
			});
		}
		return 0L;
	}

	public List<String> lPop(final String queueName, final int length) {
		return redisTemplate.execute(new RedisCallback<List<String>>() {
			// long s = System.currentTimeMillis();
			@Override
			public List<String> doInRedis(RedisConnection connection) throws DataAccessException {
				List<String> values = new ArrayList<>();
				byte[] key = redisTemplate.getStringSerializer().serialize((queueName));
				byte[] lockKey = redisTemplate.getStringSerializer().serialize((queueName) + QUESUE_LOCK_KEY_SUFFIX);
				try {
					while (!connection.setNX(lockKey, key)) {// 队列加锁
						try {
							Thread.sleep(10);
						} catch (Exception e) {
							logger.warn(" RedisQueueTemplate sleep Exception .....");
						}
					}
					connection.expire(lockKey, 5);// 设计锁的超时时间

					List<byte[]> valuesByte = connection.lRange(key, 0, length - 1);
					connection.lTrim(key, valuesByte.size(), -1);

					for (byte[] valueByte : valuesByte) {
						values.add(redisTemplate.getStringSerializer().deserialize(valueByte));
					}

				} finally {
					connection.del(lockKey);
				}
				// long s2 = System.currentTimeMillis();
				// if (s2 - 1000 > s) {
				// logger.warn("lPop WARN Time consuming: " + (s2 - 1000 > s));
				// }
				return values;
			}
		});
	}

	public List<String> lRange(final String queueName, final int length) {
		return redisTemplate.execute(new RedisCallback<List<String>>() {
			@Override
			public List<String> doInRedis(RedisConnection connection) throws DataAccessException {
				List<String> values = new ArrayList<>();
				byte[] key = redisTemplate.getStringSerializer().serialize((queueName));
				List<byte[]> valuesByte = connection.lRange(key, 0, length - 1);
				for (byte[] valueByte : valuesByte) {
					values.add(redisTemplate.getStringSerializer().deserialize(valueByte));
				}
				return values;
			}
		});
	}

	public void lRem(final String queueName, final int length) {
		redisTemplate.execute(new RedisCallback<Long>() {
			@Override
			public Long doInRedis(RedisConnection connection) throws DataAccessException {
				byte[] key = redisTemplate.getStringSerializer().serialize((queueName));
				connection.lTrim(key, length, -1);
				return 1L;
			}
		});
	}

	public long lLen(final String queueName) {
		return redisTemplate.execute(new RedisCallback<Long>() {
			@Override
			public Long doInRedis(RedisConnection connection) throws DataAccessException {
				byte[] key = redisTemplate.getStringSerializer().serialize((queueName));
				return connection.lLen(key);
			}
		});
	}

}

四、注册bean

/**
 * redis配置
 * 
 * 
 */
@Configuration
public class RedisConfig {

	@Autowired
    private StringRedisTemplate stringRedisTemplate;
	
	@Bean
	public RedisQueueTemplate redisQueueTemplate() {
		return new RedisQueueTemplate(stringRedisTemplate);
	}
}