Java 實現簡單的SQL動態組裝工具類

時間 2019-11-09

標籤 java 實現簡單 sql 動態組裝工具欄目 Java 简体版

原文原文鏈接

初版

package com.zh.oukele.util;

import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

public class CreateSqlUtil {

    public static void main(String[] args) {

        Map<String ,Object> map = new HashMap<>();
        map.put("stuName","歐可樂");
        map.put("stuAge",20);
        map.put("stuSex","男");
        map.put("Key_stuId","ASDF");
        map.put("Key_stuSex","ASDF");
        try {
            System.out.println(getSql("table_name", "delete", map, false, ""));
        } catch (Exception e) {
            e.printStackTrace();
        }

    }

    /**
     * 動態組裝 簡單sql語法
     * @param tableName 表名
     * @param operation 操做標識符 select|delete|update ,默認爲 select
     * @param mapData 數據的map集合
     * @param useMySQL true|false , false 爲使用動態組裝SQL，true爲使用自已的sql
     * @param mySql 自已的sql
     * 注意：update 這裏，where xxx = xxx ,的時候，mapData 裏的鍵必需要有 Key_ 前綴（其餘的 並不影響到）
     *
     * @return
     * @throws Exception
     */
    public static String getSql(String tableName, String operation, Map<?,?> mapData,boolean useMySQL,String mySql) throws Exception {
        String sql = null;
        // 使用組裝sql的功能
        if( !useMySQL){
            if( !(tableName != null && !tableName.equals("") && tableName.length() > 0 ) ){
                throw new Exception(" 參數 tableName 的值爲空！");
            }else if( !(mapData != null && !mapData.equals("") && mapData.size() > 0 ) ){
                throw new Exception(" 參數 mapData 的值爲空！");
            }
            // 操做標識 默認爲 select
            String operations = "select";
            String condition = " a.* from " + tableName + " a where ";
            if( operation != null && !operation.equals("") ){
                if( operation.equals("update") || operation.equals("UPDATE") ){
                    operations = "update";
                    condition = " " + tableName + " a set ";
                }else if( operation.equals("delete") || operation.equals("DELETE") ){
                    operations = "delete";
                    condition = " from " + tableName + " a where ";
                }else if( operation.equals("insert") || operation.equals("INSERT") ){
                    operations = "insert";
                    condition = " into " + tableName + " values (";
                    String link = "";
                    Iterator<?> iterator = mapData.keySet().iterator();
                    while (iterator.hasNext()) {
                        String next = (String) iterator.next();
                        condition += link + next;
                        link = ",";
                    }
                    condition += ") values( ";
                }
            }
            String value= "";
            String link ="";
            String keyValueOperations = " where ";
            Iterator<? extends Map.Entry<?, ?>> iterator = mapData.entrySet().iterator();
            while (iterator.hasNext()) {
                Map.Entry<?, ?> next = iterator.next();
                if( next.getValue() instanceof String ){
                    value = "'" + next.getValue() +"'";
                }else {
                    value = "" + next.getValue() +"";
                }
                if( next.getKey().toString().lastIndexOf("Key_") == -1 ){
                    if( !operations.equals("insert")){
                        if( operations.equals("select") || operations.equals("delete")){
                            condition += link + "a." + next.getKey();
                            condition += "=" + value;
                            link = " and ";
                        }else {
                            condition += link + "a." + next.getKey();
                            condition += "=" + value;
                            link = ",";
                        }
                    }else {
                        condition += link + value;
                        link = ",";
                    }
                }else {
                    continue;
                }
            }

            // 組裝 insert sql 的結尾
            if( operations.equals("insert") ){
                condition += ")";
            }else if(operations.equals("update")){ // 組裝 update sql 的結尾
                condition += " where ";
                String and = "";
                Iterator<? extends Map.Entry<?, ?>> iterator1 = mapData.entrySet().iterator();
                while (iterator1.hasNext()) {
                    Map.Entry<?, ?> next = iterator1.next();
                    if( next.getValue() instanceof String ){
                        value = "'" + next.getValue() +"'";
                    }else {
                        value = "" + next.getValue() +"";
                    }
                    String key = next.getKey().toString();
                    if( key.lastIndexOf("Key_") != -1 ){
                        key =  key.substring(key.indexOf("Key_")+ 4,key.length());
                       condition += and +"a." +key + "=" + value;
                       and = " and ";
                    }
                }
            }

            sql = operations + condition;
        }else { // 不使用組裝sql的功能
            sql = mySql;
        }
        return sql;
    }
}

使用案例：java

    public static void main(String[] args) throws Exception {

        Map<String ,Object> map = new HashMap<>();
        map.put("stuName","歐可樂");
        map.put("stuAge",20);
        map.put("stuSex","");
        map.put("Key_stuId","XXX");
        map.put("Key_stuSex","VVV");

        String select = getSql1("table_name", "select", map, false, "");
        System.out.println(select);

        System.out.println();

        String insert = getSql1("table_name", "insert", map, false, "");
        System.out.println(insert);

        System.out.println();

        String delete = getSql1("table_name", "delete", map, false, "");
        System.out.println(delete);

        System.out.println();

        String update = getSql1("table_name", "update", map, false, "");
        System.out.println(update);

    }

生成的SQL語句：sql

第二版

