Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Copyright 2018-2021 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* 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 org.springframework.data.redis.connection.stream;
import java.nio.ByteBuffer;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;
import org.springframework.data.redis.util.ByteUtils;
import org.springframework.lang.Nullable;
import org.springframework.util.Assert;
import org.springframework.util.ClassUtils;
import org.springframework.util.ObjectUtils;
/**
* {@link StreamRecords} provides utilities to create specific {@link Record} instances.
*
* @author Christoph Strobl
* @since 2.2
*/
public class StreamRecords {
/**
* Create a new {@link ByteRecord} for the given raw field/value pairs.
*
* @param raw must not be {@literal null}.
* @return new instance of {@link ByteRecord}.
*/
public static ByteRecord rawBytes(Map raw) {
return new ByteMapBackedRecord(null, RecordId.autoGenerate(), raw);
}
/**
* Create a new {@link ByteBufferRecord} for the given raw field/value pairs.
*
* @param raw must not be {@literal null}.
* @return new instance of {@link ByteBufferRecord}.
*/
public static ByteBufferRecord rawBuffer(Map raw) {
return new ByteBufferMapBackedRecord(null, RecordId.autoGenerate(), raw);
}
/**
* Create a new {@link ByteBufferRecord} for the given raw field/value pairs.
*
* @param raw must not be {@literal null}.
* @return new instance of {@link ByteBufferRecord}.
*/
public static StringRecord string(Map raw) {
return new StringMapBackedRecord(null, RecordId.autoGenerate(), raw);
}
/**
* Create a new {@link MapRecord} backed by the field/value pairs of the given {@link Map}.
*
* @param map must not be {@literal null}.
* @param type of the stream key.
* @param type of the map key.
* @param type of the map value.
* @return new instance of {@link MapRecord}.
*/
public static MapRecord mapBacked(Map map) {
return new MapBackedRecord<>(null, RecordId.autoGenerate(), map);
}
/**
* Create new {@link ObjectRecord} backed by the given value.
*
* @param value must not be {@literal null}.
* @param the stream key type
* @param the value type.
* @return new instance of {@link ObjectRecord}.
*/
public static ObjectRecord objectBacked(V value) {
return new ObjectBackedRecord<>(null, RecordId.autoGenerate(), value);
}
/**
* Obtain new instance of {@link RecordBuilder} to fluently create {@link Record records}.
*
* @return new instance of {@link RecordBuilder}.
*/
public static RecordBuilder newRecord() {
return new RecordBuilder<>(null, RecordId.autoGenerate());
}
// Utility constructor
private StreamRecords() {}
/**
* Builder for {@link Record}.
*
* @param stream keyy type.
*/
public static class RecordBuilder {
private RecordId id;
private S stream;
RecordBuilder(@Nullable S stream, RecordId recordId) {
this.stream = stream;
this.id = recordId;
}
/**
* Configure a stream key.
*
* @param stream the stream key, must not be null.
* @param
* @return {@literal this} {@link RecordBuilder}.
*/
public RecordBuilder in(STREAM_KEY stream) {
Assert.notNull(stream, "Stream key must not be null");
return new RecordBuilder<>(stream, id);
}
/**
* Configure a record Id given a {@link String}. Associates a user-supplied record id instead of using
* server-generated record Id's.
*
* @param id the record id.
* @return {@literal this} {@link RecordBuilder}.
* @see RecordId
*/
public RecordBuilder withId(String id) {
return withId(RecordId.of(id));
}
/**
* Configure a {@link RecordId}. Associates a user-supplied record id instead of using server-generated record Id's.
*
* @param id the record id.
* @return {@literal this} {@link RecordBuilder}.
*/
public RecordBuilder withId(RecordId id) {
Assert.notNull(id, "RecordId must not be null");
this.id = id;
return this;
}
/**
* Create a {@link MapRecord}.
*
* @param map
* @param
* @param
* @return new instance of {@link MapRecord}.
*/
public MapRecord ofMap(Map map) {
return new MapBackedRecord<>(stream, id, map);
}
/**
* Create a {@link StringRecord}.
*
* @param map
* @return new instance of {@link StringRecord}.
* @see MapRecord
*/
public StringRecord ofStrings(Map map) {
return new StringMapBackedRecord(ObjectUtils.nullSafeToString(stream), id, map);
}
/**
* Create an {@link ObjectRecord}.
*
* @param value
* @param
* @return new instance of {@link ObjectRecord}.
*/
public ObjectRecord ofObject(V value) {
return new ObjectBackedRecord<>(stream, id, value);
}
/**
* @param value
* @return new instance of {@link ByteRecord}.
*/
public ByteRecord ofBytes(Map value) {
// todo auto conversion of known values
return new ByteMapBackedRecord((byte[]) stream, id, value);
}
/**
* @param value
* @return new instance of {@link ByteBufferRecord}.
*/
public ByteBufferRecord ofBuffer(Map value) {
ByteBuffer streamKey;
if (stream instanceof ByteBuffer) {
streamKey = (ByteBuffer) stream;
} else if (stream instanceof String) {
streamKey = ByteUtils.getByteBuffer((String) stream);
} else if (stream instanceof byte[]) {
streamKey = ByteBuffer.wrap((byte[]) stream);
} else {
throw new IllegalArgumentException(String.format("Stream key %s cannot be converted to byte buffer.", stream));
}
return new ByteBufferMapBackedRecord(streamKey, id, value);
}
}
/**
* Default implementation of {@link MapRecord}.
*
* @param
* @param
* @param
*/
static class MapBackedRecord implements MapRecord {
private @Nullable S stream;
private RecordId recordId;
private final Map kvMap;
MapBackedRecord(@Nullable S stream, RecordId recordId, Map kvMap) {
this.stream = stream;
this.recordId = recordId;
this.kvMap = kvMap;
}
@Nullable
@Override
public S getStream() {
return stream;
}
@Nullable
@Override
public RecordId getId() {
return recordId;
}
@Override
public Iterator> iterator() {
return kvMap.entrySet().iterator();
}
@Override
public Map getValue() {
return kvMap;
}
@Override
public MapRecord withId(RecordId id) {
return new MapBackedRecord<>(stream, id, this.kvMap);
}
@Override
public MapRecord withStreamKey(S1 key) {
return new MapBackedRecord<>(key, recordId, this.kvMap);
}
@Override
public String toString() {
return "MapBackedRecord{" + "recordId=" + recordId + ", kvMap=" + kvMap + '}';
}
@Override
public boolean equals(Object o) {
if (o == null) {
return false;
}
if (this == o) {
return true;
}
if (!ClassUtils.isAssignable(MapBackedRecord.class, o.getClass())) {
return false;
}
MapBackedRecord that = (MapBackedRecord) o;
if (!ObjectUtils.nullSafeEquals(this.stream, that.stream)) {
return false;
}
if (!ObjectUtils.nullSafeEquals(this.recordId, that.recordId)) {
return false;
}
return ObjectUtils.nullSafeEquals(this.kvMap, that.kvMap);
}
@Override
public int hashCode() {
int result = stream != null ? stream.hashCode() : 0;
result = 31 * result + recordId.hashCode();
result = 31 * result + kvMap.hashCode();
return result;
}
}
/**
* Default implementation of {@link ByteRecord}.
*/
static class ByteMapBackedRecord extends MapBackedRecord implements ByteRecord {
ByteMapBackedRecord(byte[] stream, RecordId recordId, Map map) {
super(stream, recordId, map);
}
@Override
public ByteMapBackedRecord withStreamKey(byte[] key) {
return new ByteMapBackedRecord(key, getId(), getValue());
}
@Override
public ByteMapBackedRecord withId(RecordId id) {
return new ByteMapBackedRecord(getStream(), id, getValue());
}
}
/**
* Default implementation of {@link ByteBufferRecord}.
*/
static class ByteBufferMapBackedRecord extends MapBackedRecord
implements ByteBufferRecord {
ByteBufferMapBackedRecord(ByteBuffer stream, RecordId recordId, Map map) {
super(stream, recordId, map);
}
@Override
public ByteBufferMapBackedRecord withStreamKey(ByteBuffer key) {
return new ByteBufferMapBackedRecord(key, getId(), getValue());
}
@Override
public ByteBufferMapBackedRecord withId(RecordId id) {
return new ByteBufferMapBackedRecord(getStream(), id, getValue());
}
}
/**
* Default implementation of StringRecord.
*/
static class StringMapBackedRecord extends MapBackedRecord implements StringRecord {
StringMapBackedRecord(String stream, RecordId recordId, Map stringStringMap) {
super(stream, recordId, stringStringMap);
}
@Override
public StringRecord withStreamKey(String key) {
return new StringMapBackedRecord(key, getId(), getValue());
}
@Override
public StringMapBackedRecord withId(RecordId id) {
return new StringMapBackedRecord(getStream(), id, getValue());
}
}
/**
* Default implementation of {@link ObjectRecord}.
*
* @param
* @param
*/
static class ObjectBackedRecord implements ObjectRecord {
private @Nullable S stream;
private RecordId recordId;
private final V value;
ObjectBackedRecord(@Nullable S stream, RecordId recordId, V value) {
this.stream = stream;
this.recordId = recordId;
this.value = value;
}
@Nullable
@Override
public S getStream() {
return stream;
}
@Nullable
@Override
public RecordId getId() {
return recordId;
}
@Override
public V getValue() {
return value;
}
@Override
public ObjectRecord withId(RecordId id) {
return new ObjectBackedRecord<>(stream, id, value);
}
@Override
public ObjectRecord withStreamKey(SK key) {
return new ObjectBackedRecord<>(key, recordId, value);
}
@Override
public String toString() {
return "ObjectBackedRecord{" + "recordId=" + recordId + ", value=" + value + '}';
}
@Override
public boolean equals(Object o) {
if (this == o)
return true;
if (o == null || getClass() != o.getClass())
return false;
ObjectBackedRecord that = (ObjectBackedRecord) o;
if (!ObjectUtils.nullSafeEquals(stream, that.stream)) {
return false;
}
if (!ObjectUtils.nullSafeEquals(recordId, that.recordId)) {
return false;
}
return ObjectUtils.nullSafeEquals(value, that.value);
}
@Override
public int hashCode() {
int result = ObjectUtils.nullSafeHashCode(stream);
result = 31 * result + ObjectUtils.nullSafeHashCode(recordId);
result = 31 * result + ObjectUtils.nullSafeHashCode(value);
return result;
}
}
}
