io.microsphere.redis.replicator.spring.RedisCommandReplicator Maven / Gradle / Ivy

Go to download
package io.microsphere.redis.replicator.spring;

import io.microsphere.redis.spring.event.RedisCommandEvent;
import io.microsphere.redis.replicator.spring.event.RedisCommandReplicatedEvent;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.ApplicationListener;
import org.springframework.data.redis.connection.RedisConnection;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.util.ReflectionUtils;

import java.lang.reflect.Method;
import java.util.function.Function;

import static io.microsphere.redis.spring.beans.Wrapper.tryUnwrap;
import static io.microsphere.redis.spring.metadata.RedisMetadataRepository.findWriteCommandMethod;
import static io.microsphere.redis.spring.metadata.RedisMetadataRepository.getRedisCommandBindingFunction;


/**
 * Redis Command Replicator
 *
 * @author Mercy
 * @since 1.0.0
 */
public class RedisCommandReplicator implements ApplicationListener {

    private static final Logger logger = LoggerFactory.getLogger(RedisCommandReplicator.class);

    public static final String BEAN_NAME = "redisCommandReplicator";

    private final RedisConnectionFactory redisConnectionFactory;

    public RedisCommandReplicator(RedisConnectionFactory redisConnectionFactory) {
        this.redisConnectionFactory = tryUnwrap(redisConnectionFactory, RedisConnectionFactory.class);
    }

    @Override
    public void onApplicationEvent(RedisCommandReplicatedEvent event) {
        try {
            handleRedisCommandEvent(event);
        } catch (Throwable e) {
            logger.error("[Redis-Replicator-Event] Failed to process Redis command event [{}]", event, e);
        }
    }

    private void handleRedisCommandEvent(RedisCommandReplicatedEvent event) throws Throwable {
        RedisCommandEvent redisCommandEvent = event.getSourceEvent();
        Method method = findWriteCommandMethod(redisCommandEvent);
        if (method != null) {
            String interfaceNme = redisCommandEvent.getInterfaceName();
            RedisConnection redisConnection = getRedisConnection();
            Object[] args = redisCommandEvent.getArgs();
            Function bindingFunction = getRedisCommandBindingFunction(interfaceNme);
            Object redisCommandObject = bindingFunction.apply(redisConnection);
            // TODO: Native method implementation
            ReflectionUtils.invokeMethod(method, redisCommandObject, args);
        }
    }

    private RedisConnection getRedisConnection() {
        return redisConnectionFactory.getConnection();
    }
}