com.tvd12.ezydata.elasticsearch.concurrent.EzyEsActionHandleLoop Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ezydata-elasticsearch Show documentation
The newest version!
package com.tvd12.ezydata.elasticsearch.concurrent;

import com.tvd12.ezydata.elasticsearch.action.EzyEsAction;
import com.tvd12.ezydata.elasticsearch.action.EzyEsActionWrapper;
import com.tvd12.ezydata.elasticsearch.callback.EzyEsActionCallback;
import com.tvd12.ezydata.elasticsearch.handler.EzyEsActionHandler;
import com.tvd12.ezydata.elasticsearch.handler.EzyEsUncaughtExceptionHandler;
import com.tvd12.ezyfox.concurrent.EzyThreadList;
import com.tvd12.ezyfox.util.EzyLoggable;
import com.tvd12.ezyfox.util.EzyStartable;
import com.tvd12.ezyfox.util.EzyStoppable;

public class EzyEsActionHandleLoop
    extends EzyLoggable
    implements EzyStartable, EzyStoppable {

    protected volatile boolean active;
    protected final EzyEsActionQueue actionQueue;
    protected final EzyThreadList executorService;
    protected final EzyEsUncaughtExceptionHandler uncaughtExceptionHandler;
    
    protected static final String THREAD_NAME = "elasticsearch-action-handling";

    public EzyEsActionHandleLoop(
        int threadPoolSize,
        EzyEsActionQueue actionQueue,
        EzyEsUncaughtExceptionHandler uncaughtExceptionHandler) {
        this.actionQueue = actionQueue;
        this.executorService = newExecutorService(threadPoolSize);
        this.uncaughtExceptionHandler = uncaughtExceptionHandler;
    }

    @Override
    public void start() {
        this.active = true;
        this.executorService.execute();
    }

    protected void loop() {
        while (active) {
            this.handleActions();
        }
    }

    @SuppressWarnings({"unchecked", "rawtypes"})
    protected void handleActions() {
        Object response = null;
        Exception exception = null;
        EzyEsAction action = null;
        EzyEsActionCallback callback = null;
        try {
            EzyEsActionWrapper wrapper = actionQueue.takeAction();
            EzyEsActionHandler handler = wrapper.getHandler();
            action = wrapper.getAction();
            callback = wrapper.getCallback();
            response = handler.handle(action);
        } catch (Exception e) {
            exception = e;
        } finally {
            if (response != null && callback != null) {
                callback.onSuccess(action, response);
            }
            if (exception != null) {
                if (callback != null) {
                    callback.onError(action, exception);
                } else if (uncaughtExceptionHandler != null) {
                    uncaughtExceptionHandler.uncaughtException(action, exception);
                } else {
                    logger.error(
                        "call action ({}): {} error",
                        action != null ? action.getActionType() : "null",
                        action,
                        exception
                    );
                }
            }
        }
    }

    protected EzyThreadList newExecutorService(int threadPoolSize) {
        Runnable task = this::loop;
        return new EzyThreadList(threadPoolSize, task, THREAD_NAME);
    }

    @Override
    public void stop() {
        this.active = false;
    }
}