手把手教你寫一個java的orm(五)

何白白發表於2019-01-23

生成sql:where

上一篇裡我們實現了生成insert的sql,下面要開始實現updatedeleteselect的sql語句了。但是這些語句有一個比較麻煩的地方是:它們一般後面都會有where條件,因為在執行的時候不能把表裡所有的資料都進行操作。

所以這裡我們需要先生成條件的sql。大概是這樣的:

WHERE id = ? AND name != ? OR age >= ? 

where 後面的引數繼續用 “?” 代替。值就放在一個有序的集合中就好了。類似上一篇提到的insertColumnValues

思路

  1. 條件都是一個一個組成的,我們可以寫一個類用來描述一個條件。
  2. 寫一個工具類來快速的建立條件。
  3. 將多個條件中間用 and 或者 or 組合起來,並在最前方新增 where 就是一個完整的條件。
  4. 最後將這個條件轉成一個字串,並用一個集合將條件中的值存起來就好了。

實現

第一步

我們實現第一步,在這之前我們先看一下一個條件是有什麼組成的,例如:

1: id = ? AND
2: name != ? OR
3: age >= ? 

這裡通過觀察可以發現,每一個條件都是由一個 欄位名稱一個判斷一個佔位符 “?”和後面用於連線條件的 AND 或者 OR 所構成。這樣我們可以編寫一個類用來儲存這些資訊:

Where.java


import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * where條件 預設使用 and 連線多個條件
 *
 * @author hjx
 */
public class Where {

    protected static final String PLACEHOLDER = "#{COLUMN}";

    static final String AND = "AND ";

    static final String OR = "OR ";

    private String sql;

    private String column;

    private String connect = AND;

    private List<Object> values;

    /**
     * 是否有值(null 也代表有值)
     */
    private boolean hasValue;

    /**
     * @param column 被操作的列
     * @param sql    操作的sql
     */
    public Where(String column, String sql) {
        this.column = column;
        this.sql = sql;
        this.hasValue = false;
        this.values = new ArrayList<>();
    }

    /**
     * @param column 被操作的列
     * @param sql    操作的sql
     * @param value  sql的引數
     */
    public Where(String column, String sql, Object value) {
        this.sql = sql;
        this.column = column;
        this.values = new ArrayList<>();
        this.values.add(value);
        this.hasValue = true;
    }

    /**
     * @param column 被操作的列
     * @param sql    操作的sql
     * @param values sql的引數
     */
    public Where(String column, String sql, Object[] values) {
        this.sql = sql;
        this.column = column;
        this.values = Arrays.asList(values);
        this.hasValue = true;
    }

    public Where or() {
        this.connect = OR;
        return this;
    }

    public Where and() {
        this.connect = AND;
        return this;
    }

    /**
     * 獲取本次條件的連線符
     *
     * @return
     */
    public String getConnect() {
        return connect;
    }

    protected String getSql() {
        return sql;
    }

    protected boolean isHasValue() {
        return hasValue;
    }

    protected List<Object> getValues() {
        return values;
    }

    public String getColumn() {
        return column;
    }
}

上面中的常量 PLACEHOLDER 是作為一個佔位符使用的,下面會說道。

這樣,一個用於儲存單個條件的類就寫好了,在一個sql中有多個條件的話,只需要用一個ArrayList儲存這些條件,並按照一定的條件拼裝成sql就好了。

第二步

sql中還有一些比較常用的判斷,比如:!= , = , <= , >= 等等,我們在這裡可以建立一個工具類來快速的生成Where 這個類,可以這樣寫:

Wheres.java


import java.util.Arrays;

/**
 * 查詢條件
 * @author hjx
 */
public class Wheres {
    
    public static Where equal(final String columnName, final Object value) {
        return new Where(columnName, Where.PLACEHOLDER + " = ? ", value);
    }

    public static Where notEqual(final String columnName, final Object value) {
        return new Where(columnName, Where.PLACEHOLDER + " != ? ", value);
    }

    public static Where not(final String columnName, final Object value) {
        return new Where(columnName, Where.PLACEHOLDER + " <> ? ", value);
    }

    public static Where isNotNull(final String columnName) {
        return new Where(columnName, Where.PLACEHOLDER + " IS NOT NULL ");
    }

    public static Where isNull(final String columnName) {
        return new Where(columnName, Where.PLACEHOLDER + " IS NULL ");
    }

