org.babyfish.jimmer.sql.fetcher.impl.FetcherMergeContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jimmer-sql Show documentation

A revolutionary ORM framework for both java and kotlin

There is a newer version: 0.9.19

package org.babyfish.jimmer.sql.fetcher.impl;

import org.babyfish.jimmer.sql.fetcher.Fetcher;
import org.babyfish.jimmer.sql.fetcher.Field;

import java.util.Map;
import java.util.Objects;

class FetcherMergeContext {

    private final String path;

    FetcherMergeContext() {
        this("");
    }

    private FetcherMergeContext(String path) {
        this.path = path;
    }

    @SuppressWarnings("unchecked")
    public Fetcher merge(Fetcher fetcher1, Fetcher fetcher2, boolean implicit) throws ConflictException {
        boolean isEmbeddable = (fetcher1 != null && fetcher1.getImmutableType().isEmbeddable()) ||
                (fetcher2 != null && fetcher2.getImmutableType().isEmbeddable());
        if (fetcher1 == null) {
            return isEmbeddable ? null : fetcher2;
        }
        if (fetcher2 == null) {
            return isEmbeddable ? null : fetcher1;
        }
        Map fieldMap1 = fetcher1.getFieldMap();
        Map fieldMap2 = fetcher2.getFieldMap();
        for (Field field1 : fieldMap1.values()) {
            if (field1.getProp().isId()) {
                continue;
            }
            Field field2 = fieldMap2.get(field1.getProp().getName());
            if (field2 != null) {
                String conflictCfgName = null;
                if (field1.getBatchSize() != field2.getBatchSize()) {
                    conflictCfgName = "batchSize";
                } else if (field1.getLimit() != field2.getLimit()) {
                    conflictCfgName = "limit";
                } else if (field1.getOffset() != field2.getOffset()) {
                    conflictCfgName = "offset";
                } else if (!Objects.equals(field1.getFilter(),field1.getFilter())) {
                    conflictCfgName = "batchSize";
                } else if (!Objects.equals(field1.getRecursionStrategy(),field1.getRecursionStrategy())) {
                    conflictCfgName = "batchSize";
                }
                if (conflictCfgName != null) {
                    throw new ConflictException(path, conflictCfgName);
                }
                if (field1.getRecursionStrategy() != null || field2.getRecursionStrategy() != null) {
                    throw new IllegalArgumentException(
                            "Both \"" +
                                    field1.getProp() +
                                    "\" and \"" +
                                    field2.getProp() +
                                    "\" are fetched, so the recursion strategy cannot be specified"
                    );
                }
            }
            fetcher2 = new FetcherImpl<>(
                    (FetcherImpl