com.kenshoo.pl.entity.PLContext Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of persistence-layer Show documentation

A Java persistence layer based on JOOQ for high performance and business flow support.

There is a newer version: 0.1.121-jooq-3.16.3

package com.kenshoo.pl.entity;

import com.kenshoo.pl.entity.internal.EntitiesFetcher;
import com.kenshoo.pl.entity.spi.PersistenceLayerRetryer;
import com.kenshoo.pl.entity.spi.audit.AuditRecordPublisher;
import org.jooq.DSLContext;
import org.jooq.lambda.Seq;

import java.util.function.Predicate;

import static com.kenshoo.pl.entity.spi.PersistenceLayerRetryer.JUST_RUN_WITHOUT_CHECKING_DEADLOCKS;


public class PLContext {

    final private DSLContext dslContext;
    final private PersistenceLayerRetryer retryer;
    final private Predicate featurePredicate;
    private final AuditRecordPublisher auditRecordPublisher;

    private PLContext(final DSLContext dslContext,
                      final PersistenceLayerRetryer retryer,
                      final Predicate featurePredicate,
                      final AuditRecordPublisher auditRecordPublisher) {
        this.dslContext = dslContext;
        this.retryer = retryer;
        this.featurePredicate = featurePredicate;
        this.auditRecordPublisher = auditRecordPublisher;
    }

    public DSLContext dslContext() {
        return dslContext;
    }

    public FeatureSet generateFeatureSet() {
        return new FeatureSet(Seq.of(Feature.values()).filter(featurePredicate));
    }

    public PersistenceLayerRetryer persistenceLayerRetryer() {
        return  retryer;
    }

    public AuditRecordPublisher auditRecordPublisher() {
        return auditRecordPublisher;
    }

    /**
     * Start building a query to fetch entities with the given fields.
     *
     * @param fields the fields to fetch, must not be empty
     * @return the next step in which the entity type will be specified
     */
    public FetchFromStep select(final EntityField... fields) {
        return new FluentEntitiesFetcher(fetcher(), fields);
    }

    private EntitiesFetcher fetcher() {
        return new EntitiesFetcher(dslContext, generateFeatureSet());
    }

    public static class Builder {

        private DSLContext dslContext;
        private PersistenceLayerRetryer retryer = JUST_RUN_WITHOUT_CHECKING_DEADLOCKS;
        private Predicate featurePredicate = __ -> false;
        private AuditRecordPublisher auditRecordPublisher = AuditRecordPublisher.NO_OP;

        public Builder withFeaturePredicate(Predicate featurePredicate) {
            this.featurePredicate = featurePredicate;
            return this;
        }

        public Builder(DSLContext dslContext) {
            this.dslContext = dslContext;
        }

        public Builder withRetryer(PersistenceLayerRetryer retryer) {
            this.retryer = retryer;
            return this;
        }

        public Builder withAuditRecordPublisher(final AuditRecordPublisher auditRecordPublisher) {
            this.auditRecordPublisher = auditRecordPublisher;
            return this;
        }

        public PLContext build() {
            return new PLContext(dslContext,
                                 retryer,
                                 featurePredicate,
                                 auditRecordPublisher);
        }

    }
}