Example usage for org.springframework.jdbc.core JdbcTemplate execute

List of usage examples for org.springframework.jdbc.core JdbcTemplate execute

Introduction

In this page you can find the example usage for org.springframework.jdbc.core JdbcTemplate execute.

Prototype

@Override
    @Nullable
    public <T> T execute(String callString, CallableStatementCallback<T> action) throws DataAccessException 

Source Link

Usage

From source file:com.jbrisbin.vpc.jobsched.sql.SqlMessageHandler.java

public SqlMessage handleMessage(final SqlMessage msg) throws Exception {
    log.debug("handling message: " + msg.toString());

    DataSource ds = appCtx.getBean(msg.getDatasource(), DataSource.class);
    JdbcTemplate tmpl = new JdbcTemplate(ds);

    String sql = msg.getSql();//from w w  w. ja  v  a 2s  .c  o  m
    CallableStatementCreator stmtCreator = null;
    CallableStatementCallback<SqlMessage> callback = null;
    if (sql.startsWith("plugin:")) {
        // Use a plugin to get the sql
        String pluginName = (sql.contains("?") ? sql.substring(7, sql.indexOf('?')) : sql.substring(7));
        final Plugin plugin = groovyPluginManager.getPlugin(pluginName);
        Map<String, Object> vars = new LinkedHashMap<String, Object>();
        vars.put("message", msg);
        vars.put("datasource", ds);
        vars.put("listen", groovyClosureFactory.createListenClosure(msg));
        vars.put("mapreduce", groovyClosureFactory.createMapReduceClosure(msg));
        plugin.setContext(vars);

        // Execute this plugin
        plugin.run();

        Object o = plugin.get("sql");
        if (null != o && o instanceof Closure) {
            sql = ((Closure) o).call(msg).toString();
        } else if (o instanceof String || o instanceof GString) {
            sql = o.toString();
        } else {
            throw new IllegalStateException("Can't convert " + String.valueOf(o) + " to SQL statement.");
        }
        msg.setSql(sql);

        o = plugin.get("statementCreator");
        if (null != o && o instanceof Closure) {
            stmtCreator = new CallableStatementCreator() {
                public CallableStatement createCallableStatement(Connection con) throws SQLException {
                    Object obj = ((Closure) plugin.get("statementCreator")).call(new Object[] { con, msg });
                    log.debug("from plugin statementCreator: " + String.valueOf(obj));
                    return (CallableStatement) obj;
                }
            };
        } else {
            throw new IllegalStateException("Can't convert " + String.valueOf(o)
                    + " to CallableStatementCreator. Define a closure named 'statementCreator' in your plugin.");
        }

        o = plugin.get("callback");
        if (null != o && o instanceof Closure) {
            callback = new CallableStatementCallback<SqlMessage>() {
                public SqlMessage doInCallableStatement(CallableStatement cs)
                        throws SQLException, DataAccessException {
                    Object obj = ((Closure) plugin.get("callback")).call(new Object[] { cs, msg });
                    log.debug("from plugin callback: " + String.valueOf(obj));
                    return (SqlMessage) obj;
                }
            };
        } else {
            throw new IllegalStateException("Can't convert " + String.valueOf(o)
                    + " to CallableStatementCallback. Define a closure named 'callback' in your plugin.");
        }
    } else {
        stmtCreator = new CallableStatementCreator() {
            public CallableStatement createCallableStatement(Connection connection) throws SQLException {
                CallableStatement stmt = connection.prepareCall(msg.getSql());
                List<Object> params = msg.getParams();
                if (null != params) {
                    int index = 1;
                    for (Object obj : params) {
                        stmt.setObject(index++, obj);
                    }
                }
                return stmt;
            }
        };
        callback = new CallableStatementCallback<SqlMessage>() {
            public SqlMessage doInCallableStatement(CallableStatement callableStatement)
                    throws SQLException, DataAccessException {
                if (null == msg.getResults().getData()) {
                    msg.getResults().setData(new ArrayList<List<Object>>());
                }
                if (callableStatement.execute()) {
                    ResultSet results = callableStatement.getResultSet();

                    // Pull out column names
                    ResultSetMetaData meta = results.getMetaData();
                    String[] columns = new String[meta.getColumnCount()];
                    for (int i = 1; i <= meta.getColumnCount(); i++) {
                        columns[i - 1] = meta.getColumnName(i);
                    }
                    msg.getResults().getColumnNames().addAll(Arrays.asList(columns));

                    int total = 0;
                    while (results.next()) {
                        List<Object> row = new ArrayList<Object>(columns.length);
                        for (int i = 1; i <= columns.length; i++) {
                            row.add(results.getObject(i));
                        }
                        msg.getResults().getData().add(row);
                        total++;
                    }
                    msg.getResults().setTotalRows(total);

                } else {
                    msg.getResults().getColumnNames().add("updateCount");
                    msg.getResults().setTotalRows(1);
                    List<Object> updCnt = new ArrayList<Object>(1);
                    updCnt.add(callableStatement.getUpdateCount());
                    msg.getResults().getData().add(updCnt);
                }
                return msg;
            }
        };
    }
    try {
        tmpl.setExceptionTranslator(appCtx.getBean(SQLExceptionTranslator.class));
    } catch (NoSuchBeanDefinitionException notfound) {
        // IGNORED
    }

    if (null != stmtCreator && null != callback) {
        try {
            tmpl.execute(stmtCreator, callback);
        } catch (Throwable t) {
            log.error(t.getMessage(), t);
            List<String> errors = new ArrayList<String>();
            errors.add(t.getMessage());
            Throwable cause = t.getCause();
            if (null != cause) {
                do {
                    errors.add(cause.getMessage());
                } while (null != (cause = cause.getCause()));
            }
            msg.getResults().setErrors(errors);
        }
    } else {
        log.warn("CallableStatementCreator and/or CallableStatementCallback where empty. "
                + "Make sure your plugin provides these under 'statementCreator' and 'callback' respectively.");
    }
    return msg;
}

