X-Git-Url: https://code.wpia.club/?p=gigi.git;a=blobdiff_plain;f=src%2Forg%2Fcacert%2Fgigi%2Fping%2FDomainPinger.java;h=c5eb3926bea8d8a2fed6b0b766b11171d9a13ae0;hp=550c86afa176f6391e5b83e001c6d07efbb6d42a;hb=0b86fb147b4a61f315770fa5bba4466ca18ddfa8;hpb=33a87a70ba2320451c3b3b1faa9b487dbcbd3540 diff --git a/src/org/cacert/gigi/ping/DomainPinger.java b/src/org/cacert/gigi/ping/DomainPinger.java index 550c86af..c5eb3926 100644 --- a/src/org/cacert/gigi/ping/DomainPinger.java +++ b/src/org/cacert/gigi/ping/DomainPinger.java @@ -1,12 +1,21 @@ package org.cacert.gigi.ping; -import org.cacert.gigi.database.DatabaseConnection; +import org.cacert.gigi.database.DBEnum; import org.cacert.gigi.database.GigiPreparedStatement; import org.cacert.gigi.dbObjects.CertificateOwner; import org.cacert.gigi.dbObjects.Domain; public abstract class DomainPinger { + public static enum PingState implements DBEnum { + OPEN, SUCCESS, FAILED; + + @Override + public String getDBName() { + return toString().toLowerCase(); + } + } + public static final String PING_STILL_PENDING = null; public static final String PING_SUCCEDED = ""; @@ -14,22 +23,24 @@ public abstract class DomainPinger { public abstract void ping(Domain domain, String configuration, CertificateOwner target, int confId); protected static void enterPingResult(int configId, String state, String result, String token) { - GigiPreparedStatement enterPingResult = DatabaseConnection.getInstance().prepare("INSERT INTO `domainPinglog` SET `configId`=?, `state`=?::`pingState`, `result`=?, `challenge`=?"); - enterPingResult.setInt(1, configId); - enterPingResult.setString(2, DomainPinger.PING_STILL_PENDING == state ? "open" : DomainPinger.PING_SUCCEDED.equals(state) ? "success" : "failed"); - enterPingResult.setString(3, result); - enterPingResult.setString(4, token); - enterPingResult.execute(); - + PingState estate = DomainPinger.PING_STILL_PENDING == state ? PingState.OPEN : DomainPinger.PING_SUCCEDED.equals(state) ? PingState.SUCCESS : PingState.FAILED; + try (GigiPreparedStatement enterPingResult = new GigiPreparedStatement("INSERT INTO `domainPinglog` SET `configId`=?, `state`=?::`pingState`, `result`=?, `challenge`=?")) { + enterPingResult.setInt(1, configId); + enterPingResult.setEnum(2, estate); + enterPingResult.setString(3, result); + enterPingResult.setString(4, token); + enterPingResult.execute(); + } } - protected static void updatePingResult(int configId, String token, String state, String result) { - GigiPreparedStatement updatePingResult = DatabaseConnection.getInstance().prepare("UPDATE `domainPinglog` SET `state`=?::`pingState`, `result`=? WHERE `configId`=? AND `challenge`=?"); - updatePingResult.setString(1, DomainPinger.PING_STILL_PENDING == state ? "open" : DomainPinger.PING_SUCCEDED.equals(state) ? "success" : "failed"); - updatePingResult.setString(2, result); - updatePingResult.setInt(3, configId); - updatePingResult.setString(4, token); - updatePingResult.execute(); - + protected static void updatePingResult(int configId, String state, String result, String token) { + try (GigiPreparedStatement updatePingResult = new GigiPreparedStatement("UPDATE `domainPinglog` SET `state`=?::`pingState`, `result`=? WHERE `configId`=? AND `challenge`=?")) { + updatePingResult.setString(1, DomainPinger.PING_STILL_PENDING == state ? "open" : DomainPinger.PING_SUCCEDED.equals(state) ? "success" : "failed"); + updatePingResult.setString(2, result); + updatePingResult.setInt(3, configId); + updatePingResult.setString(4, token); + updatePingResult.execute(); + } } + }