package gudusoft.gsqlparser.dlineage.dataflow.model.xml;

import java.util.ArrayList;
import java.util.Comparator;
import java.util.HashSet;
import java.util.List;
import java.util.Objects;
import java.util.Set;

import javax.xml.bind.annotation.XmlAttribute;
import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlType;

import gudusoft.gsqlparser.util.SQLUtil;

@XmlType(propOrder = {"id", "type", "function", "effectType", "partition", "clause", "joinType", "target", "caller", "sources", "callees", "processId", "processType", "procedureId", "sqlHash", "sqlComment", "condition", "timestampMin", "timestampMax", "callStmt", "callCoordinate"})
public class relationship implements Cloneable {

    private String id;

    private String type;

    private String function;

    private String effectType;

    private String clause;

    private String joinType;

    private targetColumn target;

    private targetColumn caller;

    private List<sourceColumn> sources;

    private List<sourceColumn> callees;

    private String processId;

    private String processType;
    
    private String partition;

    private String condition;

    private String sqlHash;

    private String sqlComment;
    
    private String procedureId;

    private String timestampMin;

    private String timestampMax;

    private Boolean builtIn;
    
    private String callStmt;
    
        private String callCoordinate;

    @XmlAttribute(required = false)
    public String getId() {
        return id;
    }

    public void setId(String id) {
        this.id = id;
    }

    @XmlAttribute(required = false)
    public String getType() {
        return type;
    }

    public void setType(String type) {
        this.type = type;
    }

    @XmlElement(name = "target", required = false)
    public targetColumn getTarget() {
        return target;
    }

    public void setTarget(targetColumn target) {
        this.target = target;
    }

    @XmlElement(name = "source", required = false)
    public List<sourceColumn> getSources() {
        if (this.sources == null) {
            this.sources = new ArrayList<sourceColumn>();
        }
        return sources;
    }

    public void setSources(List<sourceColumn> sources) {
        this.sources = sources;
    }

    @XmlAttribute(required = false)
    public String getClause() {
        return clause;
    }

    public void setClause(String clause) {
        this.clause = clause;
    }

    @XmlAttribute(required = false)
    public String getCondition() {
        return condition;
    }

    public void setCondition(String condition) {
        this.condition = condition;
    }

    @XmlAttribute(required = false)
    public String getJoinType() {
        return joinType;
    }

    public void setJoinType(String joinType) {
        this.joinType = joinType;
    }

    public boolean isDataFlow() {
        return "fdd".equals(type);
    }

    @XmlAttribute(required = false)
    public String getFunction() {
        return function;
    }

    public void setFunction(String function) {
        this.function = function;
    }

    @XmlAttribute(required = false)
    public String getEffectType() {
        return effectType;
    }

    public void setEffectType(String effectType) {
        this.effectType = effectType;
    }
    
    @XmlAttribute(required = false)
    public String getPartition() {
                return partition;
        }

        public void setPartition(String partition) {
                this.partition = partition;
        }

        @Override
    public Object clone() throws CloneNotSupportedException {
        return super.clone();
    }

    @XmlAttribute(required = false)
    public String getProcessId() {
        return processId;
    }

    public void setProcessId(String processId) {
        this.processId = processId;
    }

    @XmlAttribute(required = false)
    public String getProcessType() {
        return processType;
    }

    public void setProcessType(String processType) {
        this.processType = processType;
    }

    @XmlAttribute(required = false)
    public String getProcedureId() {
                return procedureId;
        }

        public void setProcedureId(String procedureId) {
                this.procedureId = procedureId;
        }

        @XmlAttribute(required = false)
    public String getSqlHash() {
        return sqlHash;
    }

    public void setSqlHash(String sqlHash) {
        this.sqlHash = sqlHash;
    }

    @XmlAttribute(required = false)
    public String getSqlComment() {
        return sqlComment;
    }

    public void setSqlComment(String sqlComment) {
        this.sqlComment = sqlComment;
    }

    @XmlAttribute(required = false)
    public String getTimestampMin() {
        return timestampMin;
    }

    public void setTimestampMin(String timestampMin) {
        this.timestampMin = timestampMin;
    }

    @XmlAttribute(required = false)
    public String getTimestampMax() {
        return timestampMax;
    }

    public void setTimestampMax(String timestampMax) {
        this.timestampMax = timestampMax;
    }

