org.jetlinks.registry.redis.CompositeDeviceMessageSenderInterceptor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jetlinks-registry-redis Show documentation

基于Redis的设备注册中心

The newest version!

package org.jetlinks.registry.redis;

import org.jetlinks.core.device.DeviceOperation;
import org.jetlinks.core.message.DeviceMessage;
import org.jetlinks.core.message.DeviceMessageReply;
import org.jetlinks.core.message.interceptor.DeviceMessageSenderInterceptor;

import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionStage;
import java.util.concurrent.CopyOnWriteArrayList;

public class CompositeDeviceMessageSenderInterceptor implements DeviceMessageSenderInterceptor {
    private List interceptors = new CopyOnWriteArrayList<>();

    public void addInterceptor(DeviceMessageSenderInterceptor interceptor) {
        interceptors.add(interceptor);
    }

    @Override
    public DeviceMessage preSend(DeviceOperation device, DeviceMessage message) {
        for (DeviceMessageSenderInterceptor interceptor : interceptors) {
            message = interceptor.preSend(device, message);
        }
        return message;
    }

    @Override
    public  CompletionStage afterReply(DeviceOperation device, DeviceMessage message, R reply) {

        CompletableFuture future = CompletableFuture.completedFuture(reply);

        for (DeviceMessageSenderInterceptor interceptor : interceptors) {
            future = future.thenCompose(r -> interceptor.afterReply(device, message, r));
        }

        return future;

    }
}