Examples with ClassInfo siena.ClassInfo used on opensource projects

Example 26 with ClassInfo

use of siena.ClassInfo in project siena by mandubian.

the class GaePersistenceManagerAsync method insert.

public SienaFuture<Integer> insert(final Iterable<?> objects) {
    List<Entity> entities = new ArrayList<Entity>();
    for (Object obj : objects) {
        Class<?> clazz = obj.getClass();
        ClassInfo info = ClassInfo.getClassInfo(clazz);
        Field idField = info.getIdField();
        Entity entity = GaeMappingUtils.createEntityInstance(idField, info, obj);
        GaeMappingUtils.fillEntity(obj, entity);
        entities.add(entity);
    }
    Future<List<Key>> future = ds.put(entities);
    Future<Integer> wrapped = new SienaFutureWrapper<List<Key>, Integer>(future) {

        @Override
        protected Integer wrap(List<Key> generatedKeys) throws Exception {
            int i = 0;
            for (Object obj : objects) {
                Class<?> clazz = obj.getClass();
                ClassInfo info = ClassInfo.getClassInfo(clazz);
                Field idField = info.getIdField();
                GaeMappingUtils.setIdFromKey(idField, obj, generatedKeys.get(i++));
            }
            return generatedKeys.size();
        }
    };
    return new SienaFutureContainer<Integer>(wrapped);
}

Example 27 with ClassInfo

use of siena.ClassInfo in project siena by mandubian.

the class DdlGenerator method addTable.

public Table addTable(Class<?> clazz) {
    if (Modifier.isAbstract(clazz.getModifiers())) {
        return null;
    }
    Table table = new Table();
    ClassInfo info = ClassInfo.getClassInfo(clazz);
    table.setName(info.tableName);
    table.setType("MyISAM");
    database.addTable(table);
    Map<String, UniqueIndex> uniques = new HashMap<String, UniqueIndex>();
    Map<String, NonUniqueIndex> indexes = new HashMap<String, NonUniqueIndex>();
    /* columns */
    for (Field field : info.allFields) {
        String[] columns = ClassInfo.getColumnNames(field);
        boolean notNull = field.getAnnotation(NotNull.class) != null;
        Class<?> type = field.getType();
        if (!ClassInfo.isModel(type) || (ClassInfo.isModel(type) && ClassInfo.isEmbedded(field))) {
            Column column = createColumn(clazz, field, columns[0]);
            if (notNull || type.isPrimitive()) {
                column.setRequired(true);
                if (type.isPrimitive() && !ClassInfo.isId(field)) {
                    // TODO: add also Boolean, Long, Double,... ?
                    if (type == Boolean.TYPE) {
                        column.setDefaultValue("false");
                    } else {
                        column.setDefaultValue("0");
                    }
                }
            }
            Id id = field.getAnnotation(Id.class);
            if (id != null) {
                column.setPrimaryKey(true);
                column.setRequired(true);
                // auto_increments managed ONLY for long
                if (id.value() == Generator.AUTO_INCREMENT && (Long.TYPE == type || Long.class.isAssignableFrom(type)))
                    column.setAutoIncrement(true);
            // adds index on primary key
            /*UniqueIndex i = uniques.get(columns[0]);
					if(i == null) {
						i = new UniqueIndex();
						i.setName(columns[0]);
						uniques.put(columns[0], i);
						table.addIndex(i);
					}
					fillIndex(i, field);*/
            }
            table.addColumn(column);
        } else {
            List<Field> keys = ClassInfo.getClassInfo(type).keys;
            for (int i = 0; i < columns.length; i++) {
                Field f = keys.get(i);
                Column column = createColumn(clazz, f, columns[i]);
                if (notNull)
                    column.setRequired(true);
                table.addColumn(column);
            }
        }
    }
    /* indexes */
    for (Field field : info.updateFields) {
        Index index = field.getAnnotation(Index.class);
        if (index != null) {
            String[] names = index.value();
            for (String name : names) {
                NonUniqueIndex i = indexes.get(name);
                if (i == null) {
                    i = new NonUniqueIndex();
                    i.setName(name);
                    indexes.put(name, i);
                    table.addIndex(i);
                }
                fillIndex(i, field);
            }
        }
        Unique unique = field.getAnnotation(Unique.class);
        if (unique != null) {
            String[] names = unique.value();
            for (String name : names) {
                UniqueIndex i = uniques.get(name);
                if (i == null) {
                    i = new UniqueIndex();
                    i.setName(name);
                    uniques.put(name, i);
                    table.addIndex(i);
                }
                fillIndex(i, field);
            }
        }
    }
    tables.put(table.getName(), table);
    return table;
}