    @XmlElement(name = "caller", required = false)
    public targetColumn getCaller() {
        return caller;
    }

    public void setCaller(targetColumn caller) {
        this.caller = caller;
    }

    @XmlElement(name = "callee", required = false)
    public List<sourceColumn> getCallees() {
        if (this.callees == null) {
            this.callees = new ArrayList<sourceColumn>();
        }
        return callees;
    }

    public void setCallees(List<sourceColumn> callees) {
        this.callees = callees;
    }

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

    @Override
    public int hashCode() {
        return Objects.hash(id);
    }

    private Set<String> sourceIds = new HashSet<String>();
        public void addSource(sourceColumn source) {
                if(sourceIds.contains(source.getId())) {
                        return;
                }
                getSources().add(source);
                sourceIds.add(source.getId());
        }

    @XmlAttribute(required = false)
    public Boolean getBuiltIn() {
        return builtIn;
    }

    public void setBuiltIn(Boolean builtIn) {
        this.builtIn = builtIn;
    }

    @XmlAttribute(required = false)
        public String getCallStmt() {
                return callStmt;
        }

        public void setCallStmt(String callStmt) {
                this.callStmt = callStmt;
        }

    @XmlAttribute(required = false)
        public String getCallCoordinate() {
                return callCoordinate;
        }

        public void setCallCoordinate(String callCoordinate) {
                this.callCoordinate = callCoordinate;
        }

        public String toDedupKey() {
                StringBuilder sb = new StringBuilder(256);
                appendDedupKey(sb);
                return sb.toString();
        }

        public long toDedupHash() {
                StringBuilder sb = new StringBuilder(256);
                appendDedupKey(sb);
                return SQLUtil.hash64(sb);
        }

        void appendDedupKey(StringBuilder sb) {
                if (type != null) { sb.append("type=").append(type).append('|'); }
                if (function != null) { sb.append("function=").append(function).append('|'); }
                if (effectType != null) { sb.append("effectType=").append(effectType).append('|'); }
                if (partition != null) { sb.append("partition=").append(partition).append('|'); }
                if (clause != null) { sb.append("clause=").append(clause).append('|'); }
                if (joinType != null) { sb.append("joinType=").append(joinType).append('|'); }
                if (target != null) {
                        sb.append("target={");
                        target.appendDedupKey(sb);
                        sb.append("}|");
                }
                if (caller != null) {
                        sb.append("caller={");
                        caller.appendDedupKey(sb);
                        sb.append("}|");
                }
                if (sources != null && !sources.isEmpty()) {
                        sb.append("sources=[");
                        sources.stream().sorted(Comparator.comparing(sourceColumn::getId, Comparator.nullsFirst(Comparator.<String>naturalOrder())))
                                        .forEach(s -> { sb.append('{'); s.appendDedupKey(sb); sb.append("},"); });
                        sb.setLength(sb.length() - 1);
                        sb.append("]|");
                }
                if (callees != null && !callees.isEmpty()) {
                        sb.append("callees=[");
                        callees.stream().sorted(Comparator.comparing(sourceColumn::getId, Comparator.nullsFirst(Comparator.<String>naturalOrder())))
                                        .forEach(s -> { sb.append('{'); s.appendDedupKey(sb); sb.append("},"); });
                        sb.setLength(sb.length() - 1);
                        sb.append("]|");
                }
                if (processId != null) { sb.append("processId=").append(processId).append('|'); }
                if (processType != null) { sb.append("processType=").append(processType).append('|'); }
                if (procedureId != null) { sb.append("procedureId=").append(procedureId).append('|'); }
                if (sqlHash != null) { sb.append("sqlHash=").append(sqlHash).append('|'); }
                if (sqlComment != null) { sb.append("sqlComment=").append(sqlComment).append('|'); }
                if (condition != null) { sb.append("condition=").append(condition).append('|'); }
                if (timestampMin != null) { sb.append("timestampMin=").append(timestampMin).append('|'); }
                if (timestampMax != null) { sb.append("timestampMax=").append(timestampMax).append('|'); }
                if (builtIn != null) { sb.append("builtIn=").append(builtIn).append('|'); }
                if (callStmt != null) { sb.append("callStmt=").append(callStmt).append('|'); }
                sb.append("dataFlow=").append(isDataFlow()).append('|');
                if (callCoordinate != null) { sb.append("callCoordinate=").append(callCoordinate).append('|'); }
        }

}