src/org/cacert/gigi/ping/PingerDaemon.java

   1 package org.cacert.gigi.ping;
   2
   3 import java.io.FileReader;
   4 import java.io.IOException;
   5 import java.sql.PreparedStatement;
   6 import java.sql.ResultSet;
   7 import java.sql.SQLException;
   8 import java.util.HashMap;
   9 import java.util.Properties;
  10
  11 import org.cacert.gigi.database.DatabaseConnection;
  12
  13 public class PingerDaemon implements Runnable {
  14
  15     HashMap<String, DomainPinger> pingers = new HashMap<>();
  16
  17     public PingerDaemon() {
  18         // pingers.put("email",);
  19         pingers.put("ssl", new SSLPinger());
  20         pingers.put("http", new HTTPFetch());
  21         pingers.put("dns", new DNSPinger());
  22
  23     }
  24
  25     @Override
  26     public void run() {
  27         try {
  28             PreparedStatement ps = DatabaseConnection.getInstance().prepare("SELECT pingconfig.*, domains.domain FROM pingconfig LEFT JOIN domainPinglog ON domainPinglog.configId=pingconfig.id INNER JOIN domains ON domains.id=pingconfig.domainid WHERE domainPinglog.configId IS NULL ");
  29             PreparedStatement result = DatabaseConnection.getInstance().prepare("INSERT INTO domainPinglog SET configId=?, state=?, result=?");
  30
  31             ResultSet rs = ps.executeQuery();
  32             while (rs.next()) {
  33                 String type = rs.getString("type");
  34                 String config = rs.getString("info");
  35                 System.out.println(type);
  36                 System.out.println(config);
  37                 DomainPinger dp = pingers.get(type);
  38                 if (dp != null) {
  39                     result.setInt(1, rs.getInt("id"));
  40                     String resp = dp.ping(rs.getString("domain"), config);
  41                     result.setString(2, resp == DomainPinger.PING_STILL_PENDING ? "open" : resp == DomainPinger.PING_SUCCEDED ? "success" : "failed");
  42                     result.setString(3, resp);
  43                     result.execute();
  44                 }
  45             }
  46         } catch (SQLException e) {
  47             e.printStackTrace();
  48         }
  49     }
  50
  51     public static void main(String[] args) throws IOException {
  52         Properties conf = new Properties();
  53         conf.load(new FileReader("config/gigi.properties"));
  54         DatabaseConnection.init(conf);
  55         new PingerDaemon().run();
  56
  57     }
  58 }