    public static Where greater(final String columnName, final Object value, final boolean andEquals) {
        if (andEquals) {
            return new Where(columnName, Where.PLACEHOLDER + " >= ? ", value);
        }
        return new Where(columnName, Where.PLACEHOLDER + " > ? ", value);
    }

    public static Where less(final String columnName, final Object value, final boolean andEquals) {
        if (andEquals) {
            return new Where(columnName, Where.PLACEHOLDER + " <= ? ", value);
        }
        return new Where(columnName, Where.PLACEHOLDER + " < ? ", value);
    }

    public static Where like(final String columnName, final Object value) {
        return new Where(columnName, Where.PLACEHOLDER + " like ? ", value);
    }

    public static Where betweenAnd(final String columnName, final Object value1st, final Object value2nd) {
        return new Where(columnName, Where.PLACEHOLDER + " between ? and ? ", new Object[]{value1st, value2nd});
    }

    public static Where in(final String columnName, final Object[] values) {
        Object[] sqlVal = values;
        if (sqlVal.length == 0) {
            sqlVal = new Object[]{null};
        }
        StringBuffer inSql = new StringBuffer();
        inSql.append(Where.PLACEHOLDER);
        inSql.append(" IN ( ");
        String[] strings = StringUtils.repeat("?", sqlVal.length);
        inSql.append(StringUtils.join(Arrays.asList(strings), ", "));
        inSql.append(" ) ");
        return new Where(columnName, inSql.toString(), sqlVal);
    }

}

這裡只是簡單的列出了一些常用的判斷條件,如果有特殊需要的自己再加進去就好了。

關於常量 PLACEHOLDER 是這麼一回事:

在生成sql 的時候,我需要做一些欄位上的驗證。這裡在sql中使用一個佔位符放進sql中,真正參與條件的欄位放在另外一個屬性中儲存。這樣在真正生成sql的時候可以驗證條件中的欄位在不在表中,如果存在的話將欄位和佔位符進行替換就好了。並且如果使用的是屬性名稱的話,也可以根據名稱找到對應的表的欄位名。

第三步

通過上面的程式碼,我們可以很方便的建立條件了。現在我們將這些條件組裝成我們需要的完整的sql。

注意:這裡的程式碼可能和我的github上的不太一樣,因為這裡只講一下思路,具體的怎麼將所有的程式碼組裝起來讓它成為一個完整的專案,每個人都不一樣。所以~~~ 嘿嘿。

現在開始:

我們還是以之前寫的User.java為例子

List<Where> wheres = Arrays.asList(
        Wheres.equal("name", "李叔叔"),
        Wheres.notEqual("status", 1),
        Wheres.in("age", new Integer[]{1, 2, 3, 4, 5}),
        Wheres.greater("age", 20, true)
);
List<Object> sqlValue = new ArrayList<>();
StringBuilder sql = new StringBuilder();
if (wheres.size() != 0) {
    sql.append("WHERE ");
    for (int i = 0; i < wheres.size(); i++) {
        Where where = wheres.get(i);
        if (i != 0) {
            sql.append(where.getConnect());
        }
        String column = where.getColumn();
        String whereSql = where.getSql();
        sql.append(
            //這裡獲取真實sql
            whereSql.replace(Where.PLACEHOLDER, getColumnName(column))
        );
        //因為有些條件中的引數可能是有多個
        List<Object> values = where.getValues();
        for (int j = 0; j < values.size(); j++) {
            sqlValue.add(values.get(j));
        }
    }
}
System.out.println(sql.toString());
System.out.println(sqlValue.toString());

這裡說明一下:getColumnName(String name) ,這個方法是根據引數獲取真正的欄位名稱的方法。因為這個條件中可能傳入的是java屬性的名稱而不是表的欄位名稱,需要轉換成為真正的表的欄位名。這一步也是從之前生成的對映中獲取的。順便還能驗證一下表中有沒有這個欄位。這個方法我就不貼出來了,github上有。

輸出結果:

WHERE name = ? AND status != ? AND age IN ( ?, ?, ?, ?, ? ) AND age >= ? 
[李叔叔, 1, 1, 2, 3, 4, 5, 20]

這裡一個where就寫好了,並且也可以拿到條件中的引數了。

剩下的就是後面的單獨生成updatedeleteselect 型別sql的操作了。

我下一篇再寫~

相關文章