com.github.liuyehcf.framework.rpc.ares.util.AresContext Maven / Gradle / Ivy

Go to download
package com.github.liuyehcf.framework.rpc.ares.util;

import com.github.liuyehcf.framework.rpc.ares.constant.SchemaType;
import lombok.AllArgsConstructor;
import lombok.Data;

/**
 * @author hechenfeng
 * @date 2019/11/29
 */
public abstract class AresContext {

    private static final ThreadLocal ENDPOINT = new ThreadLocal<>();

    public static  T invokeWithEndpoint(SchemaType schemaType, String host, Integer port, Invoker invoker) {
        try {
            setEndpoint(schemaType, host, port);
            return invoker.invoke();
        } finally {
            removeEndpoint();
        }
    }

    public static void setEndpoint(SchemaType schemaType, String host, Integer port) {
        ENDPOINT.set(new Endpoint(schemaType, host, port));
    }

    public static Endpoint getEndpoint() {
        return ENDPOINT.get();
    }

    public static void removeEndpoint() {
        ENDPOINT.remove();
    }

    public interface Invoker {

        /**
         * invoker wraps http biz logic
         */
        T invoke();
    }

    @Data
    @AllArgsConstructor
    public static final class Endpoint {
        /**
         * http schema
         */
        private final SchemaType schema;

        /**
         * http host
         */
        private final String host;

        /**
         * http port
         */
        private final Integer port;
    }
}