constraint.java

来自「opennms得相关源码 请大家看看」· Java 代码 · 共 251 行

//
// This file is part of the OpenNMS(R) Application.
//
// OpenNMS(R) is Copyright (C) 2002-2005 The OpenNMS Group, Inc.  All rights reserved.
// OpenNMS(R) is a derivative work, containing both original code, included code and modified
// code that was published under the GNU General Public License. Copyrights for modified 
// and included code are below.
//
// OpenNMS(R) is a registered trademark of The OpenNMS Group, Inc.
//
// The code in this file is Copyright (C) 2004 DJ Gregor.
//
// Based on install.pl which was Copyright (C) 1999-2001 Oculan Corp.  All rights reserved.
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
//
// For more information contact:
//      OpenNMS Licensing       <[e-mail]>
//      http://www.opennms.org/
//      http://www.opennms.com/
//

package org.opennms.install;

import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class Constraint {
    public static final int PRIMARY_KEY = 1;

    public static final int FOREIGN_KEY = 2;

    private String m_name;

    private int m_type;

    private String m_column;

    private String m_ftable;

    private String m_fcolumn;

    private String m_fdeltype;

    public Constraint(String constraint) throws Exception {
        this.parse(constraint);
    }

    public Constraint(String name, String column) {
        m_name = name;
        m_type = PRIMARY_KEY;
        m_column = column;
    }

    public Constraint(String name, String column, String ftable, String fcolumn, String fdeltype) throws Exception {
        m_name = name;
        m_type = FOREIGN_KEY;
        m_column = column;
        m_ftable = ftable;
        m_fcolumn = fcolumn;

        setForeignDelType(fdeltype);
    }

    public String getName() {
        return m_name;
    }

    public void setName(String name) {
        m_name = name.toLowerCase();
    }

    public int getType() {
        return m_type;
    }

    public void setType(int type) {
        m_type = type;
    }

    public String getColumn() {
        return m_column;
    }

    public void setColumn(String column) {
        m_column = column.toLowerCase();
    }

    public String getForeignTable() {
        return m_ftable;
    }

    public void setForeignTable(String ftable) {
        m_ftable = ftable.toLowerCase();
    }

    public String getForeignColumn() {
        return m_fcolumn;
    }

    public void setForeignColumn(String fcolumn) {
        m_fcolumn = fcolumn.toLowerCase();
    }

    public String getForeignDelType() {
        return m_fdeltype;
    }

    public void setForeignDelType(String fdeltype) throws Exception {
        if (fdeltype.equals("a") || fdeltype.equals("c")) {
            m_fdeltype = fdeltype;
        } else {
            throw new Exception("confdeltype \"" + fdeltype + "\" unknown");
        }
    }

    public String toString() {
        StringBuffer b = new StringBuffer();

        b.append("constraint ");
        b.append(m_name);

        switch (m_type) {
        case PRIMARY_KEY:
            b.append(" primary key (");
            break;

        case FOREIGN_KEY:
            b.append(" foreign key (");
            break;
        }

        b.append(m_column);
        b.append(")");

        if (m_type == FOREIGN_KEY) {
            b.append(" references ");
            b.append(m_ftable);
            b.append(" (");
            b.append(m_fcolumn);
            b.append(")");
        }

        if ("c".equals(m_fdeltype)) {
            b.append(" on delete cascade");
        }

        return b.toString();
    }

    public void parse(String constraint) throws Exception {
        Matcher m;

        m = Pattern.compile("(?i)constraint (\\S+) " + "primary key \\((\\S+)\\)").matcher(constraint);
        if (m.matches()) {
            setName(m.group(1));
            setType(PRIMARY_KEY);
            setColumn(m.group(2));
            return;
        }

        m = Pattern.compile("(?i)constraint (\\S+) " + "foreign key \\((\\S+)\\) " + "references (\\S+)" + "(?: \\((\\S+)\\))?" + "(\\s+on\\s+delete\\s+cascade)?").matcher(constraint);
        if (m.matches()) {
            setName(m.group(1));
            setType(FOREIGN_KEY);
            setColumn(m.group(2));
            setForeignTable(m.group(3));
            if (m.group(4) == null) {
                setForeignColumn(m.group(2));
            } else {
                setForeignColumn(m.group(4));
            }
            if (m.group(5) == null) {
                setForeignDelType("a");
            } else {
                setForeignDelType("c");
            }
            return;
        }

        throw new Exception("Cannot parse constraint: " + constraint);
    }

    public boolean equals(Object other_o) {
        return equals(other_o, false);
    }

    public boolean equals(Object other_o, boolean ignoreFdelType) {
        Constraint other = (Constraint) other_o;

        if ((m_name == null && other.getName() != null) || (m_name != null && other.getName() == null)) {
            return false;
        }
        if (m_name != null && other.getName() != null && !m_name.equals(other.getName())) {
            return false;
        }

        if (m_type != other.getType()) {
            return false;
        }

        if ((m_column == null && other.getColumn() != null) || (m_column != null && other.getColumn() == null)) {
            return false;
        }
        if (m_column != null && other.getColumn() != null && !m_column.equals(other.getColumn())) {
            return false;
        }

        if ((m_ftable == null && other.getForeignTable() != null) || (m_ftable != null && other.getForeignTable() == null)) {
            return false;
        }
        if (m_ftable != null && other.getForeignTable() != null && !m_ftable.equals(other.getForeignTable())) {
            return false;
        }

        if ((m_fcolumn == null && other.getForeignColumn() != null) || (m_fcolumn != null && other.getForeignColumn() == null)) {
            return false;
        }
        if (m_fcolumn != null && other.getForeignColumn() != null && !m_fcolumn.equals(other.getForeignColumn())) {
            return false;
        }

        if (!ignoreFdelType) {
            if ((m_fdeltype == null && other.getForeignDelType() != null) || (m_fdeltype != null && other.getForeignDelType() == null)) {
                return false;
            }
            if (m_fdeltype != null && other.m_fdeltype != null && !m_fdeltype.equals(other.getForeignDelType())) {
                return false;
            }
        }

        return true;
    }

    public int hashCode() {
        return m_name.hashCode() + new Integer(m_type).hashCode() + m_column.hashCode() + m_ftable.hashCode() + m_fcolumn.hashCode() + m_fdeltype.hashCode();
    }
}