如何在 Guava 中表示 SQL Table 和 SELECT 查询?

How to represent SQL Table and SELECT Query in Guava?

考虑一个典型的 客户 table:

和一个典型的SQL模式匹配查询:

SELECT FirstName, LastName, City 

FROM Customers 

WHERE FirstName LIKE '%ab%'

OR LastName LIKE '%an%'

我们如何使用 Google 番石榴库 Table 表示此 table 和 SQL SELECT 查询集合和 Predicates?

编辑

正如 mfulton26 所提到的,Guava Table 可能不是最理想的相当于数据库的数据结构 table。

那么,内存中数据table最合适的数据结构能够提供:

1) 迭代(可能使用迭代器)

2) 过滤器(可能带有谓词)

3) 具有索引的多个数据列可实现快速访问。

Table

public class CustomerTable {

    public enum Column {
        FIRST_NAME, LAST_NAME, ADDRESS_LINE1, CITY, STATE_PROVINCE_CD, POSTAL_CODE;
    }

    private Table<Integer, Column, String> table = HashBasedTable.create();

    @Override
    public String toString() {
        return table.toString();
    }

    public void createRow(String[] values) {
        if (Column.values().length != values.length) {
            throw new IllegalArgumentException();
        }
        Integer rowNum = table.rowKeySet().size() + 1;
        for(int i=0; i < values.length; i ++) {
            table.put(rowNum, Column.values()[i], values[i]);
        }
    }

    public Table<Integer, Column, String> query(Predicate<Map<Column, String>> query) {
        return query(query, allOf(Column.class));
    }

    public Table<Integer, Column, String> query(Predicate<Map<Column, String>> query, EnumSet<Column> columns) {
        Map<Integer, Map<Column, String>> filtered = Maps.filterValues(table.rowMap(), query);
        return createResultTable(filtered, columns);
    }

    private Table<Integer, Column, String> createResultTable(Map<Integer, Map<Column, String>> resultMap, final EnumSet<Column> columns) {

        int i = 0;
        Table<Integer, Column, String> result = HashBasedTable.create();
        for (Map<Column, String> row : resultMap.values()) {
            i++;
            for (Column column : row.keySet()){
                if (columns.contains(column)) {
                    result.put(i, column, row.get(column));
                }
            }
        }
        return result;
    }

谓词

class LikePredicate implements Predicate<Map<CustomerTable.Column, String>> {

    private Column column;
    private String value;

    public LikePredicate(Column column, String value) {
        this.column = column;
        this.value = value;
    }

    @Override
    public boolean apply(Map<Column, String> input) {
        return input.get(column) != null && input.get(column).contains(value);
    }

    public static LikePredicate like(Column column, String value) {
        return new LikePredicate(column, value);
    }
}

用法示例

public static void main(String[] args) {

    CustomerTable customerTable = new CustomerTable();
    customerTable.createRow(new String[]{"Ben", "Miller", "101 Candy Rd.", "Redmond", "WA", "98052"});
    customerTable.createRow(new String[]{"Garret", "Vargas", "10203 Acorn Avenue", "Calgary", "AB", "T2P 2G8"});
    //Create other rows or read rows from a file


    Table<Integer, Column, String> result;
    /*
    SELECT FirstName, LastName, City 
    FROM Customers 
    WHERE FirstName LIKE '%ab%'
          OR LastName LIKE '%an%'
    */
    result = customerTable.query(or(like(Column.FIRST_NAME, "ab"), like(Column.LAST_NAME, "an")),
            EnumSet.of(Column.FIRST_NAME, Column.LAST_NAME, Column.CITY));

    System.out.println(result);
}

请注意 SQL Table 通常不会映射到 Guava Table。 Guava 的 Table 适用于需要两个索引的情况(参见 NewCollectionTypesExplained · google/guava Wiki)。 SQL Table 结果集通常表示为简单的 CollectionIterable.

话虽如此,以下是如何 "query" 番石榴 table:

Java 8

table.rowMap().values().stream().filter(row -> {
    return row.get("FirstName").contains("ab") || row.get("LastName").contains("an");
});

Java 7/6

FluentIterable.from(table.rowMap().values()).filter(new Predicate<Map<String, String>>() {
    @Override
    public boolean apply(Map<String, String> row) {
        return row.get("FirstName").contains("ab") || row.get("LastName").contains("an");
    }
});

另请注意,Guava 的 Table 不像数据库 table,您可以在其中添加额外的索引等。您只有两个索引:行和列。