primarykey.java

来自「cwbbs 云网论坛源码」· Java 代码 · 共 291 行

package cn.js.fan.db;

import java.util.*;
import java.io.Serializable;

public class PrimaryKey implements Serializable, Cloneable {
    public static final int TYPE_STRING = 0;
    public static final int TYPE_INT = 1;
    public static final int TYPE_COMPOUND = 2; 
    public static final int TYPE_LONG = 3;

    HashMap keys; 

    public PrimaryKey() {

    }

    public PrimaryKey(String name, int type) {
        this.name = name;
        this.type = type;
    }

    public PrimaryKey(HashMap keys) {
        this.keys = keys;
        this.type = this.TYPE_COMPOUND;
    }

    public PrimaryKey(String name, Object value) {
        this.name = name;

        if (value.getClass().isInstance(new Integer(0))) {
            this.type = TYPE_INT;
            this.intValue = ((Integer)value).intValue();
        }
        if (value.getClass().isInstance(new Long(0))) {
            this.type = TYPE_LONG;
            this.longValue = ((Long)value).longValue();
        }
        if (value.getClass().isInstance(new String(""))) {
            this.type = TYPE_STRING;
            this.strValue = (String)value;
        }
    }

    public Object clone() {
        PrimaryKey o = null;
        try {
            o = (PrimaryKey)super.clone();
        } catch (CloneNotSupportedException e) {
            e.printStackTrace();
        }
        if (keys!=null) {
            o.keys = new HashMap();
            Iterator ir = keys.keySet().iterator();
            while (ir.hasNext()) {
                String key = (String)ir.next();
                KeyUnit ku = (KeyUnit)((KeyUnit)keys.get(key)).clone();
                o.keys.put(key, ku);
            }
        }
        return o;
    }

    public void setValue(Object value) {
        if (value.getClass().isInstance(new Integer(0))) {
            this.type = TYPE_INT;
            this.intValue = ((Integer)value).intValue();
        }
        else  if (value.getClass().isInstance(new String(""))) {
            this.type = TYPE_STRING;
            this.strValue = (String)value;
        }
        else if (value.getClass().isInstance(new Long(0))) {
            this.type = TYPE_LONG;
            this.longValue = ((Long)value).longValue();
        }
        else if (value.getClass().isInstance(new HashMap())) {
            this.type = TYPE_COMPOUND;
            this.keys = (HashMap)value;
        }
        else {
            throw new IllegalArgumentException("PrimaryKey setValue: Object value " + value +
                         " are not valid type. value.getClass()=" + value.getClass() + " Object=" + value);
        }
    }

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

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

    public int getIntValue() {
        return intValue;
    }

    public long getLongValue() {
        return longValue;
    }

    public int getType() {
        return type;
    }

    
    public int getKeyCount() {
        if (type==this.TYPE_INT || type==this.TYPE_STRING || type==this.TYPE_LONG)
            return 1;
        if (type==this.TYPE_COMPOUND) {
            if (keys!=null) {
                return keys.size();
            }
            else
                return 0;
        }
        return 0;
    }

    public HashMap getKeys() {
        return keys;
    }

    public String getName() {
        if (type==this.TYPE_INT || type==this.TYPE_STRING || type==this.TYPE_LONG)
            return name;
        else {
            if (keys!=null) {
                Iterator ir = keys.keySet().iterator();
                String str = "";
                while (ir.hasNext()) {
                    String objname = (String)ir.next();
                    if (str.equals(""))
                        str += objname;
                    else
                        str += "," + objname;
                }
                return str;
            }
            else
                return null;
        }
    }

    public String getStrValue() {
        return strValue;
    }

    public Object getValue() {
        if (type==TYPE_STRING)
            return strValue;
        else if (type==TYPE_INT)
            return new Integer(intValue);
        else if (type==TYPE_LONG)
            return new Long(longValue);
        else if (type==this.TYPE_COMPOUND) {
            if (keys!=null) {
                Iterator ir = keys.keySet().iterator();
                String str = "";
                while (ir.hasNext()) {
                    String keyname = (String)ir.next();
                    KeyUnit ku = (KeyUnit)keys.get(keyname);
                    if (str.equals(""))
                        str += ku.getStrValue();
                    else
                        str += "|" + ku.getStrValue();
                }
                return str;
            }
            else
                return null;
        }
        else
            return null;
    }

    public Object getKeyValue(String keyName) {
        if (keys!=null) {
            KeyUnit ku = (KeyUnit)keys.get(keyName);
            if (ku==null) {
                throw new IllegalArgumentException("keyName=" + keyName + " is not exist.");
            }
            return ku.getValue();
        }
        else
            return null;
    }

    public void setKeyValue(String keyName, Object value) {
        KeyUnit ku = (KeyUnit)keys.get(keyName);
        if (ku!=null) {
            
            ku.setValue(value);
        }
        else {
            
            keys.put(keyName, new KeyUnit(value));
        }
    }

    public long getKeyLongValue(String keyName) {
        Object obj = getKeyValue(keyName);
        if (obj==null) {
            return -65536;
        }
        else
            return ((Long)obj).longValue();
    }

    public int getKeyIntValue(String keyName) {
        Object obj = getKeyValue(keyName);
        if (obj==null) {
            return -65536;
        }
        else
            return ((Integer)obj).intValue();
    }

    public String getKeyStrValue(String keyName) {
        Object obj = getKeyValue(keyName);
        if (obj==null) {
            return null;
        }
        else
            return (String)obj;
    }

    public Object[] toObjectArray() {
        
        
        if (type == TYPE_STRING)
            return new Object[] {strValue};
        else if (type == TYPE_INT)
            return new Object[] {new Integer(intValue)};
        else if (type == TYPE_LONG)
            return new Object[] {new Long(longValue)};
        else if (type == TYPE_COMPOUND) {
            if (keys != null) {
                Object[] objs = new Object[keys.size()];
                Iterator ir = keys.keySet().iterator();
                int k = 0;
                while (ir.hasNext()) {
                    String keyname = (String) ir.next();
                    KeyUnit ku = (KeyUnit) keys.get(keyname);
                    objs[ku.getOrders()] = ku.getValue();
                    
                    k++;
                }
                return objs;
            } else
                return null;
        } else
            return null;
    }

    public String toString() {
        if (type==TYPE_STRING)
            return strValue;
        else if (type==TYPE_INT)
            return "" + intValue;
        else if (type==TYPE_LONG)
            return "" + longValue;
        else if (type==TYPE_COMPOUND) {
            if (keys!=null) {
                Iterator ir = keys.keySet().iterator();
                String str = "";
                while (ir.hasNext()) {
                    String keyname = (String)ir.next();
                    KeyUnit ku = (KeyUnit)keys.get(keyname);
                    if (str.equals(""))
                        str += ku.getStrValue();
                    else
                        str += "|" + ku.getStrValue();
                }
                return str;
            }
            else
                return "";
        }
        else
            return "";
    }

    private int type;
    private int intValue;
    private long longValue;
    private String strValue;
    private String name;
}