intellij-code-from-ds/src/com/rthoni/intellij/codefromds/dbo/template/DataSourceDbo.java

package com.rthoni.intellij.codefromds.dbo.template;

import java.util.List;
import java.util.Vector;

/**
 * Created by robin on 11/18/16.
 */
public class DataSourceDbo {

    private String _name;

    private List<TableDataSourceDbo> _tables = new Vector<>();

    private List<TableDataSourceDbo> _types = new Vector<>();

    private List<StoredProcedureDbo> _storedProcedures = new Vector<>();

    public List<StoredProcedureDbo> getStoredProcedures() {
        return _storedProcedures;
    }

    public void addStoredProcedure(StoredProcedureDbo storedProcedures) {
        _storedProcedures.add(storedProcedures);
    }

    public String getName() {
        return _name;
    }

    public void setName(String name) {
        _name = name;
    }

    public List<TableDataSourceDbo> getTables() {
        return _tables;
    }

    public void addTable(TableDataSourceDbo table) {
        _tables.add(table);
    }

    public List<TableDataSourceDbo> getTypes() {
        return _types;
    }

    public List<TableDataSourceDbo> getTablesAndTypes() {
        List<TableDataSourceDbo> all = new Vector<>();
        all.addAll(_tables);
        all.addAll(_types);
        return all;
    }

    public void addType(TableDataSourceDbo types) {
        _types.add(types);
    }

    public TableDataSourceDbo findTable(String name)
    {
        for (TableDataSourceDbo table : _tables) {
            if (table.getName().equals(name)) {
                return table;
            }
        }
        return null;
    }

    public TableDataSourceDbo findType(String name)
    {
        for (TableDataSourceDbo table : _types) {
            if (table.getName().equals(name)) {
                return table;
            }
        }
        return null;
    }

    public TableDataSourceDbo findTypeOrTable(String name)
    {
        TableDataSourceDbo type = findType(name);
        if (type == null) {
            return findTable(name);
        }
        return type;
    }

    public StoredProcedureDbo findStoredProcedure(String name)
    {
        for (StoredProcedureDbo sp : _storedProcedures) {
            if (sp.getFullName().equals(name)) {
                return sp;
            }
        }
        return null;
    }
}