silentbalanceyh/vertx-zero

package io.vertx.up.unity;

import io.horizon.eon.em.typed.ChangeFlag;
import io.modello.atom.normalize.KRuleTerm;
import io.vertx.core.json.JsonArray;
import io.vertx.core.json.JsonObject;
import io.vertx.up.commune.record.Apt;
import io.vertx.up.eon.KName;
import io.vertx.up.util.Ut;

import java.util.Collection;
import java.util.Objects;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.function.BiFunction;

/**
 * @author <a href="http://www.origin-x.cn">Lang</a>
 */
class Unique {

    private Unique() {
    }

    static JsonObject ruleAll(final Collection<KRuleTerm> rules, final JsonObject input) {
        final boolean isMatch = rules.stream().allMatch(term -> Objects.nonNull(term.dataMatch(input)));
        return isMatch ? input : null;
    }

    static JsonObject ruleAll(final Collection<KRuleTerm> rules, final JsonObject recordO, final JsonObject recordN) {
        final boolean isMatch = rules.stream().allMatch(rule -> ruleMatch(rule, recordO, recordN));
        return isMatch ? ruleTwins(recordO, recordN) : null;
    }

    static JsonObject ruleAll(final Collection<KRuleTerm> rules, final JsonArray source, final JsonObject record) {
        return Ut.itJArray(source).map(recordR -> ruleAll(rules, record, recordR))
            .filter(Objects::nonNull).findFirst().orElse(null);
    }

    static JsonObject ruleAny(final Collection<KRuleTerm> rules, final JsonObject input) {
        final boolean isMatch = rules.stream().anyMatch(term -> Objects.nonNull(term.dataMatch(input)));
        return isMatch ? input : null;
    }

    static JsonObject ruleAny(final Collection<KRuleTerm> rules, final JsonObject recordO, final JsonObject recordN) {
        final boolean isMatch = rules.stream().anyMatch(rule -> ruleMatch(rule, recordO, recordN));
        return isMatch ? ruleTwins(recordO, recordN) : null;
    }

    static JsonObject ruleAny(final Collection<KRuleTerm> rules, final JsonArray source, final JsonObject record) {
        return Ut.itJArray(source).map(recordR -> ruleAny(rules, record, recordR))
            .filter(Objects::nonNull).findFirst().orElse(null);
    }

    static ConcurrentMap<Boolean, JsonArray> ruleAll(final Collection<KRuleTerm> rules, final JsonArray input) {
        return ruleSplit(rules, input, Unique::ruleAll);
    }

    static ConcurrentMap<Boolean, JsonArray> ruleAny(final Collection<KRuleTerm> rules, final JsonArray input) {
        return ruleSplit(rules, input, Unique::ruleAny);
    }

    static JsonObject ruleTwins(final JsonObject recordO, final JsonObject recordN) {
        final JsonObject data = new JsonObject();
        data.put(KName.__.OLD, recordO);
        data.put(KName.__.NEW, recordN);
        return data;
    }

    static Apt ruleApt(final JsonArray twins, final boolean isReplaced) {
        final JsonArray oldQueue = new JsonArray();
        final JsonArray newQueue = new JsonArray();
        final JsonArray updatedQueue = new JsonArray();
        final JsonArray addQueue = new JsonArray();
        Ut.itJArray(twins).forEach(json -> {
            final JsonObject oldRecord = json.getJsonObject(KName.__.OLD);
            final JsonObject newRecord = json.getJsonObject(KName.__.NEW);
            /*
             * 新旧任意一个不为空
             */
            if (Objects.nonNull(oldRecord) && Objects.nonNull(newRecord)) {
                /*
                 * oldRecord != null, newRecord != null
                 * UPDATE - 编辑
                 */
                oldQueue.add(oldRecord);
                newQueue.add(newRecord);
                /*
                 * Accomplish 中使用
                 */
                if (isReplaced) {
                    /*
                     * Replace mode
                     */
                    updatedQueue.add(oldRecord.copy().mergeIn(newRecord, true));
                } else {
                    /*
                     * Append mode ( Remove new record `null` )
                     */
                    final JsonObject merged = oldRecord.copy();
                    newRecord.fieldNames().stream().filter(key -> Objects.nonNull(newRecord.getValue(key)))
                        .forEach(key -> merged.put(key, newRecord.getValue(key)));
                    updatedQueue.add(merged);
                }
            }
            if (Objects.nonNull(oldRecord) && Objects.isNull(newRecord)) {
                /*
                 * oldRecord != null, newRecord == null
                 * DELETE - 删除
                 */
                oldQueue.add(oldRecord);
            }
            if (Objects.isNull(oldRecord) && Objects.nonNull(newRecord)) {
                /*
                 * oldRecord = null, newRecord != null
                 * ADD - 添加
                 */
                newQueue.add(newRecord);
                addQueue.add(newRecord);
            }
        });
        final Apt apt = Apt.create(oldQueue, newQueue);
        if (!addQueue.isEmpty()) {
            apt.comparedA(addQueue);
        }
        if (!updatedQueue.isEmpty()) {
            apt.comparedU(updatedQueue);
        }
        return apt;
    }

    static JsonObject ruleNil(final JsonObject twins, final ChangeFlag flag) {
        final JsonObject oldRecord = twins.getJsonObject(KName.__.OLD);
        final JsonObject newRecord = twins.getJsonObject(KName.__.NEW);
        final JsonObject normalized = new JsonObject();
        if (ChangeFlag.ADD == flag) {
            normalized.mergeIn(Objects.isNull(newRecord) ? oldRecord : newRecord, true);
        } else if (ChangeFlag.DELETE == flag) {
            normalized.mergeIn(Objects.isNull(oldRecord) ? newRecord : oldRecord, true);
        } else if (ChangeFlag.UPDATE == flag) {
            normalized.mergeIn(oldRecord.copy(), true);
            normalized.mergeIn(newRecord.copy(), true);
        }
        return normalized;
    }

    private static boolean ruleMatch(final KRuleTerm rule, final JsonObject recordL, final JsonObject recordR) {
        if (Objects.isNull(rule)) {
            /* Compare record directly */
            return recordL.equals(recordR);
        } else {
            /* Extract */
            final JsonObject recordLR = rule.dataRule(recordL);
            final JsonObject recordRR = rule.dataRule(recordR);
            if (Ut.isNil(recordLR) || Ut.isNil(recordRR)) {
                return false;
            } else {
                /* Compare record */
                return recordLR.equals(recordRR);
            }
        }
    }

    private static ConcurrentMap<Boolean, JsonArray> ruleSplit(final Collection<KRuleTerm> rules, final JsonArray input,
                                                               final BiFunction<Collection<KRuleTerm>, JsonObject, JsonObject> eachFn) {
        final JsonArray valid = new JsonArray();
        final JsonArray inValid = new JsonArray();
        Ut.itJArray(input).forEach(item -> {
            final JsonObject validItem = eachFn.apply(rules, item);
            if (Objects.isNull(validItem)) {
                inValid.add(item);
            } else {
                valid.add(validItem);
            }
        });
        return new ConcurrentHashMap<Boolean, JsonArray>() {
            {
                this.put(Boolean.TRUE, valid);
                this.put(Boolean.FALSE, inValid);
            }
        };
    }
}