ddl-manager/lib/cache/trigger-builder/commutative/CommutativeTriggerBuilder.ts

store postgres procedures and triggers in files

import { AbstractTriggerBuilder } from "../AbstractTriggerBuilder";
import { buildCommutativeBody } from "../body/buildCommutativeBody";
import { Update, Expression } from "../../../ast";
import { SetItemsFactory } from "../../processor/SetItemsFactory";

export class CommutativeTriggerBuilder
extends AbstractTriggerBuilder {

    private setItems = new SetItemsFactory(this.context);

    createTriggers() {
        return [{
            trigger: this.createDatabaseTriggerOnDIU(),
            procedure: this.createDatabaseFunction(
                this.createBody()
            )
        }];
    }

    protected createBody() {
        const deltaUpdate = new Update({
            table: this.context.cache.for.toString(),
            set: this.setItems.plus(),
            where: this.conditions.simpleWhere("new")
        });

        const body = buildCommutativeBody(
            this.needListenInsert(),
            this.conditions.hasMutableColumns(),
            this.conditions.noChanges(),
            {
                hasReferenceWithoutJoins: this.conditions.hasReferenceWithoutJoins("old"),
                needUpdate: this.conditions.filtersWithJoins("old"),
                update: new Update({
                    table: this.context.cache.for.toString(),
                    set: this.setItems.minus(),
                    where: this.conditions.simpleWhere("old")
                })
            },
            {
                hasReferenceWithoutJoins: this.conditions.hasReferenceWithoutJoins("new"),
                needUpdate: this.conditions.filtersWithJoins("new"),
                update: new Update({
                    table: this.context.cache.for.toString(),
                    set: this.setItems.plus(),
                    where: this.conditions.simpleWhere("new")
                })
            },
            {
                needUpdate: this.conditions.noReferenceChanges(),
                update: deltaUpdate,
                exitIf: this.conditions.exitFromDeltaUpdateIf(),
                old: {
                    needUpdate: this.conditions.needUpdateConditionOnUpdate("old"),
                    update: this.needUpdateInDelta() ? new Update({
                        table: this.context.cache.for.toString(),
                        set: this.setItems.minus(),
                        where: this.conditions.simpleWhereOnUpdate("old")
                    }) : undefined
                },
                new: {
                    needUpdate: this.conditions.needUpdateConditionOnUpdate("new"),
                    update: this.needUpdateInDelta() ? new Update({
                        table: this.context.cache.for.toString(),
                        set: this.setItems.plus(),
                        where: this.conditions.simpleWhereOnUpdate("new")
                    }) : undefined
                }
            }
        );
        
        return body;
    }

    protected needListenInsert() {
        if ( this.context.withoutInsertCase() ) {
            return false;
        }
        
        const isReferenceByTriggerTableId = this.context.referenceMeta
            .expressions.some(condition => {
                if ( !condition.isBinary("=") ) {
                    return false;
                }
                const [left, right] = condition.splitBy("=");
                return (
                    this.isTriggerTableId(left) ||
                    this.isTriggerTableId(right) 
                );
            });

        return !isReferenceByTriggerTableId;
    }

    private isTriggerTableId(condition: Expression): boolean {
        const columnRefs = condition.getColumnReferences();
        const isTriggerTableIdColumn = (
            columnRefs.length === 1 &&
            columnRefs[0].name === "id" &&
            this.context.isColumnRefToTriggerTable(columnRefs[0])
        );
        return isTriggerTableIdColumn;
    }

    private needUpdateInDelta() {
        const hasCondition = !!this.conditions.needUpdateConditionOnUpdate("new");
        const hasUnknownReferenceExpressions = this.context.referenceMeta.unknownExpressions.length;

        return (
            hasCondition ||
            hasUnknownReferenceExpressions
        );
    }
}