blob: 7e8e05afd855f5a78f2de72b143c93a032d42f31 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
/*-
* ============LICENSE_START=======================================================
* openECOMP : SDN-C
* ================================================================================
* Copyright (C) 2017 AT&T Intellectual Property. All rights
* reserved.
* ================================================================================
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* ============LICENSE_END=========================================================
*/
package jtest.util.org.openecomp.sdnc.ra;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.support.rowset.SqlRowSet;
public class TestTable {
private String tableName;
private String[] columnList;
private String idName;
private String insertSql;
private JdbcTemplate jdbcTemplate;
public TestTable(JdbcTemplate jdbcTemplate, String tableName, String idName, String... columnList) {
this.jdbcTemplate = jdbcTemplate;
this.tableName = tableName;
this.idName = idName;
this.columnList = columnList;
createInsertSql();
}
private void createInsertSql() {
StringBuilder ss = new StringBuilder();
ss.append("INSERT INTO ").append(tableName).append(" (");
for (String s : columnList)
ss.append(s).append(", ");
ss.setLength(ss.length() - 2);
ss.append(") VALUES (");
for (int i = 0; i < columnList.length; i++)
ss.append("?, ");
ss.setLength(ss.length() - 2);
ss.append(")");
insertSql = ss.toString();
}
public void add(Object... values) {
jdbcTemplate.update(insertSql, values);
}
public long getLastId() {
return jdbcTemplate.queryForObject("SELECT max(" + idName + ") FROM " + tableName, Long.class);
}
public Long getId(String where) {
String selectSql = "SELECT " + idName + " FROM " + tableName + " WHERE " + where;
SqlRowSet rs = jdbcTemplate.queryForRowSet(selectSql);
if (rs.first())
return rs.getLong(idName);
return null;
}
public boolean exists(String where) {
String selectSql = "SELECT * FROM " + tableName + " WHERE " + where;
SqlRowSet rs = jdbcTemplate.queryForRowSet(selectSql);
return rs.first();
}
public void delete(String where) {
jdbcTemplate.update("DELETE FROM " + tableName + " WHERE " + where);
}
}
|