org.babyfish.jimmer.sql.runtime.ExecutionPurpose 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
The newest version!
package org.babyfish.jimmer.sql.runtime;

import org.babyfish.jimmer.sql.ast.mutation.QueryReason;

public interface ExecutionPurpose {

    ExecutionPurpose QUERY = new SimpleExecutionPurpose(Type.QUERY);
    ExecutionPurpose UPDATE = new SimpleExecutionPurpose(Type.UPDATE);
    ExecutionPurpose LOAD = new SimpleExecutionPurpose(Type.LOAD);
    ExecutionPurpose EXPORT = new SimpleExecutionPurpose(Type.EXPORT);
    ExecutionPurpose MUTATE = new SimpleExecutionPurpose(Type.MUTATE);
    ExecutionPurpose EVICT = new SimpleExecutionPurpose(Type.EVICT);

    static Command delete(QueryReason queryReason) {
        return new DeletePurpose(queryReason);
    }

    static Command command(QueryReason queryReason) {
        return new CommandPurpose(queryReason);
    }

    Type getType();

    interface Command extends ExecutionPurpose {
        QueryReason getQueryReason();
    }

    enum Type {
        QUERY,
        UPDATE,
        DELETE,
        LOAD,
        EXPORT,
        MUTATE,
        EVICT,
        COMMAND
    }
}

class SimpleExecutionPurpose implements ExecutionPurpose {

    private final Type type;

    SimpleExecutionPurpose(Type type) {
        this.type = type;
    }

    @Override
    public Type getType() {
        return type;
    }

    @Override
    public String toString() {
        return type.name();
    }
}

class DeletePurpose implements ExecutionPurpose.Command {

    private final QueryReason queryReason;

    DeletePurpose(QueryReason queryReason) {
        this.queryReason = queryReason;
    }

    @Override
    public Type getType() {
        return Type.COMMAND;
    }

    public QueryReason getQueryReason() {
        return queryReason;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;

        DeletePurpose dp = (DeletePurpose) o;
        return queryReason == dp.queryReason;
    }

    @Override
    public int hashCode() {
        return queryReason.hashCode();
    }

    @Override
    public String toString() {
        if (queryReason == QueryReason.NONE) {
            return "DELETE";
        }
        return "DELETE(" + queryReason.name() + ")";
    }
}

class CommandPurpose implements ExecutionPurpose.Command {

    private final QueryReason queryReason;

    CommandPurpose(QueryReason queryReason) {
        this.queryReason = queryReason;
    }

    @Override
    public Type getType() {
        return Type.COMMAND;
    }

    public QueryReason getQueryReason() {
        return queryReason;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;

        CommandPurpose cp = (CommandPurpose) o;
        return queryReason == cp.queryReason;
    }

    @Override
    public int hashCode() {
        return queryReason.hashCode();
    }

    @Override
    public String toString() {
        if (queryReason == QueryReason.NONE) {
            return "COMMAND";
        }
        return "COMMAND(" + queryReason.name() + ")";
    }
}