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 2015-2018 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
*
* http://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.core;
import java.util.Collection;
import java.util.List;
import java.util.Set;
import org.springframework.data.redis.connection.RedisClusterCommands.AddSlots;
import org.springframework.data.redis.connection.RedisClusterConnection;
import org.springframework.data.redis.connection.RedisClusterNode;
import org.springframework.data.redis.connection.RedisClusterNode.SlotRange;
import org.springframework.data.redis.connection.RedisServerCommands.MigrateOption;
import org.springframework.lang.Nullable;
import org.springframework.util.Assert;
/**
* Default {@link ClusterOperations} implementation.
*
* @author Christoph Strobl
* @since 1.7
* @param
* @param
*/
class DefaultClusterOperations extends AbstractOperations implements ClusterOperations {
private final RedisTemplate template;
/**
* Creates new {@link DefaultClusterOperations} delegating to the given {@link RedisTemplate}.
*
* @param template must not be {@literal null}.
*/
DefaultClusterOperations(RedisTemplate template) {
super(template);
this.template = template;
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.RedisClusterOperations#keys(org.springframework.data.redis.connection.RedisNode, byte[])
*/
@Override
public Set keys(final RedisClusterNode node, final K pattern) {
Assert.notNull(node, "ClusterNode must not be null.");
return execute(connection -> deserializeKeys(connection.keys(node, rawKey(pattern))));
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.RedisClusterOperations#randomKey(org.springframework.data.redis.connection.RedisNode)
*/
@Override
public K randomKey(final RedisClusterNode node) {
Assert.notNull(node, "ClusterNode must not be null.");
return execute(connection -> deserializeKey(connection.randomKey(node)));
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.RedisClusterOperations#ping(org.springframework.data.redis.connection.RedisNode)
*/
@Override
public String ping(final RedisClusterNode node) {
Assert.notNull(node, "ClusterNode must not be null.");
return execute(connection -> connection.ping(node));
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.RedisClusterOperations#addSlots(org.springframework.data.redis.connection.RedisClusterNode, int[])
*/
@Override
public void addSlots(final RedisClusterNode node, final int... slots) {
Assert.notNull(node, "ClusterNode must not be null.");
execute((RedisClusterCallback) connection -> {
connection.clusterAddSlots(node, slots);
return null;
});
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.RedisClusterOperations#addSlots(org.springframework.data.redis.connection.RedisClusterNode, org.springframework.data.redis.connection.RedisClusterNode.SlotRange)
*/
@Override
public void addSlots(RedisClusterNode node, SlotRange range) {
Assert.notNull(node, "ClusterNode must not be null.");
Assert.notNull(range, "Range must not be null.");
addSlots(node, range.getSlotsArray());
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.RedisClusterOperations#bgReWriteAof(org.springframework.data.redis.connection.RedisClusterNode)
*/
@Override
public void bgReWriteAof(final RedisClusterNode node) {
Assert.notNull(node, "ClusterNode must not be null.");
execute((RedisClusterCallback) connection -> {
connection.bgReWriteAof(node);
return null;
});
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.RedisClusterOperations#bgSave(org.springframework.data.redis.connection.RedisClusterNode)
*/
@Override
public void bgSave(final RedisClusterNode node) {
Assert.notNull(node, "ClusterNode must not be null.");
execute((RedisClusterCallback) connection -> {
connection.bgSave(node);
return null;
});
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.RedisClusterOperations#meet(org.springframework.data.redis.connection.RedisClusterNode)
*/
@Override
public void meet(final RedisClusterNode node) {
Assert.notNull(node, "ClusterNode must not be null.");
execute((RedisClusterCallback) connection -> {
connection.clusterMeet(node);
return null;
});
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.RedisClusterOperations#forget(org.springframework.data.redis.connection.RedisClusterNode)
*/
@Override
public void forget(final RedisClusterNode node) {
Assert.notNull(node, "ClusterNode must not be null.");
execute((RedisClusterCallback) connection -> {
connection.clusterForget(node);
return null;
});
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.RedisClusterOperations#flushDb(org.springframework.data.redis.connection.RedisClusterNode)
*/
@Override
public void flushDb(final RedisClusterNode node) {
Assert.notNull(node, "ClusterNode must not be null.");
execute((RedisClusterCallback) connection -> {
connection.flushDb(node);
return null;
});
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.RedisClusterOperations#getSlaves(org.springframework.data.redis.connection.RedisClusterNode)
*/
@Override
public Collection getSlaves(final RedisClusterNode node) {
Assert.notNull(node, "ClusterNode must not be null.");
return execute(connection -> connection.clusterGetSlaves(node));
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.RedisClusterOperations#save(org.springframework.data.redis.connection.RedisClusterNode)
*/
@Override
public void save(final RedisClusterNode node) {
Assert.notNull(node, "ClusterNode must not be null.");
execute((RedisClusterCallback) connection -> {
connection.save(node);
return null;
});
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.RedisClusterOperations#shutdown(org.springframework.data.redis.connection.RedisClusterNode)
*/
@Override
public void shutdown(final RedisClusterNode node) {
Assert.notNull(node, "ClusterNode must not be null.");
execute((RedisClusterCallback) connection -> {
connection.shutdown(node);
return null;
});
}
/*
* (non-Javadoc)
* @see org.springframework.data.redis.core.ClusterOperations#reshard(org.springframework.data.redis.connection.RedisClusterNode, int, org.springframework.data.redis.connection.RedisClusterNode)
*/
@Override
public void reshard(final RedisClusterNode source, final int slot, final RedisClusterNode target) {
Assert.notNull(source, "Source node must not be null.");
Assert.notNull(target, "Target node must not be null.");
execute((RedisClusterCallback) connection -> {
connection.clusterSetSlot(target, slot, AddSlots.IMPORTING);
connection.clusterSetSlot(source, slot, AddSlots.MIGRATING);
List keys = connection.clusterGetKeysInSlot(slot, Integer.MAX_VALUE);
for (byte[] key : keys) {
connection.migrate(key, source, 0, MigrateOption.COPY);
}
connection.clusterSetSlot(target, slot, AddSlots.NODE);
return null;
});
}
/**
* Executed wrapped command upon {@link RedisClusterConnection}.
*
* @param callback must not be {@literal null}.
* @return execution result. Can be {@literal null}.
*/
@Nullable
public T execute(RedisClusterCallback callback) {
Assert.notNull(callback, "ClusterCallback must not be null!");
RedisClusterConnection connection = template.getConnectionFactory().getClusterConnection();
try {
return callback.doInRedis(connection);
} finally {
connection.close();
}
}
}
