Refactored DBBean and added DBColumn annotation

2015-12-15 23:53:35 +01:00 · 2015-12-15 23:53:35 +01:00 · 74e8c4aa68
commit 74e8c4aa68
parent 431d238c19
4 changed files with 124 additions and 97 deletions
--- a/src/zutil/db/bean/DBBeanConfig.java
+++ b/src/zutil/db/bean/DBBeanConfig.java
@ -0,0 +1,88 @@
+package zutil.db.bean;
+
+import zutil.log.LogUtil;
+
+import java.lang.reflect.Field;
+import java.lang.reflect.Modifier;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.logging.Logger;
+
+/**
+ * A Class that contains information about a bean
+ */
+class DBBeanConfig{
+    private static final Logger logger = LogUtil.getLogger();
+    /** This is a cache of all the initialized beans */
+    private static HashMap<String,DBBeanConfig> beanConfigs = new HashMap<String,DBBeanConfig>();
+
+
+	/** The name of the table in the DB **/
+    protected String tableName;
+    /** The name of the id column **/
+    protected String idColumn;
+    /** All the fields in the bean **/
+    protected ArrayList<Field> fields;
+
+    protected DBBeanConfig(){
+        fields = new ArrayList<Field>();
+    }
+
+
+
+    /**
+     * @return the configuration object for the specified class
+     */
+    protected static DBBeanConfig getBeanConfig(Class<? extends DBBean> c){
+        if( !beanConfigs.containsKey( c.getName() ) )
+            initBeanConfig( c );
+        return beanConfigs.get( c.getName() );
+    }
+
+    /**
+     * Caches the fields
+     */
+    private static void initBeanConfig(Class<? extends DBBean> c){
+        logger.fine("Initiating new DBBeanConfig( "+c.getName()+" )");
+        DBBeanConfig config = new DBBeanConfig();
+        // Find the table name
+        DBBean.DBTable tableAnn = c.getAnnotation(DBBean.DBTable.class);
+        if( tableAnn != null ){
+            config.tableName = tableAnn.value();
+            config.idColumn  = tableAnn.idColumn();
+        }
+        else{
+            config.tableName = c.getSimpleName();
+            config.idColumn  = "id";
+        }
+        // Add the fields in the bean and all the super classes fields
+        for(Class<?> cc = c; cc != DBBean.class ;cc = cc.getSuperclass()){
+            Field[] fields = cc.getDeclaredFields();
+            for( Field field : fields ){
+                int mod = field.getModifiers();
+                if( !Modifier.isTransient( mod ) &&
+                        !Modifier.isAbstract( mod ) &&
+                        !Modifier.isFinal( mod ) &&
+                        !Modifier.isStatic( mod ) &&
+                        !Modifier.isInterface( mod ) &&
+                        !Modifier.isNative( mod ) &&
+                        !config.fields.contains( field )){
+                    config.fields.add( field );
+                }
+            }
+            if( tableAnn == null || !tableAnn.superBean() )
+                break;
+        }
+
+        beanConfigs.put(c.getName(), config);
+    }
+
+    protected static String getFieldName(Field field){
+        String name = null;
+        if(field.getDeclaredAnnotation(DBBean.DBColumn.class) != null)
+            name = field.getDeclaredAnnotation(DBBean.DBColumn.class).value();
+        else
+            name = field.getName();
+        return name;
+    }
+}