From source file:org.kuali.coeus.common.impl.krms.StoredFunctionDao.java

public String executeFunction(final String functionName, final List<Object> paramValues) {

    final JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);

    String result = jdbcTemplate.execute(new CallableStatementCreator() {
        @Override//  w w  w .  j a  v  a2 s .co m
        public CallableStatement createCallableStatement(Connection con) throws SQLException {
            String paramSyntaxString = "";
            int paramCount = paramValues.size();
            for (int i = 0; i < paramCount; i++) {
                if (i == 0)
                    paramSyntaxString += "(?";
                else if (i == paramCount - 1)
                    paramSyntaxString += ",?)";
                else
                    paramSyntaxString += ",?";
            }
            if (paramCount == 1)
                paramSyntaxString += ")";
            CallableStatement cs = con.prepareCall("{ ? = call " + functionName + paramSyntaxString + "}");
            cs.registerOutParameter(1, Types.VARCHAR);
            for (int i = 0; i < paramValues.size(); i++) {
                cs.setObject(i + 2, paramValues.get(i));
            }
            return cs;
        }
    }, new CallableStatementCallback<String>() {
        @Override
        public String doInCallableStatement(CallableStatement cs) throws SQLException, DataAccessException {
            cs.execute();
            String result = cs.getString(1);
            return result;
        }

    });
    LOG.debug(functionName + " result: " + result);
    return result;
}

From source file:org.kuali.rice.kew.documentoperation.web.DocumentContentOperationAction.java

private String getDocumentContent(final String documentId) {
    final DataSource dataSource = KEWServiceLocator.getDataSource();
    JdbcTemplate template = new JdbcTemplate(dataSource);
    String docContent = template.execute(new PreparedStatementCreator() {
        public PreparedStatement createPreparedStatement(Connection connection) throws SQLException {
            String sql = "SELECT doc_cntnt_txt FROM krew_doc_hdr_cntnt_t WHERE doc_hdr_id = ?";
            PreparedStatement statement = connection.prepareStatement(sql);
            return statement;
        }//from w  w w .j  a va2s  .c  om
    }, new PreparedStatementCallback<String>() {
        public String doInPreparedStatement(PreparedStatement statement)
                throws SQLException, DataAccessException {
            String docContent = "";
            statement.setString(1, documentId);
            ResultSet rs = statement.executeQuery();
            try {
                while (rs.next()) {
                    docContent = rs.getString("doc_cntnt_txt");
                }
            } finally {
                if (rs != null) {
                    rs.close();
                }
            }
            return docContent;
        }
    });
    return docContent;
}

From source file:org.kuali.rice.kew.documentoperation.web.DocumentContentOperationAction.java

private void saveDocumentContent(final String documentId, final String docContent) {
    if (StringUtils.isBlank(documentId) || StringUtils.isBlank(docContent)) {
        LOG.info("The document Id or the doc content was blank");
        return;/*  w ww .ja v a2  s  .  co m*/
    }
    final DataSource dataSource = KEWServiceLocator.getDataSource();
    JdbcTemplate template = new JdbcTemplate(dataSource);
    template.execute(new PreparedStatementCreator() {
        public PreparedStatement createPreparedStatement(Connection connection) throws SQLException {
            String sql = "UPDATE krew_doc_hdr_cntnt_t SET doc_cntnt_txt = ? WHERE doc_hdr_id = ?";
            PreparedStatement statement = connection.prepareStatement(sql);
            return statement;
        }
    }, new PreparedStatementCallback<String>() {
        public String doInPreparedStatement(PreparedStatement statement)
                throws SQLException, DataAccessException {
            statement.setString(1, docContent);
            statement.setString(2, documentId);
            ResultSet rs = statement.executeQuery();
            if (rs != null) {
                rs.close();
            }
            return "";
        }
    });
}

From source file:org.kuali.rice.kim.impl.role.RoleDaoJdbc.java

