用注解编写创建表的SQL语句(2)-木庄网络博客

当前第2页返回上一页

/**
* Project Name:myannotation
* File Name:TableCreator.java
* Package Name:com.iflytek.table
* Date:2016-8-29下午10:57:52
* Copyright (c) 2016, syzhao@iflytek.com All Rights Reserved.
*
*/

package com.iflytek.table;

import java.lang.annotation.Annotation;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;

import com.iflytek.db.Constraints;
import com.iflytek.db.DBTable;
import com.iflytek.db.SQLInteger;
import com.iflytek.db.SQLString;

public class TableCreator
{
  public static void main(String[] args)
  {
    createTable(Member.class);
  }
  
  //创建表SQL语句
  private static void createTable(Class<?> cl)
  {
    //获取DBTable注解
    DBTable dbTable = cl.getAnnotation(DBTable.class);
    //判断DBTable注解是否存在
    if (dbTable == null)
    {
      System.out.println("没有找到关于DBTable");
      return;
    }
    
    //如果@DBTable注解存在获取表明 
    String tableName = dbTable.name();
    //判断表名是否存在
    if (tableName.length() < 1)
    {
      //不存在，说明默认就是类名，通过 cl.getSimpleName()获取类名并且大写
      tableName = cl.getSimpleName().toUpperCase();
    }
    
    //定义获取column的容器
    List<String> columnDefs = new ArrayList<String>();
    //循环属性字段
    //说明：getDeclaredFields()获得某个类的所有申明的字段，即包括public、private和proteced，但是不包括父类的申明字段。 
    //getFields()获得某个类的所有的公共（public）的字段，包括父类。 
    for (Field field : cl.getDeclaredFields())
    {
      //定义表字段名称变量
      String columnName = null;
      //获取字段上的注解（现在字段允许多个注解，因此返回的是数组）
      Annotation[] anns = field.getDeclaredAnnotations();
      //判断属性是否存在注解
      if (anns.length < 1)
        continue;
      
      //判断是否是我们定义的数据类型
      if (anns[0] instanceof SQLInteger)
      {
        //获取SQLInteger 注解
        SQLInteger sInt = (SQLInteger)anns[0];
        //判断是否注解的name是否有值
        if (sInt.name().length() < 1)
        {
          //如果没有值，说明是类的属性字段，获取属性并转换大写
          columnName = field.getName().toUpperCase();
        }
        else
        { //如果有值，获取设置的name值
          columnName = sInt.name();
        }
        //放到属性的容器内
        columnDefs.add(columnName + " INT " + getConstraints(sInt.constraints()));
      }
      
      //同上SQLInteger，这里不写注释了
      if (anns[0] instanceof SQLString)
      {
        SQLString sString = (SQLString)anns[0];
        if (sString.name().length() < 1)
        {
          columnName = field.getName().toUpperCase();
        }
        else
        {
          columnName = sString.name();
        }
        columnDefs.add(columnName + " VARCHAR(" + sString.value() + ")" + getConstraints(sString.constraints()));
      }
      
      //定义生成创建表的SQL语句
      StringBuilder createCommand = new StringBuilder("CREATE TABLE " + tableName + "(");
      //循环上面属性容器，
      for (String columnDef : columnDefs)
      {
        //把属性添加到sql语句中
        createCommand.append("\n  " + columnDef + ",");
        //去掉最后一个逗号
        String tableCreate = createCommand.substring(0, createCommand.length() - 1) + ");";
        //打印
        System.out.println("Table creation SQL for " + cl.getName() + " is :\n" + tableCreate);
      }
    }
  }
  
  private static String getConstraints(Constraints con)
  {
    String constraints = "";
    //判断是否为null
    if (!con.allowNull())
    {
      constraints += " NOT NULL ";
    }
    //判断是否是主键
    if (con.primaryKey())
    {
      constraints += " PRIMARY KEY ";
    }
    //是否唯一
    if (con.unique())
    {
      constraints += " UNIQUE ";
    }
    
    return constraints;
  }
}

以上代码拷贝出来，就可以运行了！

上面虽然是简单的创建表语句，但我们可以蔓延到hibernate的domain类里的注解，各种CURD ，何尝不是这样处理的呢，只是hibernate有很多东西，但是万变不离其宗，以后有机会研究一下hibernate 。

收获：

读了以后，对于注解知道为什么要这么用了，其实顾名思义就是一个注解，只是有一个处理器来处理这个注解，这对我以后用到注解方面应该有帮助的，

时间不早了，就写到这里！

结果：

标签：SQL

返回前面的内容