com.github.linyuzai.connection.loadbalance.sse.servlet.ServletSubscriberSseConnection Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of concept-connection-loadbalance-sse Show documentation

Load Balance 4 SSE

The newest version!

package com.github.linyuzai.connection.loadbalance.sse.servlet;

import com.github.linyuzai.connection.loadbalance.sse.concept.SubscriberSseConnection;
import lombok.Getter;
import lombok.RequiredArgsConstructor;
import lombok.Setter;

import java.net.HttpURLConnection;
import java.util.function.Consumer;

@Getter
@Setter
@RequiredArgsConstructor
public class ServletSubscriberSseConnection extends SubscriberSseConnection {

    private HttpURLConnection http;

    @Override
    public void doClose(Object reason, Runnable onSuccess, Consumer onError, Runnable onComplete) {
        try {
            if (http != null) {
                http.disconnect();
            }
            onSuccess.run();
        } catch (Throwable e) {
            onError.accept(e);
        } finally {
            onComplete.run();
        }
    }
}