JChargePointProtocol/jcpp-protocol-api/src/main/java/sanbing/jcpp/protocol/domain/ProtocolSession.java

/**
 * 抖音关注：程序员三丙
 * 知识星球：https://t.zsxq.com/j9b21
 */
package sanbing.jcpp.protocol.domain;

import com.github.benmanes.caffeine.cache.Cache;
import com.github.benmanes.caffeine.cache.Caffeine;
import lombok.Getter;
import lombok.Setter;
import lombok.extern.slf4j.Slf4j;
import sanbing.jcpp.proto.gen.ProtocolProto.DownlinkRestMessage;
import sanbing.jcpp.protocol.forwarder.Forwarder;

import java.io.Closeable;
import java.time.Duration;
import java.time.LocalDateTime;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.Set;
import java.util.UUID;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ScheduledFuture;
import java.util.function.Function;

/**
 * @author baigod
 */
@Getter
@Slf4j
public abstract class ProtocolSession implements Closeable {

    private static final int REQUEST_CACHE_LIMIT = 1000;

    protected final String protocolName;

    protected final UUID id;

    @Setter
    protected LocalDateTime lastActivityTime;

    protected final Set<String> pileCodeSet;

    private final Map<String, ScheduledFuture<?>> scheduledFutures = new ConcurrentHashMap<>();

    private final Cache<String, Object> requestCache = Caffeine.newBuilder()
            .initialCapacity(REQUEST_CACHE_LIMIT)
            .maximumSize(REQUEST_CACHE_LIMIT)
            .expireAfterAccess(Duration.ofMinutes(1))
            .build();

    @Setter
    private Forwarder forwarder;

    public ProtocolSession(String protocolName) {
        this.protocolName = protocolName;
        this.pileCodeSet = new LinkedHashSet<>();
        this.id = UUID.randomUUID();
        this.lastActivityTime = LocalDateTime.now();
    }

    public abstract void onDownlink(DownlinkRestMessage downlinkMsg);

    public void close() {
        close(SessionCloseReason.DESTRUCTION);
    }

    public void close(SessionCloseReason reason) {
        log.info("[{}] Protocol会话关闭，原因: {}", this, reason);

        scheduledFutures.values().forEach(scheduledFuture -> scheduledFuture.cancel(true));
        scheduledFutures.clear();
    }

    @Override
    public String toString() {
        return "[" + id + "]" + pileCodeSet;
    }

    public void addPileCode(String pileCode) {
        this.pileCodeSet.add(pileCode);
    }

    public void addSchedule(String name, Function<String, ScheduledFuture<?>> scheduledFutureFunction) {
        scheduledFutures.computeIfAbsent(name, scheduledFutureFunction);
    }
}