net.binis.codegen.spring.AsyncEntityModifier Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of code-generator-spring Show documentation
Spring extension for Binis Code Generation
There is a newer version: 1.2.22
Show newest version
package net.binis.codegen.spring;

/*-
 * #%L
 * code-generator-spring
 * %%
 * Copyright (C) 2021 Binis Belev
 * %%
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */

import lombok.extern.slf4j.Slf4j;
import net.binis.codegen.factory.CodeFactory;
import net.binis.codegen.spring.async.AsyncDispatcher;
import net.binis.codegen.spring.async.AsyncModifier;
import net.binis.codegen.spring.async.executor.CodeExecutor;
import net.binis.codegen.spring.async.executor.CodeGenCompletableFuture;

import java.time.Duration;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.TimeUnit;
import java.util.function.Consumer;
import java.util.function.Supplier;

import static java.util.Objects.nonNull;

@Slf4j
public class AsyncEntityModifier extends BaseEntityModifier {

    static {
        CodeFactory.registerType(AsyncDispatcher.class, CodeFactory.singleton(CodeExecutor.defaultDispatcher()), null);
    }

    public AsyncModifier async() {
        return new AsyncImpl();
    }

    protected class AsyncImpl implements AsyncModifier {

        private String flow = CodeExecutor.DEFAULT;
        private long delay;
        private TimeUnit unit;

        @Override
        public AsyncModifier flow(String flow) {
            this.flow = flow;
            return this;
        }

        @Override
        public AsyncModifier delay(long delay, TimeUnit unit) {
            this.delay = delay;
            this.unit = unit;
            return this;
        }

        @Override
        public AsyncModifier delay(Duration duration) {
            this.delay = duration.toMillis();
            this.unit = TimeUnit.MILLISECONDS;
            return this;
        }

        @Override
        public CompletableFuture save() {
            return execute(AsyncEntityModifier.this::save);
        }

        @Override
        public CompletableFuture delete() {
            return execute(AsyncEntityModifier.this::delete);
        }

        @Override
        public CompletableFuture execute(Consumer task) {
            return execute(() ->
                    AsyncEntityModifier.this.transaction(m -> {
                        task.accept(m);
                        return null;
                    }));
        }

        private CompletableFuture execute(Supplier supplier) {
            var executor = CodeFactory.create(AsyncDispatcher.class).flow(flow);

            if (delay > 0 && nonNull(unit)) {
                executor = CompletableFuture.delayedExecutor(delay, unit, executor);
            }

            return CodeGenCompletableFuture.newSupplyAsync(executor, supplier);
        }
    }

}