修改版本一組裝insert語法時的一些bug，新增組裝查詢SQL時, 可以使用 a.xxx is not null 條件查詢ide

    /**
     * 動態組裝 簡單sql語法
     * @param tableName 表名
     * @param operation 操做標識符 select|delete|update ,默認爲 select
     * @param mapData 數據的map集合
     * @param useMySQL true|false , false 爲使用動態組裝SQL，true爲使用自已的sql
     * @param mySql 自已的sql
     * 注意：update 這裏，where xxx = xxx ,的時候，mapData 裏的鍵必需要有 Key_ 前綴（其餘的 並不影響到）
     *
     * @return
     * @throws Exception
     */
    public static String getSql2(String tableName, String operation, Map<?,?> mapData,boolean useMySQL,String mySql) throws Exception {
        String sql = null;
        // 使用組裝sql的功能
        if( !useMySQL){
            if( !(tableName != null && !tableName.equals("") && tableName.length() > 0 ) ){
                throw new Exception(" 參數 tableName 的值爲空！");
            }else if( !(mapData != null && !mapData.equals("") && mapData.size() > 0 ) ){
                throw new Exception(" 參數 mapData 的值爲空！");
            }
            // 鍵組裝
            // 操做標識 默認爲 select
            String operations = "select";
            String condition = " a.* from " + tableName + " a where ";
            if( operation != null && !operation.equals("") ){
                if( operation.equals("update") || operation.equals("UPDATE") ){
                    operations = "update";
                    condition = " " + tableName + " a set ";
                }else if( operation.equals("delete") || operation.equals("DELETE") ){
                    operations = "delete";
                    condition = " from " + tableName + " a where ";
                }else if( operation.equals("insert") || operation.equals("INSERT") ){
                    operations = "insert";
                    condition = " into " + tableName + " values (";
                    String link = "";
                    Iterator<?> iterator = mapData.keySet().iterator();
                    while (iterator.hasNext()) {
                        String next = (String) iterator.next();
                        if( next.lastIndexOf("Key_") == -1){
                            condition += link + next;
                            link = ",";
                        }
                    }
                    condition += ") values( ";
                }
            }

            // 值組裝
            String value= "";
            String link ="";
            String keyValueOperations = " where ";
            Iterator<? extends Map.Entry<?, ?>> iterator = mapData.entrySet().iterator();
            while (iterator.hasNext()) {
                Map.Entry<?, ?> next = iterator.next();
                if( next.getValue() instanceof String ){
                    value = "'" + next.getValue() +"'";
                }else {
                    if( next.getValue() == null ){
                        value = "";
                    }else {
                        value = "" + next.getValue() +"";
                    }
                }
                if( next.getKey().toString().lastIndexOf("Key_") == -1 ){
                    if( !operations.equals("insert")){
                        if( operations.equals("select") || operations.equals("delete")){
                            condition += link + "a." + next.getKey();
                            if( value.equals("") ){
                                condition += value;
                            }else {
                                condition += "=" + value;
                            }
                            link = " and ";
                        }else {
                            condition += link + " a." + next.getKey();
                            condition += "=" + value;
                            link = ",";
                        }
                    }else {
                        condition += link + value;
                        link = ",";
                    }
                }else {
                    continue;
                }
            }

            // 組裝 insert sql 的結尾
            if( operations.equals("insert") ){
                condition += ")";
            }else if(operations.equals("update")){ // 組裝 update sql 的結尾
                condition += " where ";
                String and = "";
                Iterator<? extends Map.Entry<?, ?>> iterator1 = mapData.entrySet().iterator();
                while (iterator1.hasNext()) {
                    Map.Entry<?, ?> next = iterator1.next();
                    if( next.getValue() instanceof String ){
                        value = "'" + next.getValue() +"'";
                    }else {
                        value = "" + next.getValue() +"";
                    }
                    String key = next.getKey().toString();
                    if( key.lastIndexOf("Key_") != -1 ){
                        key =  key.substring(key.indexOf("Key_")+ 4,key.length());
                        condition += and +"a." +key + "=" + value;
                        and = " and ";
                    }
                }
            }

            sql = operations + condition;
        }else { // 不使用組裝sql的功能
            sql = mySql;
        }
        return sql;
    }

View Code

使用案例：spa

    public static void main(String[] args) throws Exception {

        Map<String ,Object> map = new HashMap<>();
        map.put("stuName","歐可樂");
        map.put("stuAge",20);
        map.put("stuSex","");
        map.put("Key_stuId","XXX");
        map.put("Key_stuSex","VVV");

        String select = getSql2("table_name", "select", map, false, "");
        System.out.println(select);

        System.out.println();

        String insert = getSql2("table_name", "insert", map, false, "");
        System.out.println(insert);

        System.out.println();

        String delete = getSql2("table_name", "delete", map, false, "");
        System.out.println(delete);

        System.out.println();

        String update = getSql2("table_name", "update", map, false, "");
        System.out.println(update);

    }

生成的SQL語句：code

簡單動態組裝select語法案例：blog

    public static void main(String[] args) throws Exception {

        Map<String ,Object> map = new HashMap<>();
        map.put("stuName","歐可樂");
        map.put("stuSex","男");
        map.put("stuSex is not null or a.stuAge > 19 ",null);
        
        String select = getSql2("table_name", "select", map, false, "");
        System.out.println(select);

    }