io.github.sinri.drydock.common.QueueMixin Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of DryDock Show documentation
Prototype Library for Keel Based Java Projects.
The newest version!
package io.github.sinri.drydock.common;

import io.github.sinri.keel.logger.issue.recorder.KeelIssueRecorder;
import io.github.sinri.keel.servant.queue.KeelQueue;
import io.github.sinri.keel.servant.queue.KeelQueueNextTaskSeeker;
import io.github.sinri.keel.servant.queue.QueueManageIssueRecord;
import io.github.sinri.keel.servant.queue.QueueWorkerPoolManager;
import io.vertx.core.DeploymentOptions;
import io.vertx.core.Future;
import io.vertx.core.ThreadingModel;

import javax.annotation.Nonnull;

/**
 * @since 1.1.0
 */
public interface QueueMixin extends CommonUnit {
    default KeelQueue buildQueue() {
        KeelIssueRecorder issueRecorder = this.generateIssueRecorder(QueueManageIssueRecord.TopicQueue, QueueManageIssueRecord::new);
        return new KeelQueue() {
            @Nonnull
            @Override
            protected KeelIssueRecorder buildIssueRecorder() {
                return issueRecorder;
            }

            @Override
            protected @Nonnull KeelQueueNextTaskSeeker getNextTaskSeeker() {
                return buildQueueNextTaskSeeker();
            }

            @Override
            protected @Nonnull SignalReader getSignalReader() {
                return buildSignalReader();
            }

            @Nonnull
            @Override
            protected QueueWorkerPoolManager getQueueWorkerPoolManager() {
                var x = configuredQueueWorkerPoolSize();
                return new QueueWorkerPoolManager(x);
            }
        };
    }

    /**
     * @return zero for unlimited.
     */
    default int configuredQueueWorkerPoolSize() {
        return 0;
    }

    KeelQueue.SignalReader buildSignalReader();

    KeelQueueNextTaskSeeker buildQueueNextTaskSeeker();

    default Future loadQueue() {
        return Future.succeededFuture(this.buildQueue())
                .compose(queue -> {
                    if (queue == null) return Future.succeededFuture();
                    return this.beforeLoadingQueue()
                            .compose(v -> {
                                return queue.deployMe(new DeploymentOptions().setThreadingModel(ThreadingModel.WORKER))
                                        .onFailure(throwable -> {
                                            getLogger().exception(throwable, "Failed to load queue");
                                        })
                                        .compose(deploymentId -> {
                                            getLogger().info("Loaded queue: " + deploymentId);
                                            return Future.succeededFuture();
                                        });
                            });
                });
    }

    /**
     * Execute before `io.github.sinri.drydock.common.QueueMixin#loadQueue()`,
     * to clean up the left RUNNING tasks (let them ERROR).
     *
     * @since 1.5.8
     */
    default Future beforeLoadingQueue() {
        return Future.succeededFuture();
    }
}