@Override
public List<RoleMemberBo> getRoleMembersForRoleIds(Collection<String> roleIds, String memberTypeCode,
        Map<String, String> qualification) {
    JdbcTemplate template = new JdbcTemplate(dataSource);
    final List<String> roleIDs = new ArrayList<String>(roleIds);
    final String memberTypeCd = memberTypeCode;
    final Map<String, String> qual = qualification;
    final List<RoleMemberBo> roleMemberBos = new ArrayList<RoleMemberBo>();
    template.execute(new PreparedStatementCreator() {

        /*/*from w  w w. ja v  a  2s  . c o m*/
         SAMPLE QUERY
                
        SELECT A0.ROLE_MBR_ID AS ROLE_MBR_ID,A0.ROLE_ID AS ROLE_ID,A0.MBR_ID AS MBR_ID,A0.MBR_TYP_CD AS MBR_TYP_CD,A0.VER_NBR AS ROLE_MBR_VER_NBR,A0.OBJ_ID AS ROLE_MBR_OBJ_ID,A0.ACTV_FRM_DT AS ROLE_MBR_ACTV_FRM_DT ,A0.ACTV_TO_DT AS ROLE_MBR_ACTV_TO_DT,
        BO.KIM_TYP_ID AS KIM_TYP_ID, BO.KIM_ATTR_DEFN_ID AS KIM_ATTR_DEFN_ID, BO.ATTR_VAL AS ATTR_VAL, BO.ATTR_DATA_ID AS ATTR_DATA_ID, BO.OBJ_ID AS ATTR_DATA_OBJ_ID, BO.VER_NBR AS ATTR_DATA_VER_NBR,
        CO.OBJ_ID AS ATTR_DEFN_OBJ_ID, CO.VER_NBR as ATTR_DEFN_VER_NBR, CO.NM AS ATTR_NAME, CO.LBL as ATTR_DEFN_LBL, CO.ACTV_IND as ATTR_DEFN_ACTV_IND, CO.NMSPC_CD AS ATTR_DEFN_NMSPC_CD, CO.CMPNT_NM AS ATTR_DEFN_CMPNT_NM
        FROM KRIM_ROLE_MBR_T A0 JOIN KRIM_ROLE_MBR_ATTR_DATA_T BO ON A0.ROLE_MBR_ID = BO.ROLE_MBR_ID  JOIN KRIM_ATTR_DEFN_T CO ON BO.KIM_ATTR_DEFN_ID = CO.KIM_ATTR_DEFN_ID
        WHERE A0.ROLE_ID in ('100000')
                
        UNION ALL
                
        SELECT D0.ROLE_MBR_ID AS ROLE_MBR_ID,D0.ROLE_ID AS ROLE_ID,D0.MBR_ID AS MBR_ID,D0.MBR_TYP_CD AS MBR_TYP_CD,D0.VER_NBR AS ROLE_MBR_VER_NBR,D0.OBJ_ID AS ROLE_MBR_OBJ_ID,D0.ACTV_FRM_DT AS ROLE_MBR_ACTV_FRM_DT ,D0.ACTV_TO_DT AS ROLE_MBR_ACTV_TO_DT,
        '' AS KIM_TYP_ID, '' AS KIM_ATTR_DEFN_ID, '' AS ATTR_VAL, '' AS ATTR_DATA_ID, '' AS ATTR_DATA_OBJ_ID, NULL AS ATTR_DATA_VER_NBR,
        '' AS ATTR_DEFN_OBJ_ID, NULL as ATTR_DEFN_VER_NBR, '' AS ATTR_NAME, '' as ATTR_DEFN_LBL, '' as ATTR_DEFN_ACTV_IND, '' AS ATTR_DEFN_NMSPC_CD, '' AS ATTR_DEFN_CMPNT_NM
        FROM KRIM_ROLE_MBR_T D0
        WHERE D0.ROLE_MBR_ID NOT IN (SELECT DISTINCT (E0.ROLE_MBR_ID) FROM KRIM_ROLE_MBR_ATTR_DATA_T E0)
        AND D0.ROLE_ID IN ('100000')
        */

        @Override
        public PreparedStatement createPreparedStatement(Connection connection) throws SQLException {
            /*
             The query returns multiple lines for each role by joining a role with each of its members. This allows us to get all the role member
             and role data in a single query (even though we are duplicating the role information across the role members). The cost of this
             comes out to be cheaper than firing indiviudual queries for each role in cases where there are over 500 roles
            */
            StringBuilder sql1 = new StringBuilder("SELECT "
                    + " A0.ROLE_MBR_ID AS ROLE_MBR_ID,A0.ROLE_ID AS ROLE_ID,A0.MBR_ID AS MBR_ID,A0.MBR_TYP_CD AS MBR_TYP_CD,A0.VER_NBR AS ROLE_MBR_VER_NBR,A0.OBJ_ID AS ROLE_MBR_OBJ_ID,A0.ACTV_FRM_DT AS ROLE_MBR_ACTV_FRM_DT ,A0.ACTV_TO_DT AS ROLE_MBR_ACTV_TO_DT, "
                    + " BO.KIM_TYP_ID AS KIM_TYP_ID, BO.KIM_ATTR_DEFN_ID AS KIM_ATTR_DEFN_ID, BO.ATTR_VAL AS ATTR_VAL, BO.ATTR_DATA_ID AS ATTR_DATA_ID, BO.OBJ_ID AS ATTR_DATA_OBJ_ID, BO.VER_NBR AS ATTR_DATA_VER_NBR,  "
                    + " C0.KIM_ATTR_DEFN_ID AS KIM_ATTR_DEFN_ID, C0.OBJ_ID AS ATTR_DEFN_OBJ_ID, C0.VER_NBR as ATTR_DEFN_VER_NBR, C0.NM AS ATTR_NAME, C0.LBL as ATTR_DEFN_LBL, C0.ACTV_IND as ATTR_DEFN_ACTV_IND, C0.NMSPC_CD AS ATTR_DEFN_NMSPC_CD, C0.CMPNT_NM AS ATTR_DEFN_CMPNT_NM "
                    + " FROM KRIM_ROLE_MBR_T A0 JOIN KRIM_ROLE_MBR_ATTR_DATA_T BO ON A0.ROLE_MBR_ID = BO.ROLE_MBR_ID "
                    + " JOIN KRIM_ATTR_DEFN_T C0 ON BO.KIM_ATTR_DEFN_ID = C0.KIM_ATTR_DEFN_ID  ");

            StringBuilder sql2 = new StringBuilder("SELECT"
                    + " D0.ROLE_MBR_ID AS ROLE_MBR_ID,D0.ROLE_ID AS ROLE_ID,D0.MBR_ID AS MBR_ID,D0.MBR_TYP_CD AS MBR_TYP_CD,D0.VER_NBR AS ROLE_MBR_VER_NBR,D0.OBJ_ID AS ROLE_MBR_OBJ_ID,D0.ACTV_FRM_DT AS ROLE_MBR_ACTV_FRM_DT ,D0.ACTV_TO_DT AS ROLE_MBR_ACTV_TO_DT, "
                    + " '' AS KIM_TYP_ID, '' AS KIM_ATTR_DEFN_ID, '' AS ATTR_VAL, '' AS ATTR_DATA_ID, '' AS ATTR_DATA_OBJ_ID, NULL AS ATTR_DATA_VER_NBR,"
                    + " '' AS KIM_ATTR_DEFN_ID,'' AS ATTR_DEFN_OBJ_ID, NULL as ATTR_DEFN_VER_NBR, '' AS ATTR_NAME, '' as ATTR_DEFN_LBL, '' as ATTR_DEFN_ACTV_IND, '' AS ATTR_DEFN_NMSPC_CD, '' AS ATTR_DEFN_CMPNT_NM "
                    + " FROM KRIM_ROLE_MBR_T D0 "
                    + " WHERE D0.ROLE_MBR_ID NOT IN (SELECT DISTINCT (E0.ROLE_MBR_ID) FROM KRIM_ROLE_MBR_ATTR_DATA_T E0)");

            StringBuilder criteria = new StringBuilder();

            List<String> params1 = new ArrayList<String>();
            List<String> params2 = new ArrayList<String>();

            if (roleIDs != null && !roleIDs.isEmpty()) {
                criteria.append("A0.ROLE_ID IN (");

                for (String roleId : roleIDs) {
                    criteria.append("?,");
                    params1.add(roleId);
                    params2.add(roleId);
                }
                criteria.deleteCharAt(criteria.length() - 1);
                criteria.append(")");
            }

            if (memberTypeCd != null) {
                if (criteria.length() > 0) {
                    criteria.append(" AND ");
                }

                criteria.append("A0.MBR_TYP_CD = ?");
                params1.add(memberTypeCd);
                params2.add(memberTypeCd);
            }

            // Assuming that at least a role id or role member type code is specified
            if (criteria.length() > 0) {
                sql1.append(" WHERE ");
                sql2.append(" AND ");
                sql1.append(criteria);
                sql2.append(criteria.toString().replaceAll("A0", "D0"));
            }

            if (qual != null && CollectionUtils.isNotEmpty(qual.keySet())) {

                // If Qualifiers present then sql2 should not be returning any result as it finds
                // rolemembers with now attributes
                sql2 = new StringBuilder();

                if (criteria.length() > 0) {
                    sql1.append(" AND ");
                } else {
                    sql1.append(" WHERE ");
                }

                sql1.append(" EXISTS (SELECT B1.ROLE_MBR_ID FROM KRIM_ROLE_MBR_ATTR_DATA_T B1 WHERE ");
                int conditionCount = 0;
                for (Map.Entry<String, String> qualifier : qual.entrySet()) {
                    if (StringUtils.isNotEmpty(qualifier.getValue())) {
                        // advance the number of times we have found a non-null qualifier
                        conditionCount++;

                        // add '(' if encountering a non-null qualifier for the first time
                        if (conditionCount == 1) {
                            sql1.append("(");
                        }

                        // add the qualifier template with the parameters
                        String value = (qualifier.getValue()).replace('*', '%');
                        sql1.append(" (B1.ATTR_VAL LIKE ? AND B1.KIM_ATTR_DEFN_ID = ?) ");
                        params1.add(value);
                        params1.add(qualifier.getKey());
                    }

                    sql1.append("OR");
                }
                // remove the last OR
                sql1.delete(sql1.length() - 2, sql1.length());
                // add ') AND' if we encountered a non-null qualifier sub-query above
                if (conditionCount != 0) {
                    sql1.append(") AND");
                }
                sql1.append(" B1.ROLE_MBR_ID = A0.ROLE_MBR_ID)");
            }

            StringBuilder sql = new StringBuilder(sql1.toString());

            if (sql2.length() > 0) {
                sql.append(" UNION ALL ");
                sql.append(sql2.toString());
            }

            sql.append(" ORDER BY ROLE_MBR_ID ");

            PreparedStatement statement = connection.prepareStatement(sql.toString());
            int i = 1;
            for (String param : params1) {
                statement.setString(i, param);
                i++;
            }

            if (sql2.length() > 0) {
                for (String param : params2) {
                    statement.setString(i, param);
                    i++;
                }
            }

            return statement;
        }
    }, new PreparedStatementCallback<List<RoleMemberBo>>() {
        @Override
        public List<RoleMemberBo> doInPreparedStatement(PreparedStatement statement)
                throws SQLException, DataAccessException {
            ResultSet rs = statement.executeQuery();
            try {
                RoleMemberBo lastRoleMember = null;
                while (rs.next()) {
                    boolean processRolemember = true;

                    String roleId = rs.getString("ROLE_ID");
                    String id = rs.getString("ROLE_MBR_ID");
                    String memberId = rs.getString("MBR_ID");

                    MemberType memberType = MemberType.fromCode(rs.getString("MBR_TYP_CD"));
                    DateTime activeFromDate = rs.getDate("ROLE_MBR_ACTV_FRM_DT") == null ? null
                            : new DateTime(rs.getDate("ROLE_MBR_ACTV_FRM_DT"));
                    DateTime activeToDate = rs.getDate("ROLE_MBR_ACTV_TO_DT") == null ? null
                            : new DateTime(rs.getDate("ROLE_MBR_ACTV_TO_DT"));

                    // Since we are joining role members and attributes we would have multiple role member rows
                    // but one row per attribute so check if its the first time we are seeing the role member
                    if (lastRoleMember == null || !id.equals(lastRoleMember.getId())) {
                        RoleMember roleMember = RoleMember.Builder.create(roleId, id, memberId, memberType,
                                activeFromDate, activeToDate, new HashMap<String, String>(), "", "").build();
                        Long roleVersionNbr = rs.getLong("ROLE_MBR_VER_NBR");
                        String roleObjId = rs.getString("ROLE_MBR_OBJ_ID");

                        RoleMemberBo roleMemberBo = RoleMemberBo.from(roleMember);
                        roleMemberBo.setVersionNumber(roleVersionNbr);
                        roleMemberBo.setObjectId(roleObjId);
                        List<RoleMemberAttributeDataBo> roleMemAttrBos = new ArrayList<RoleMemberAttributeDataBo>();

                        roleMemberBo.setAttributeDetails(roleMemAttrBos);
                        if (roleMemberBo.isActive(new Timestamp(System.currentTimeMillis()))) {
                            roleMemberBos.add(roleMemberBo);
                        } else {
                            processRolemember = false;
                        }

                        lastRoleMember = roleMemberBo;
                    }

                    String kimTypeId = rs.getString("KIM_TYP_ID");
                    if (processRolemember && StringUtils.isNotEmpty(kimTypeId)) {
                        KimType theType = KimApiServiceLocator.getKimTypeInfoService().getKimType(kimTypeId);
                        // Create RoleMemberAttributeDataBo for this row
                        RoleMemberAttributeDataBo roleMemAttrDataBo = new RoleMemberAttributeDataBo();

                        KimAttribute.Builder attrBuilder = KimAttribute.Builder.create(
                                rs.getString("ATTR_DEFN_CMPNT_NM"), rs.getString("ATTR_NAME"),
                                rs.getString("ATTR_DEFN_NMSPC_CD"));
                        attrBuilder.setActive(Truth.strToBooleanIgnoreCase(rs.getString("ATTR_DEFN_ACTV_IND")));
                        attrBuilder.setAttributeLabel(rs.getString("ATTR_DEFN_LBL"));
                        attrBuilder.setId(rs.getString("KIM_ATTR_DEFN_ID"));
                        attrBuilder.setObjectId(rs.getString("ATTR_DEFN_OBJ_ID"));
                        attrBuilder.setVersionNumber(rs.getLong("ATTR_DEFN_VER_NBR"));

                        roleMemAttrDataBo.setId(rs.getString("ATTR_DATA_ID"));
                        roleMemAttrDataBo.setAssignedToId(id);
                        roleMemAttrDataBo.setKimTypeId(kimTypeId);
                        roleMemAttrDataBo.setKimType(KimTypeBo.from(theType));
                        roleMemAttrDataBo.setKimAttributeId(attrBuilder.getId());
                        roleMemAttrDataBo.setAttributeValue(rs.getString("ATTR_VAL"));
                        roleMemAttrDataBo.setVersionNumber(attrBuilder.getVersionNumber());
                        roleMemAttrDataBo.setObjectId(attrBuilder.getObjectId());

                        roleMemAttrDataBo.setKimAttribute(KimAttributeBo.from(attrBuilder.build()));
                        lastRoleMember.getAttributeDetails().add(roleMemAttrDataBo);
                    }

                }
            } finally {
                if (rs != null) {
                    rs.close();
                }
            }
            return roleMemberBos;
        }
    });
    return roleMemberBos;
}

