net.intelie.liverig.witsml.etp.processor.DiscoveryMessageProcessor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of liverig-witsml Show documentation

liverig witsml

The newest version!

package net.intelie.liverig.witsml.etp.processor;

import Energistics.Datatypes.MessageHeader;
import Energistics.Protocol.Discovery.GetResourcesResponse;
import net.intelie.liverig.witsml.etp.protocol.MessageFlags;
import org.jetbrains.annotations.NotNull;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.concurrent.CountDownLatch;

public class DiscoveryMessageProcessor extends ETPMessageProcessor {

    private static final Logger LOGGER = LoggerFactory.getLogger(DiscoveryMessageProcessor.class);

    public long getDefaultTimeout() {
        return 30;
    }

    @Override
    public GetResourcesResponse parse(@NotNull CountDownLatch messageLatch, @NotNull MessageHeader messageHeader, @NotNull GetResourcesResponse body) {

        if (body != null && body.getResource() != null)
            LOGGER.info("Returning discovery response: {} {} {}", messageHeader.getMessageFlags(), body.getResource().getUri(), body.getResource().getResourceType());

        if (MessageFlags.FinalPart.getValue() == messageHeader.getMessageFlags() ||
                MessageFlags.NoData.getValue() == messageHeader.getMessageFlags() ||
                MessageFlags.None.getValue() == messageHeader.getMessageFlags()) {
            messageLatch.countDown();
        }
        return body;
    }
}