Example 28 with ClassInfo

use of siena.ClassInfo in project siena by mandubian.

the class JdbcPersistenceManager method count.

public <T> int count(Query<T> query) {
    ClassInfo info = ClassInfo.getClassInfo(query.getQueriedClass());
    List<Object> parameters = new ArrayList<Object>();
    StringBuilder sql = new StringBuilder("SELECT COUNT(*) FROM ");
    sql.append(info.tableName);
    appendSqlWhere(query, sql, parameters);
    PreparedStatement statement = null;
    ResultSet rs = null;
    try {
        statement = createStatement(sql.toString(), parameters);
        rs = statement.executeQuery();
        rs.next();
        return rs.getInt(1);
    } catch (SQLException e) {
        throw new SienaException(e);
    } finally {
        JdbcDBUtils.closeResultSet(rs);
        JdbcDBUtils.closeStatementAndConnection(this, statement);
    }
}

Example 29 with ClassInfo

use of siena.ClassInfo in project siena by mandubian.

the class JdbcPersistenceManager method delete.

public <T> int delete(Query<T> query) {
    ClassInfo info = ClassInfo.getClassInfo(query.getQueriedClass());
    List<Object> parameters = new ArrayList<Object>();
    StringBuilder sql = new StringBuilder("DELETE FROM ");
    sql.append(info.tableName);
    appendSqlWhere(query, sql, parameters);
    PreparedStatement statement = null;
    ResultSet rs = null;
    try {
        statement = createStatement(sql.toString(), parameters);
        return statement.executeUpdate();
    } catch (SQLException e) {
        throw new SienaException(e);
    } finally {
        JdbcDBUtils.closeResultSet(rs);
        JdbcDBUtils.closeStatementAndConnection(this, statement);
    }
}

Example 30 with ClassInfo

use of siena.ClassInfo in project siena by mandubian.

the class PostgresqlPersistenceManager method appendSqlSearch.

@Override
public <T> void appendSqlSearch(QueryFilterSearch qf, Class<?> clazz, JdbcClassInfo info, StringBuilder sql, List<Object> parameters) {
    List<String> cols = new ArrayList<String>();
    try {
        for (String field : qf.fields) {
            Field f = Util.getField(clazz, field);
            Class<?> cl = f.getType();
            // if a number or date, doesn't try to coalesce
            if (Number.class.isAssignableFrom(cl) || Date.class.isAssignableFrom(cl)) {
                String[] columns = ClassInfo.getColumnNames(f, info.tableName);
                for (String col : columns) {
                    cols.add(col);
                }
            } else // if is model, gets the key type and does the same as herebefore
            if (ClassInfo.isModel(cl)) {
                ClassInfo ci = ClassInfo.getClassInfo(cl);
                if (ci.keys.size() == 1) {
                    Field key = ci.keys.get(0);
                    if (Number.class.isAssignableFrom(key.getType()) || Date.class.isAssignableFrom(key.getType())) {
                        cols.add(f.getName());
                    } else {
                        cols.add("coalesce(" + f.getName() + ", '')");
                    }
                } else {
                    for (Field key : ci.keys) {
                        String[] columns = ClassInfo.getColumnNamesWithPrefix(key, f.getName() + "_");
                        if (Number.class.isAssignableFrom(key.getType()) || Date.class.isAssignableFrom(key.getType())) {
                            for (String col : columns) {
                                cols.add(col);
                            }
                        } else {
                            for (String col : columns) {
                                cols.add("coalesce(" + col + ", '')");
                            }
                        }
                    }
                }
            } else {
                String[] columns = ClassInfo.getColumnNames(f, info.tableName);
                for (String col : columns) {
                    cols.add("coalesce(" + col + ", '')");
                }
            }
        }
        QueryOption opt = qf.option;
        if (opt != null) {
            // only manages QueryOptionJdbcSearch
            if (QueryOptionPostgresqlSearch.class.isAssignableFrom(opt.getClass())) {
                String lang = ((QueryOptionPostgresqlSearch) opt).language;
                if (lang != null && !"".equals(lang)) {
                    sql.append("to_tsvector('" + lang + "', " + Util.join(cols, " || ' ' || ") + ") @@ to_tsquery(?)");
                } else {
                    sql.append("to_tsvector('english', " + Util.join(cols, " || ' ' || ") + ") @@ to_tsquery(?)");
                }
            } else {
            }
        } else {
            sql.append("to_tsvector('english', " + Util.join(cols, " || ' ' || ") + ") @@ to_tsquery(?)");
        }
        parameters.add(qf.match);
    } catch (Exception e) {
        throw new SienaException(e);
    }
}