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

import java.util.ArrayList;
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;

@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;
        }
    
}