From source file:org.kuali.rice.kim.impl.role.RoleDaoOjb.java

public List<RoleMemberBo> getRoleMembersForRoleIds(Collection<String> roleIds, String memberTypeCode,
        Map<String, String> qualification) {
    JdbcTemplate template = new JdbcTemplate(dataSource);
    final List<String> roleIDs = new ArrayList<String>(roleIds);
    final String memberTypeCd = memberTypeCode;
    final Map<String, String> qual = qualification;
    final List<RoleMemberBo> roleMemberBos = new ArrayList<RoleMemberBo>();
    List<RoleMemberBo> results = template.execute(new PreparedStatementCreator() {

        /*/*from  w  w w  . ja v a2  s  .c  om*/
         SAMPLE QUERY
                
        SELECT A0.ROLE_MBR_ID AS ROLE_MBR_ID,A0.ROLE_ID AS ROLE_ID,A0.MBR_ID AS MBR_ID,A0.MBR_TYP_CD AS MBR_TYP_CD,A0.VER_NBR AS ROLE_MBR_VER_NBR,A0.OBJ_ID AS ROLE_MBR_OBJ_ID,A0.ACTV_FRM_DT AS ROLE_MBR_ACTV_FRM_DT ,A0.ACTV_TO_DT AS ROLE_MBR_ACTV_TO_DT,
        BO.KIM_TYP_ID AS KIM_TYP_ID, BO.KIM_ATTR_DEFN_ID AS KIM_ATTR_DEFN_ID, BO.ATTR_VAL AS ATTR_VAL, BO.ATTR_DATA_ID AS ATTR_DATA_ID, BO.OBJ_ID AS ATTR_DATA_OBJ_ID, BO.VER_NBR AS ATTR_DATA_VER_NBR,
        CO.OBJ_ID AS ATTR_DEFN_OBJ_ID, CO.VER_NBR as ATTR_DEFN_VER_NBR, CO.NM AS ATTR_NAME, CO.LBL as ATTR_DEFN_LBL, CO.ACTV_IND as ATTR_DEFN_ACTV_IND, CO.NMSPC_CD AS ATTR_DEFN_NMSPC_CD, CO.CMPNT_NM AS ATTR_DEFN_CMPNT_NM
        FROM KRIM_ROLE_MBR_T A0 JOIN KRIM_ROLE_MBR_ATTR_DATA_T BO ON A0.ROLE_MBR_ID = BO.ROLE_MBR_ID  JOIN KRIM_ATTR_DEFN_T CO ON BO.KIM_ATTR_DEFN_ID = CO.KIM_ATTR_DEFN_ID
        WHERE A0.ROLE_ID in ('100000')
                
        UNION ALL
                
        SELECT D0.ROLE_MBR_ID AS ROLE_MBR_ID,D0.ROLE_ID AS ROLE_ID,D0.MBR_ID AS MBR_ID,D0.MBR_TYP_CD AS MBR_TYP_CD,D0.VER_NBR AS ROLE_MBR_VER_NBR,D0.OBJ_ID AS ROLE_MBR_OBJ_ID,D0.ACTV_FRM_DT AS ROLE_MBR_ACTV_FRM_DT ,D0.ACTV_TO_DT AS ROLE_MBR_ACTV_TO_DT,
        '' AS KIM_TYP_ID, '' AS KIM_ATTR_DEFN_ID, '' AS ATTR_VAL, '' AS ATTR_DATA_ID, '' AS ATTR_DATA_OBJ_ID, NULL AS ATTR_DATA_VER_NBR,
        '' AS ATTR_DEFN_OBJ_ID, NULL as ATTR_DEFN_VER_NBR, '' AS ATTR_NAME, '' as ATTR_DEFN_LBL, '' as ATTR_DEFN_ACTV_IND, '' AS ATTR_DEFN_NMSPC_CD, '' AS ATTR_DEFN_CMPNT_NM
        FROM KRIM_ROLE_MBR_T D0
        WHERE D0.ROLE_MBR_ID NOT IN (SELECT DISTINCT (E0.ROLE_MBR_ID) FROM KRIM_ROLE_MBR_ATTR_DATA_T E0)
        AND D0.ROLE_ID IN ('100000')
        */

        public PreparedStatement createPreparedStatement(Connection connection) throws SQLException {
            /*
             The query returns multiple lines for each role by joining a role with each of its members. This allows us to get all the role member
             and role data in a single query (even though we are duplicating the role information across the role members). The cost of this
             comes out to be cheaper than firing indiviudual queries for each role in cases where there are over 500 roles
            */
            StringBuffer sql1 = new StringBuffer("SELECT "
                    + " A0.ROLE_MBR_ID AS ROLE_MBR_ID,A0.ROLE_ID AS ROLE_ID,A0.MBR_ID AS MBR_ID,A0.MBR_TYP_CD AS MBR_TYP_CD,A0.VER_NBR AS ROLE_MBR_VER_NBR,A0.OBJ_ID AS ROLE_MBR_OBJ_ID,A0.ACTV_FRM_DT AS ROLE_MBR_ACTV_FRM_DT ,A0.ACTV_TO_DT AS ROLE_MBR_ACTV_TO_DT, "
                    + " BO.KIM_TYP_ID AS KIM_TYP_ID, BO.KIM_ATTR_DEFN_ID AS KIM_ATTR_DEFN_ID, BO.ATTR_VAL AS ATTR_VAL, BO.ATTR_DATA_ID AS ATTR_DATA_ID, BO.OBJ_ID AS ATTR_DATA_OBJ_ID, BO.VER_NBR AS ATTR_DATA_VER_NBR,  "
                    + " C0.KIM_ATTR_DEFN_ID AS KIM_ATTR_DEFN_ID, C0.OBJ_ID AS ATTR_DEFN_OBJ_ID, C0.VER_NBR as ATTR_DEFN_VER_NBR, C0.NM AS ATTR_NAME, C0.LBL as ATTR_DEFN_LBL, C0.ACTV_IND as ATTR_DEFN_ACTV_IND, C0.NMSPC_CD AS ATTR_DEFN_NMSPC_CD, C0.CMPNT_NM AS ATTR_DEFN_CMPNT_NM "
                    + " FROM KRIM_ROLE_MBR_T A0 JOIN KRIM_ROLE_MBR_ATTR_DATA_T BO ON A0.ROLE_MBR_ID = BO.ROLE_MBR_ID "
                    + " JOIN KRIM_ATTR_DEFN_T C0 ON BO.KIM_ATTR_DEFN_ID = C0.KIM_ATTR_DEFN_ID  ");

            StringBuffer sql2 = new StringBuffer("SELECT"
                    + " D0.ROLE_MBR_ID AS ROLE_MBR_ID,D0.ROLE_ID AS ROLE_ID,D0.MBR_ID AS MBR_ID,D0.MBR_TYP_CD AS MBR_TYP_CD,D0.VER_NBR AS ROLE_MBR_VER_NBR,D0.OBJ_ID AS ROLE_MBR_OBJ_ID,D0.ACTV_FRM_DT AS ROLE_MBR_ACTV_FRM_DT ,D0.ACTV_TO_DT AS ROLE_MBR_ACTV_TO_DT, "
                    + " '' AS KIM_TYP_ID, '' AS KIM_ATTR_DEFN_ID, '' AS ATTR_VAL, '' AS ATTR_DATA_ID, '' AS ATTR_DATA_OBJ_ID, NULL AS ATTR_DATA_VER_NBR,"
                    + " '' AS KIM_ATTR_DEFN_ID,'' AS ATTR_DEFN_OBJ_ID, NULL as ATTR_DEFN_VER_NBR, '' AS ATTR_NAME, '' as ATTR_DEFN_LBL, '' as ATTR_DEFN_ACTV_IND, '' AS ATTR_DEFN_NMSPC_CD, '' AS ATTR_DEFN_CMPNT_NM "
                    + " FROM KRIM_ROLE_MBR_T D0 "
                    + " WHERE D0.ROLE_MBR_ID NOT IN (SELECT DISTINCT (E0.ROLE_MBR_ID) FROM KRIM_ROLE_MBR_ATTR_DATA_T E0)");

            StringBuffer criteria = new StringBuffer();

            List<String> params1 = new ArrayList<String>();
            List<String> params2 = new ArrayList<String>();

            if (roleIDs != null && !roleIDs.isEmpty()) {
                criteria.append("A0.ROLE_ID IN (");

                for (String roleId : roleIDs) {
                    criteria.append("?,");
                    params1.add(roleId);
                    params2.add(roleId);
                }
                criteria.deleteCharAt(criteria.length() - 1);
                criteria.append(")");
            }

            if (memberTypeCd != null) {
                if (criteria.length() > 0) {
                    criteria.append(" AND ");
                }

                criteria.append("A0.MBR_TYP_CD = ?");
                params1.add(memberTypeCd);
                params2.add(memberTypeCd);
            }

            // Assuming that at least a role id or role member type code is specified
            if (criteria.length() > 0) {
                sql1.append(" WHERE ");
                sql2.append(" AND ");
                sql1.append(criteria);
                sql2.append(criteria.toString().replaceAll("A0", "D0"));
            }

            if (qual != null && CollectionUtils.isNotEmpty(qual.keySet())) {

                // If Qualifiers present then sql2 should not be returning any result as it finds
                // rolemembers with now attributes
                sql2 = new StringBuffer();

                if (criteria.length() > 0) {
                    sql1.append(" AND ");
                } else {
                    sql1.append(" WHERE ");
                }

                sql1.append(" EXISTS (SELECT B1.ROLE_MBR_ID FROM KRIM_ROLE_MBR_ATTR_DATA_T B1 WHERE (");
                for (Map.Entry<String, String> qualifier : qual.entrySet()) {
                    if (StringUtils.isNotEmpty(qualifier.getValue())) {
                        String value = (qualifier.getValue()).replace('*', '%');
                        sql1.append(" (B1.ATTR_VAL LIKE ? AND B1.KIM_ATTR_DEFN_ID = ? ) ");
                        params1.add(value);
                        params1.add(qualifier.getKey());
                    }
                    sql1.append("OR");
                }
                sql1.delete(sql1.length() - 2, sql1.length());
                sql1.append(") AND B1.ROLE_MBR_ID = A0.ROLE_MBR_ID )");

            }

            StringBuffer sql = new StringBuffer(sql1.toString());

            if (sql2.length() > 0) {
                sql.append(" UNION ALL ");
                sql.append(sql2.toString());
            }

            sql.append(" ORDER BY ROLE_MBR_ID ");

            PreparedStatement statement = connection.prepareStatement(sql.toString());
            int i = 1;
            for (String param : params1) {
                statement.setString(i, param);
                i++;
            }

            if (sql2.length() > 0) {
                for (String param : params2) {
                    statement.setString(i, param);
                    i++;
                }
            }

            return statement;
        }
    }, new PreparedStatementCallback<List<RoleMemberBo>>() {
        public List<RoleMemberBo> doInPreparedStatement(PreparedStatement statement)
                throws SQLException, DataAccessException {
            ResultSet rs = statement.executeQuery();
            try {
                RoleMemberBo lastRoleMember = null;
                while (rs.next()) {
                    boolean processRolemember = true;

                    String roleId = rs.getString("ROLE_ID");
                    String id = rs.getString("ROLE_MBR_ID");
                    String memberId = rs.getString("MBR_ID");

                    MemberType memberType = MemberType.fromCode(rs.getString("MBR_TYP_CD"));
                    DateTime activeFromDate = rs.getDate("ROLE_MBR_ACTV_FRM_DT") == null ? null
                            : new DateTime(rs.getDate("ROLE_MBR_ACTV_FRM_DT"));
                    DateTime activeToDate = rs.getDate("ROLE_MBR_ACTV_TO_DT") == null ? null
                            : new DateTime(rs.getDate("ROLE_MBR_ACTV_TO_DT"));

                    // Since we are joining role members and attributes we would have multiple role member rows
                    // but one row per attribute so check if its the first time we are seeing the role member
                    if (lastRoleMember == null || !id.equals(lastRoleMember.getId())) {
                        RoleMember roleMember = RoleMember.Builder.create(roleId, id, memberId, memberType,
                                activeFromDate, activeToDate, new HashMap<String, String>(), "", "").build();
                        Long roleVersionNbr = rs.getLong("ROLE_MBR_VER_NBR");
                        String roleObjId = rs.getString("ROLE_MBR_OBJ_ID");

                        RoleMemberBo roleMemberBo = RoleMemberBo.from(roleMember);
                        roleMemberBo.setVersionNumber(roleVersionNbr);
                        roleMemberBo.setObjectId(roleObjId);
                        List<RoleMemberAttributeDataBo> roleMemAttrBos = new ArrayList<RoleMemberAttributeDataBo>();

                        roleMemberBo.setAttributeDetails(roleMemAttrBos);
                        if (roleMemberBo.isActive(new Timestamp(System.currentTimeMillis()))) {
                            roleMemberBos.add(roleMemberBo);
                        } else {
                            processRolemember = false;
                        }

                        lastRoleMember = roleMemberBo;
                    }

                    String kimTypeId = rs.getString("KIM_TYP_ID");
                    String attrKey = rs.getString("KIM_ATTR_DEFN_ID");
                    String attrVal = rs.getString("ATTR_VAL");
                    if (processRolemember && StringUtils.isNotEmpty(kimTypeId)) {
                        KimType theType = KimApiServiceLocator.getKimTypeInfoService().getKimType(kimTypeId);
                        // Create RoleMemberAttributeDataBo for this row
                        RoleMemberAttributeDataBo roleMemAttrDataBo = new RoleMemberAttributeDataBo();

                        KimAttribute.Builder attrBuilder = KimAttribute.Builder.create(
                                rs.getString("ATTR_DEFN_CMPNT_NM"), rs.getString("ATTR_NAME"),
                                rs.getString("ATTR_DEFN_NMSPC_CD"));
                        attrBuilder.setActive(Truth.strToBooleanIgnoreCase(rs.getString("ATTR_DEFN_ACTV_IND")));
                        attrBuilder.setAttributeLabel(rs.getString("ATTR_DEFN_LBL"));
                        attrBuilder.setId(rs.getString("KIM_ATTR_DEFN_ID"));
                        attrBuilder.setObjectId(rs.getString("ATTR_DEFN_OBJ_ID"));
                        attrBuilder.setVersionNumber(rs.getLong("ATTR_DEFN_VER_NBR"));

                        roleMemAttrDataBo.setId(rs.getString("ATTR_DATA_ID"));
                        roleMemAttrDataBo.setAssignedToId(id);
                        roleMemAttrDataBo.setKimTypeId(kimTypeId);
                        roleMemAttrDataBo.setKimType(KimTypeBo.from(theType));
                        roleMemAttrDataBo.setKimAttributeId(attrBuilder.getId());
                        roleMemAttrDataBo.setAttributeValue(attrVal);
                        roleMemAttrDataBo.setVersionNumber(rs.getLong("ATTR_DATA_VER_NBR"));
                        roleMemAttrDataBo.setObjectId(rs.getString("ATTR_DATA_OBJ_ID"));

                        roleMemAttrDataBo.setKimAttribute(KimAttributeBo.from(attrBuilder.build()));
                        lastRoleMember.getAttributeDetails().add(roleMemAttrDataBo);
                    }

                }
            } finally {
                if (rs != null) {
                    rs.close();
                }
            }
            return roleMemberBos;
        }
    });
    return roleMemberBos;
}