blob: 35849b481fd11faa6940590352c6e5f31c4787ab (
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
|
package vid.automation.test.services;
import com.google.common.collect.ImmutableList;
import vid.automation.test.utils.DB_CONFIG;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.List;
public class AsyncJobsService {
public void dropAllAsyncJobs() {
runStatementsInDb(ImmutableList.of(
"DELETE FROM `vid_service_info`",
"DELETE FROM `vid_job`",
"DELETE FROM `vid_job_audit_status`",
"DELETE FROM `vid_name_counter`"
));
}
public void runStatementsInDb(List<String> sqlStatements) {
System.out.println("Connecting database...");
try (Connection connection = DriverManager.getConnection(DB_CONFIG.url, DB_CONFIG.username, DB_CONFIG.password)) {
System.out.println("Database connected!");
try (Statement stmt = connection.createStatement()) {
for (String sql: sqlStatements) {
stmt.addBatch(sql);
}
stmt.executeBatch();
}
} catch (SQLException e) {
throw new IllegalStateException("Cannot connect the database!", e);
}
}
public void runStatementInDb(String sqlStatement) {
runStatementsInDb(ImmutableList.of(sqlStatement));
}
public void muteAllAsyncJobs() {
runStatementInDb("UPDATE `vid_job` SET `TAKEN_BY`='muteAllAsyncJobs', `JOB_STATUS`=CONCAT('MUTED_', `JOB_STATUS`), `AGE`=`AGE`+5 WHERE `TAKEN_BY` is NULL or `TAKEN_BY`<>'muteAllAsyncJobs'");
}
public void dropAllFromNameCounter() {
runStatementInDb("DELETE FROM `vid_name_counter`");
}
}
|
