net.dreamlu.mica.redis.stream.DefaultRStreamTemplate Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of mica-redis Show documentation
An enhanced toolkit of Spring cloud to simplify development.
There is a newer version: 3.3.2
/*
 * Copyright (c) 2019-2029, Dreamlu 卢春梦 ([email protected] & www.dreamlu.net).
 * 
 * Licensed under the GNU LESSER GENERAL PUBLIC LICENSE 3.0;
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 

 * http://www.gnu.org/licenses/lgpl.html
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package net.dreamlu.mica.redis.stream;

import org.springframework.data.redis.connection.RedisStreamCommands;
import org.springframework.data.redis.connection.stream.MapRecord;
import org.springframework.data.redis.connection.stream.Record;
import org.springframework.data.redis.connection.stream.RecordId;
import org.springframework.data.redis.core.RedisCallback;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StreamOperations;
import org.springframework.data.redis.core.convert.RedisCustomConversions;
import org.springframework.data.redis.serializer.RedisSerializer;
import org.springframework.data.redis.serializer.StringRedisSerializer;

import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;

/**
 * 默认的 RStreamTemplate
 *
 * @author L.cm
 */
public class DefaultRStreamTemplate implements RStreamTemplate {
	private static final RedisCustomConversions CUSTOM_CONVERSIONS = new RedisCustomConversions();
	private final RedisTemplate redisTemplate;
	private final StreamOperations streamOperations;

	public DefaultRStreamTemplate(RedisTemplate redisTemplate) {
		this.redisTemplate = redisTemplate;
		this.streamOperations = redisTemplate.opsForStream();
	}

	@Override
	public RecordId send(Record record) {
		// 1. MapRecord
		if (record instanceof MapRecord) {
			return streamOperations.add(record);
		}
		String stream = Objects.requireNonNull(record.getStream(), "RStreamTemplate send stream name is null.");
		Object recordValue = Objects.requireNonNull(record.getValue(), () -> "RStreamTemplate send stream: " + stream + " value is null.");
		Class valueClass = recordValue.getClass();
		// 2. 普通类型的 ObjectRecord
		if (CUSTOM_CONVERSIONS.isSimpleType(valueClass)) {
			return streamOperations.add(record);
		}
		// 3. 自定义类型处理
		Map payload = new HashMap<>();
		payload.put(RStreamTemplate.OBJECT_PAYLOAD_KEY, recordValue);
		MapRecord mapRecord = MapRecord.create(stream, payload);
		return streamOperations.add(mapRecord);
	}

	@Override
	public RecordId send(String name, String key, byte[] data, RedisStreamCommands.XAddOptions options) {
		RedisSerializer stringSerializer = StringRedisSerializer.UTF_8;
		byte[] nameBytes = Objects.requireNonNull(stringSerializer.serialize(name), "redis stream name is null.");
		byte[] keyBytes = Objects.requireNonNull(stringSerializer.serialize(key), "redis stream key is null.");
		Map mapDate = Collections.singletonMap(keyBytes, data);
		return redisTemplate.execute((RedisCallback) redis -> {
			RedisStreamCommands streamCommands = redis.streamCommands();
			return streamCommands.xAdd(MapRecord.create(nameBytes, mapDate), options);
		});
	}

	@Override
	public Long delete(String name, String... recordIds) {
		return streamOperations.delete(name, recordIds);
	}

	@Override
	public Long delete(String name, RecordId... recordIds) {
		return streamOperations.delete(name, recordIds);
	}

	@Override
	public Long trim(String name, long count, boolean approximateTrimming) {
		return streamOperations.trim(name, count, approximateTrimming);
	}

	@Override
	public Long acknowledge(String name, String group, String... recordIds) {
		return streamOperations.acknowledge(name, group, recordIds);
	}

	@Override
	public Long acknowledge(String name, String group, RecordId... recordIds) {
		return streamOperations.acknowledge(name, group, recordIds);
	}

	@Override
	public Long acknowledge(String group, Record record) {
		return streamOperations.acknowledge(group, record);
	}

}