package at.gv.egiz.moazs.repository;


import at.gv.egiz.moazs.scheme.MsgResponse;
import at.gv.zustellung.app2mzs.xsd.DeliveryRequestType;
import com.google.common.cache.Cache;
import com.google.common.cache.CacheBuilder;
import org.springframework.context.annotation.Profile;
import org.springframework.stereotype.Repository;

import java.util.Optional;
import java.util.concurrent.TimeUnit;

import static java.util.Optional.ofNullable;

@Repository
@Profile("!cluster")
public class InMemoryDeliveryRepository implements DeliveryRepository {

    private final Cache<String, DeliveryRequestType> requestRepository = CacheBuilder.newBuilder()
            .expireAfterWrite(30, TimeUnit.MINUTES)
            .build();

    private final Cache<String, MsgResponse> responseRepository = CacheBuilder.newBuilder()
            .expireAfterWrite(30, TimeUnit.MINUTES)
            .build();

    @Override
    public void add(DeliveryRequestType request) {
        var key = request.getMetaData().getAppDeliveryID();
        requestRepository.put(key, request);
    }

    @Override
    public Optional<DeliveryRequestType> getDeliveryRequest(String appDeliveryID) {
        return ofNullable(requestRepository.getIfPresent(appDeliveryID));
    }

    @Override
    public void add(MsgResponse response) {
        String key = response.getResponseID();
        responseRepository.put(key, response);
    }

    @Override
    public Optional<MsgResponse> getResponse(String responseID) {
        return ofNullable(responseRepository.getIfPresent(responseID));
    }
}