play.db.DB Java Examples

The following examples show how to use play.db.DB. You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example #1
Source File: DBTestUtil.java    From dr-elephant with Apache License 2.0 6 votes vote down vote up 
public static void initDBUtil(String fileName) throws IOException, SQLException {
  String query = "";
  FileInputStream inputStream = new FileInputStream(fileName);

  try {
    query = IOUtils.toString(inputStream);
  } finally {
    inputStream.close();
  }

  Connection connection = DB.getConnection();

  try {
    Statement statement = connection.createStatement();
    statement.execute(query);
  } finally {
    connection.close();
  }
}
Example #2
Source File: SQLBasedJoinExpander.java    From samantha with MIT License 5 votes vote down vote up 
public static EntityExpander getExpander(Configuration expanderConfig,
                                         Injector injector,
                                         RequestContext requestContext) {
    Settings settings = new Settings()
            .withStatementType(StatementType.STATIC_STATEMENT);
    DSLContext create = DSL.using(DB.getDataSource(expanderConfig.getString("db")), SQLDialect.DEFAULT, settings);
    return new SQLBasedJoinExpander(expanderConfig.getConfigList("expandFields"), create);
}
Example #3
Source File: SQLBasedRetrieverConfig.java    From samantha with MIT License 5 votes vote down vote up 
public Retriever getRetriever(RequestContext requestContext) {
    Settings settings = new Settings()
            .withStatementType(StatementType.STATIC_STATEMENT);
    DSLContext create = DSL.using(DB.getDataSource(db), SQLDialect.DEFAULT, settings);
    return new SQLBasedRetriever(config, setCursorKey,
            limit, offset, selectSqlKey, matchFields,
            greaterFields, lessFields, matchFieldTypes, greaterFieldTypes, lessFieldTypes,
            create, selectFields, table, orderByFields, renameMap, requestContext, injector);
}
Example #4
Source File: SQLBasedIndexerConfig.java    From samantha with MIT License 5 votes vote down vote up 
public Indexer getIndexer(RequestContext requestContext) {
    SamanthaConfigService configService = injector.instanceOf(SamanthaConfigService.class);
    Settings settings = new Settings()
            .withStatementType(StatementType.STATIC_STATEMENT);
    DSLContext create = DSL.using(DB.getDataSource(db), SQLDialect.DEFAULT, settings);
    return new SQLBasedIndexer(configService, daoConfigs,
            create, tableKey, table, injector, daoConfigKey,
            fields, fieldTypes, matchFields, matchFieldTypes,
            retrieverName, setCursorKey, daoNameKey, daoName,
            cacheJsonFile, filePathKey, separatorKey,
            config, 128, requestContext);
}
Example #5
Source File: JdbcHelper.java    From restcommander with Apache License 2.0 5 votes vote down vote up 
public static ResultSet execute(String sql, Object ... params) throws SQLException {
    PreparedStatement pst = DB.getConnection().prepareStatement(sql, ResultSet.TYPE_SCROLL_SENSITIVE, ResultSet.CONCUR_UPDATABLE);
    int index = 0;
    for (Object param : params) {
        pst.setObject(++index, param);
    }
    return pst.executeQuery();
}
Example #6
Source File: JdbcHelper.java    From restcommander with Apache License 2.0 5 votes vote down vote up 
public static ResultSet executeList(String sql, List<Object> params) throws SQLException {
    PreparedStatement pst = DB.getConnection().prepareStatement(sql, ResultSet.TYPE_SCROLL_SENSITIVE, ResultSet.CONCUR_UPDATABLE);
    int index = 0;
    for (Object param : params) {
        pst.setObject(++index, param);
    }
    return pst.executeQuery();
}
Example #7
Source File: Fixtures.java    From restcommander with Apache License 2.0 5 votes vote down vote up 
public static void executeSQL(String sqlScript) {
    for(String sql: sqlScript.split(";")) {
        if(sql.trim().length() > 0) {
            DB.execute(sql);
        }
    }
}
Example #8
Source File: Fixtures.java    From restcommander with Apache License 2.0 4 votes vote down vote up 
private static void disableForeignKeyConstraints() {
    if (DBPlugin.url.startsWith("jdbc:oracle:")) {
        DB.execute("begin\n"
                + "for i in (select constraint_name, table_name from user_constraints where constraint_type ='R'\n"
                + "and status = 'ENABLED') LOOP\n"
                + "execute immediate 'alter table '||i.table_name||' disable constraint '||i.constraint_name||'';\n"
                + "end loop;\n"
                + "end;"
        );
        return;
    }

    if (DBPlugin.url.startsWith("jdbc:hsqldb:")) {
        DB.execute("SET REFERENTIAL_INTEGRITY FALSE");
        return;
    }

    if (DBPlugin.url.startsWith("jdbc:h2:")) {
        DB.execute("SET REFERENTIAL_INTEGRITY FALSE");
        return;
    }

    if (DBPlugin.url.startsWith("jdbc:mysql:")) {
        DB.execute("SET foreign_key_checks = 0;");
        return;
    }

    if (DBPlugin.url.startsWith("jdbc:postgresql:")) {
        DB.execute("SET CONSTRAINTS ALL DEFERRED");
        return;
    }

    if (DBPlugin.url.startsWith("jdbc:sqlserver:")) {
        Statement exec=null;

        try {
            List<String> names = new ArrayList<String>();
            Connection connection = DB.getConnection();

            ResultSet rs = connection.getMetaData().getTables(null, null, null, new String[]{"TABLE"});
            while (rs.next()) {
                String name = rs.getString("TABLE_NAME");
                names.add(name);
            }

                // Then we disable all foreign keys
            exec = connection.createStatement();
            for (String tableName:names)
                exec.addBatch("ALTER TABLE " + tableName+" NOCHECK CONSTRAINT ALL");
            exec.executeBatch();
            exec.close();

            return;
        } catch (SQLException ex) {
            throw new DatabaseException("Error while disabling foreign keys", ex);
        }
    }

    // Maybe Log a WARN for unsupported DB ?
    Logger.warn("Fixtures : unable to disable constraints, unsupported database : " + DBPlugin.url);
}
Example #9
Source File: Fixtures.java    From restcommander with Apache License 2.0 4 votes vote down vote up 
private static void enableForeignKeyConstraints() {
    if (DBPlugin.url.startsWith("jdbc:oracle:")) {
        DB.execute("begin\n"
                + "for i in (select constraint_name, table_name from user_constraints where constraint_type ='R'\n"
                + "and status = 'DISABLED') LOOP\n"
                + "execute immediate 'alter table '||i.table_name||' enable constraint '||i.constraint_name||'';\n"
                + "end loop;\n"
                + "end;"
        );
        return;
    }

    if (DBPlugin.url.startsWith("jdbc:hsqldb:")) {
        DB.execute("SET REFERENTIAL_INTEGRITY TRUE");
        return;
    }

    if (DBPlugin.url.startsWith("jdbc:h2:")) {
        DB.execute("SET REFERENTIAL_INTEGRITY TRUE");
        return;
    }

    if (DBPlugin.url.startsWith("jdbc:mysql:")) {
        DB.execute("SET foreign_key_checks = 1;");
        return;
    }

    if (DBPlugin.url.startsWith("jdbc:postgresql:")) {
        return;
    }

    if (DBPlugin.url.startsWith("jdbc:sqlserver:")) {
       Connection connect = null;
        Statement exec=null;
        try {
            connect = DB.getConnection();
            // We must first drop all foreign keys
            ArrayList<String> checkFKCommands=new ArrayList<String>();
            exec=connect.createStatement();
            ResultSet rs=exec.executeQuery("SELECT 'ALTER TABLE ' + TABLE_SCHEMA + '.[' + TABLE_NAME +'] WITH CHECK CHECK CONSTRAINT [' + CONSTRAINT_NAME + ']' FROM INFORMATION_SCHEMA.TABLE_CONSTRAINTS WHERE CONSTRAINT_TYPE = 'FOREIGN KEY'");
            while (rs.next())
            {
                checkFKCommands.add(rs.getString(1));
            }
            exec.close();
            exec=null;

             // Now we have the drop commands, let's execute them
            exec=connect.createStatement();
            for (String sql:checkFKCommands)
                exec.addBatch(sql);
            exec.executeBatch();
            exec.close();
        } catch (SQLException ex) {
            throw new DatabaseException("Cannot enable foreign keys", ex);
        }
        return;
      }

    Logger.warn("Fixtures : unable to enable constraints, unsupported database : " + DBPlugin.url);
}