001package gudusoft.gsqlparser.dlineage.dataflow.model;
002
003import gudusoft.gsqlparser.EDbVendor;
004import gudusoft.gsqlparser.dlineage.dataflow.listener.DataFlowHandleListener;
005
006import java.util.Arrays;
007import java.util.HashSet;
008import java.util.Set;
009
010public class DataflowRemoveOption implements Cloneable {
011
012    private boolean removeVariable;
013
014    private boolean removeCursor;
015
016    private Set<ResultSetType> removeResultSetTypes = new HashSet<ResultSetType>();
017
018    public boolean isRemoveVariable() {
019        return removeVariable;
020    }
021
022    public void setRemoveVariable(boolean removeVariable) {
023        this.removeVariable = removeVariable;
024    }
025
026    public boolean isRemoveCursor() {
027        return removeCursor;
028    }
029
030    public void setRemoveCursor(boolean removeCursor) {
031        this.removeCursor = removeCursor;
032    }
033
034    public Set<ResultSetType> getRemoveResultSetTypes() {
035        return removeResultSetTypes;
036    }
037
038    public void setRemoveResultSetTypes(ResultSetType... types) {
039        if(types!=null) {
040            removeResultSetTypes.addAll(Arrays.asList(types));
041        }
042    }
043
044    public void setRemoveResultSetTypes(String... types) {
045        if (types != null) {
046            for (String type : types) {
047                ResultSetType resultSetType = ResultSetType.of(type);
048                if (resultSetType != null) {
049                    removeResultSetTypes.add(resultSetType);
050                }
051            }
052        }
053    }
054}