List of usage examples for org.springframework.jdbc.core JdbcTemplate getDataSource
@Nullable
public DataSource getDataSource()
From source file:dao.AdvSearchDAO.java
public static ObjectNode search(JsonNode searchOpt, int page, int size) { ObjectNode resultNode = Json.newObject(); int count = 0; List<String> scopeInList = new ArrayList<String>(); List<String> scopeNotInList = new ArrayList<String>(); List<String> tableInList = new ArrayList<String>(); List<String> tableNotInList = new ArrayList<String>(); List<String> fieldAnyList = new ArrayList<String>(); List<String> fieldAllList = new ArrayList<String>(); List<String> fieldNotInList = new ArrayList<String>(); String fieldAllIDs = ""; String comments = ""; if (searchOpt != null && (searchOpt.isContainerNode())) { if (searchOpt.has("scope")) { JsonNode scopeNode = searchOpt.get("scope"); if (scopeNode != null && scopeNode.isContainerNode()) { if (scopeNode.has("in")) { JsonNode scopeInNode = scopeNode.get("in"); if (scopeInNode != null) { String scopeInStr = scopeInNode.asText(); if (StringUtils.isNotBlank(scopeInStr)) { String[] scopeInArray = scopeInStr.split(","); if (scopeInArray != null) { for (String value : scopeInArray) { if (StringUtils.isNotBlank(value)) { scopeInList.add(value.trim()); }/* www . j a v a2 s.c o m*/ } } } } } if (scopeNode.has("not")) { JsonNode scopeNotInNode = scopeNode.get("not"); if (scopeNotInNode != null) { String scopeNotInStr = scopeNotInNode.asText(); if (StringUtils.isNotBlank(scopeNotInStr)) { String[] scopeNotInArray = scopeNotInStr.split(","); if (scopeNotInArray != null) { for (String value : scopeNotInArray) { if (StringUtils.isNotBlank(value)) { scopeNotInList.add(value.trim()); } } } } } } } } if (searchOpt.has("table")) { JsonNode tableNode = searchOpt.get("table"); if (tableNode != null && tableNode.isContainerNode()) { if (tableNode.has("in")) { JsonNode tableInNode = tableNode.get("in"); if (tableInNode != null) { String tableInStr = tableInNode.asText(); if (StringUtils.isNotBlank(tableInStr)) { String[] tableInArray = tableInStr.split(","); if (tableInArray != null) { for (String value : tableInArray) { if (StringUtils.isNotBlank(value)) { tableInList.add(value.trim()); } } } } } } if (tableNode.has("not")) { JsonNode tableNotInNode = tableNode.get("not"); if (tableNotInNode != null) { String tableNotInStr = tableNotInNode.asText(); if (StringUtils.isNotBlank(tableNotInStr)) { String[] tableNotInArray = tableNotInStr.split(","); if (tableNotInArray != null) { for (String value : tableNotInArray) { if (StringUtils.isNotBlank(value)) { tableNotInList.add(value.trim()); } } } } } } } } if (searchOpt.has("fields")) { JsonNode fieldNode = searchOpt.get("fields"); if (fieldNode != null && fieldNode.isContainerNode()) { if (fieldNode.has("any")) { JsonNode fieldAnyNode = fieldNode.get("any"); if (fieldAnyNode != null) { String fieldAnyStr = fieldAnyNode.asText(); if (StringUtils.isNotBlank(fieldAnyStr)) { String[] fieldAnyArray = fieldAnyStr.split(","); if (fieldAnyArray != null) { for (String value : fieldAnyArray) { if (StringUtils.isNotBlank(value)) { fieldAnyList.add(value.trim()); } } } } } } if (fieldNode.has("all")) { JsonNode fieldAllNode = fieldNode.get("all"); if (fieldAllNode != null) { String fieldAllStr = fieldAllNode.asText(); if (StringUtils.isNotBlank(fieldAllStr)) { String[] fieldAllArray = fieldAllStr.split(","); if (fieldAllArray != null) { for (String value : fieldAllArray) { if (StringUtils.isNotBlank(value)) { fieldAllList.add(value.trim()); } } } } } } if (fieldNode.has("not")) { JsonNode fieldNotInNode = fieldNode.get("not"); if (fieldNotInNode != null) { String fieldNotInStr = fieldNotInNode.asText(); if (StringUtils.isNotBlank(fieldNotInStr)) { String[] fieldNotInArray = fieldNotInStr.split(","); if (fieldNotInArray != null) { for (String value : fieldNotInArray) { if (StringUtils.isNotBlank(value)) { fieldNotInList.add(value.trim()); } } } } } } } } String datasetSources = ""; if (searchOpt.has("sources")) { JsonNode sourcesNode = searchOpt.get("sources"); if (sourcesNode != null) { datasetSources = sourcesNode.asText(); } } boolean needAndKeyword = false; int fieldQueryIndex = 0; if (fieldAllList.size() > 0) { String fieldAllQuery = "SELECT DISTINCT f1.dataset_id FROM dict_field_detail f1 "; String fieldWhereClause = " WHERE "; for (String field : fieldAllList) { fieldQueryIndex++; if (fieldQueryIndex == 1) { fieldWhereClause += "f1.field_name LIKE '%" + field + "%' "; } else { fieldAllQuery += "JOIN dict_field_detail f" + fieldQueryIndex + " ON f" + (fieldQueryIndex - 1) + ".dataset_id = f" + fieldQueryIndex + ".dataset_id "; fieldWhereClause += " and f" + fieldQueryIndex + ".field_name LIKE '%" + field + "%' "; } } fieldAllQuery += fieldWhereClause; List<Map<String, Object>> rows = getJdbcTemplate().queryForList(fieldAllQuery); for (Map row : rows) { fieldAllIDs += (Long) row.get("dataset_id") + ","; } if (fieldAllIDs.length() > 0) { fieldAllIDs = fieldAllIDs.substring(0, fieldAllIDs.length() - 1); } if (StringUtils.isBlank(fieldAllIDs)) { fieldAllIDs = Integer.toString(0); } } List<Dataset> pagedDatasets = new ArrayList<Dataset>(); final JdbcTemplate jdbcTemplate = getJdbcTemplate(); javax.sql.DataSource ds = jdbcTemplate.getDataSource(); DataSourceTransactionManager tm = new DataSourceTransactionManager(ds); TransactionTemplate txTemplate = new TransactionTemplate(tm); ObjectNode result; if (searchOpt.has("comments")) { JsonNode commentsNode = searchOpt.get("comments"); if (commentsNode != null) { comments = commentsNode.asText(); if (StringUtils.isNotBlank(comments)) { if (scopeInList.size() == 0 && scopeNotInList.size() == 0 && tableInList.size() == 0 && tableNotInList.size() == 0 && fieldAllList.size() == 0 && fieldAnyList.size() == 0 && fieldNotInList.size() == 0) { final String commentsQueryStr = SEARCH_DATASETS_BY_COMMENTS_WITH_PAGINATION .replace("$keyword", comments); result = txTemplate.execute(new TransactionCallback<ObjectNode>() { public ObjectNode doInTransaction(TransactionStatus status) { List<Map<String, Object>> rows = null; rows = jdbcTemplate.queryForList(commentsQueryStr, (page - 1) * size, size); for (Map row : rows) { Dataset ds = new Dataset(); ds.id = (Long) row.get("id"); ds.name = (String) row.get("name"); ds.source = (String) row.get("source"); ds.urn = (String) row.get("urn"); ds.schema = (String) row.get("schema"); pagedDatasets.add(ds); } long count = 0; try { count = jdbcTemplate.queryForObject("SELECT FOUND_ROWS()", Long.class); } catch (EmptyResultDataAccessException e) { Logger.error("Exception = " + e.getMessage()); } ObjectNode resultNode = Json.newObject(); resultNode.put("count", count); resultNode.put("page", page); resultNode.put("itemsPerPage", size); resultNode.put("totalPages", (int) Math.ceil(count / ((double) size))); resultNode.set("data", Json.toJson(pagedDatasets)); return resultNode; } }); return result; } } } } String query = ""; if (StringUtils.isNotBlank(comments)) { query = "SELECT DISTINCT d.id FROM dict_dataset d"; } else { query = "SELECT SQL_CALC_FOUND_ROWS " + "DISTINCT d.id, d.name, d.schema, d.source, d.urn, " + "FROM_UNIXTIME(d.source_modified_time) as modified FROM dict_dataset d"; } if (fieldAllList.size() > 0 || fieldAnyList.size() > 0 || fieldNotInList.size() > 0) { String fieldQuery = "SELECT DISTINCT dataset_id FROM dict_field_detail f WHERE ("; query += " WHERE d.id IN ( "; query += fieldQuery; String whereClause = ""; boolean fieldNeedAndKeyword = false; if (fieldAnyList.size() > 0) { whereClause = " ("; int indexForAnyList = 0; for (String field : fieldAnyList) { if (indexForAnyList == 0) { whereClause += "f.field_name LIKE '%" + field + "%'"; } else { whereClause += " or f.field_name LIKE '%" + field + "%'"; } indexForAnyList++; } whereClause += " ) "; fieldNeedAndKeyword = true; query += whereClause; } if (fieldAllList.size() > 0 && StringUtils.isNotBlank(fieldAllIDs)) { if (fieldNeedAndKeyword) { whereClause = " and ("; } else { whereClause = " ("; } whereClause += "f.dataset_id IN (" + fieldAllIDs + ")"; whereClause += " ) "; query += whereClause; fieldNeedAndKeyword = true; } if (fieldNotInList.size() > 0) { if (fieldNeedAndKeyword) { whereClause = " and ( f.dataset_id not in (select dataset_id from dict_field_detail where"; } else { whereClause = " ( f.dataset_id not in (select dataset_id from dict_field_detail where"; } int indexForNotInList = 0; for (String field : fieldNotInList) { if (indexForNotInList == 0) { whereClause += " field_name LIKE '%" + field + "%'"; } else { whereClause += " or field_name LIKE '%" + field + "%'"; } indexForNotInList++; } whereClause += " )) "; query += whereClause; fieldNeedAndKeyword = true; } needAndKeyword = true; query += ") )"; } if (scopeInList.size() > 0 || scopeNotInList.size() > 0) { if (needAndKeyword) { query += " and"; } else { query += " where"; } boolean scopeNeedAndKeyword = false; if (scopeInList.size() > 0) { query += " d.parent_name in ("; scopeNeedAndKeyword = true; int indexForScopeInList = 0; for (String scope : scopeInList) { if (indexForScopeInList == 0) { query += "'" + scope + "'"; } else { query += ", '" + scope + "'"; } indexForScopeInList++; } query += ") "; } if (scopeNotInList.size() > 0) { if (scopeNeedAndKeyword) { query += " and d.parent_name not in ("; } else { query += " d.parent_name not in ("; } int indexForScopeNotInList = 0; for (String scope : scopeNotInList) { if (indexForScopeNotInList == 0) { query += "'" + scope + "'"; } else { query += ", '" + scope + "'"; } indexForScopeNotInList++; } query += ") "; } needAndKeyword = true; } String condition1 = ""; String condition2 = ""; String condition3 = ""; String condition4 = ""; if (tableInList.size() > 0 || tableNotInList.size() > 0) { if (needAndKeyword) { query += " and"; } else { query += " where"; } boolean tableNeedAndKeyword = false; if (tableInList.size() > 0) { query += " ("; int indexForTableInList = 0; for (String table : tableInList) { if (indexForTableInList == 0) { query += "d.name LIKE '%" + table + "%'"; } else { condition1 += " or "; condition2 += " or "; condition3 += " or "; condition4 += " or "; query += " or d.name LIKE '%" + table + "%'"; } condition1 += "name = '" + table + "'"; condition2 += "name LIKE '" + table + "%'"; condition3 += "name LIKE '%" + table + "'"; condition4 += "name LIKE '%" + table + "%'"; indexForTableInList++; } query += " ) "; tableNeedAndKeyword = true; } if (tableNotInList.size() > 0) { if (tableNeedAndKeyword) { query += " and ("; } else { query += " ("; } int indexForTableNotInList = 0; for (String table : tableNotInList) { if (indexForTableNotInList == 0) { query += "d.name NOT LIKE '%" + table + "%'"; } else { query += " and d.name NOT LIKE '%" + table + "%'"; } indexForTableNotInList++; } query += " ) "; } needAndKeyword = true; } if (StringUtils.isNotBlank(datasetSources)) { if (needAndKeyword) { query += " and"; } else { query += " WHERE"; } query += " d.source in ("; String[] dataestSourceArray = datasetSources.split(","); for (int i = 0; i < dataestSourceArray.length; i++) { query += "'" + dataestSourceArray[i] + "'"; if (i != (dataestSourceArray.length - 1)) { query += ","; } } query += ")"; } if ((tableInList.size() > 0 || tableNotInList.size() > 0) && StringUtils.isNotBlank(condition1) && StringUtils.isNotBlank(condition2) && StringUtils.isNotBlank(condition3) && StringUtils.isNotBlank(condition4)) { query += ADVSEARCH_RANK_CLAUSE.replace("$condition1", condition1).replace("$condition2", condition2) .replace("$condition3", condition3).replace("$condition4", condition4); } else { query += " ORDER BY CASE WHEN urn LIKE 'teradata://DWH_%' THEN 2 " + "WHEN urn LIKE 'hdfs://data/tracking/%' THEN 1 " + "WHEN urn LIKE 'teradata://DWH/%' THEN 3 " + "WHEN urn LIKE 'hdfs://data/databases/%' THEN 4 " + "WHEN urn LIKE 'hdfs://data/dervied/%' THEN 5 ELSE 99 end, urn"; } if (StringUtils.isBlank(comments)) { query += " LIMIT " + (page - 1) * size + ", " + size; final String queryString = query; result = txTemplate.execute(new TransactionCallback<ObjectNode>() { public ObjectNode doInTransaction(TransactionStatus status) { List<Map<String, Object>> rows = null; rows = jdbcTemplate.queryForList(queryString); for (Map row : rows) { Dataset ds = new Dataset(); ds.id = (Long) row.get("id"); ds.name = (String) row.get("name"); ds.source = (String) row.get("source"); ds.urn = (String) row.get("urn"); ds.schema = (String) row.get("schema"); pagedDatasets.add(ds); } long count = 0; try { count = jdbcTemplate.queryForObject("SELECT FOUND_ROWS()", Long.class); } catch (EmptyResultDataAccessException e) { Logger.error("Exception = " + e.getMessage()); } ObjectNode resultNode = Json.newObject(); resultNode.put("count", count); resultNode.put("page", page); resultNode.put("itemsPerPage", size); resultNode.put("totalPages", (int) Math.ceil(count / ((double) size))); resultNode.set("data", Json.toJson(pagedDatasets)); return resultNode; } }); return result; } else { String datasetIDStr = ""; final String queryString = query; datasetIDStr = txTemplate.execute(new TransactionCallback<String>() { public String doInTransaction(TransactionStatus status) { List<Map<String, Object>> rows = null; rows = jdbcTemplate.queryForList(queryString); String idsString = ""; for (Map row : rows) { Long id = (Long) row.get("id"); idsString += Long.toString(id) + ","; } if (StringUtils.isNotBlank(idsString)) { idsString = idsString.substring(0, idsString.length() - 1); } return idsString; } }); if (StringUtils.isBlank(datasetIDStr)) { resultNode.put("count", 0); resultNode.put("page", page); resultNode.put("itemsPerPage", size); resultNode.put("totalPages", 0); resultNode.set("data", Json.toJson("")); return resultNode; } final String commentsQueryWithConditionStr = DATASET_BY_COMMENT_PAGINATION_IN_CLAUSE .replace("$keyword", comments).replace("$id_list", datasetIDStr); result = txTemplate.execute(new TransactionCallback<ObjectNode>() { public ObjectNode doInTransaction(TransactionStatus status) { List<Map<String, Object>> rows = null; rows = jdbcTemplate.queryForList(commentsQueryWithConditionStr, (page - 1) * size, size); for (Map row : rows) { Dataset ds = new Dataset(); ds.id = (Long) row.get("id"); ds.name = (String) row.get("name"); ds.source = (String) row.get("source"); ds.urn = (String) row.get("urn"); ds.schema = (String) row.get("schema"); pagedDatasets.add(ds); } long count = 0; try { count = jdbcTemplate.queryForObject("SELECT FOUND_ROWS()", Long.class); } catch (EmptyResultDataAccessException e) { Logger.error("Exception = " + e.getMessage()); } ObjectNode resultNode = Json.newObject(); resultNode.put("count", count); resultNode.put("page", page); resultNode.put("itemsPerPage", size); resultNode.put("totalPages", (int) Math.ceil(count / ((double) size))); resultNode.set("data", Json.toJson(pagedDatasets)); return resultNode; } }); return result; } } resultNode.put("count", 0); resultNode.put("page", page); resultNode.put("itemsPerPage", size); resultNode.put("totalPages", 0); resultNode.set("data", Json.toJson("")); return resultNode; }
From source file:org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfigurationTests.java
@Test public void testJdbcTemplateExists() throws Exception { this.context.register(DataSourceAutoConfiguration.class, PropertyPlaceholderAutoConfiguration.class); this.context.refresh(); JdbcTemplate jdbcTemplate = this.context.getBean(JdbcTemplate.class); assertNotNull(jdbcTemplate);/* ww w . j a va2 s . c o m*/ assertNotNull(jdbcTemplate.getDataSource()); }
From source file:org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfigurationTests.java
@Test public void testJdbcTemplateExistsWithCustomDataSource() throws Exception { this.context.register(TestDataSourceConfiguration.class, DataSourceAutoConfiguration.class, PropertyPlaceholderAutoConfiguration.class); this.context.refresh(); JdbcTemplate jdbcTemplate = this.context.getBean(JdbcTemplate.class); assertNotNull(jdbcTemplate);/* w w w. j av a2s .c om*/ assertTrue(jdbcTemplate.getDataSource() instanceof BasicDataSource); }
From source file:org.springframework.boot.autoconfigure.jdbc.JdbcTemplateAutoConfigurationTests.java
@Test public void testJdbcTemplateExists() throws Exception { this.context.register(DataSourceAutoConfiguration.class, JdbcTemplateAutoConfiguration.class, PropertyPlaceholderAutoConfiguration.class); this.context.refresh(); JdbcTemplate jdbcTemplate = this.context.getBean(JdbcTemplate.class); assertThat(jdbcTemplate).isNotNull(); assertThat(jdbcTemplate.getDataSource()).isNotNull(); }
From source file:org.springframework.boot.autoconfigure.jdbc.JdbcTemplateAutoConfigurationTests.java
@Test public void testJdbcTemplateExistsWithCustomDataSource() throws Exception { this.context.register(TestDataSourceConfiguration.class, DataSourceAutoConfiguration.class, JdbcTemplateAutoConfiguration.class, PropertyPlaceholderAutoConfiguration.class); this.context.refresh(); JdbcTemplate jdbcTemplate = this.context.getBean(JdbcTemplate.class); assertThat(jdbcTemplate).isNotNull(); assertThat(jdbcTemplate.getDataSource() instanceof BasicDataSource).isTrue(); }
From source file:org.springframework.jdbc.core.JdbcTemplateTests.java
public void testBeanProperties() throws Exception { replay();/*from w ww . j a v a 2 s . c om*/ JdbcTemplate template = new JdbcTemplate(mockDataSource); assertTrue("datasource ok", template.getDataSource() == mockDataSource); assertTrue("ignores warnings by default", template.isIgnoreWarnings()); template.setIgnoreWarnings(false); assertTrue("can set NOT to ignore warnings", !template.isIgnoreWarnings()); }