");
out.println("");
out.println("");
}
/**
* Handles POST method requests. Reads values as entered by the user in the
* harvester registration form and checks them for validity. Then either
* inserts, updates, or removes a record in the HARVEST_SITE_SCHEDULE table.
*
* @param req the request
* @param res the response
* @throws ServletException
* @throws IOException
*/
public void doPost(HttpServletRequest req, HttpServletResponse res)
throws ServletException, IOException {
Connection conn = getConnection();
int maxValue;
boolean remove = false; // if true, remove record
int siteScheduleID;
String contactEmail;
String dateLastHarvest;
String dateNextHarvest;
String documentListURL;
HttpSession httpSession;
String ldapDN;
String ldapPwd;
PrintWriter out;
ParameterParser parameterParser = new ParameterParser(req);
String register;
String unit;
String unregister;
int updateFrequency;
boolean validValues;
httpSession = req.getSession(false);
if (httpSession == null) {
System.out.println("User did not log in.");
return;
}
// The user name and password are stored as session attributes by the
// HarvesterRegistrationLogin servlet
ldapDN = (String) httpSession.getAttribute("username");
ldapPwd = (String) httpSession.getAttribute("password");
contactEmail = parameterParser.getStringParameter("contactEmail", "None");
documentListURL = parameterParser.getStringParameter("documentListURL", "");
unit = parameterParser.getStringParameter("unit", "days");
updateFrequency = parameterParser.getIntParameter("updateFrequency", 1);
register = parameterParser.getStringParameter("register", "");
unregister = parameterParser.getStringParameter("unregister", "");
remove = (unregister.equalsIgnoreCase("Unregister"));
siteScheduleID = getSiteScheduleID(conn, ldapDN);
dateLastHarvest = getDateLastHarvest(conn, siteScheduleID);
res.setContentType("text/plain");
out = res.getWriter();
if (!remove) {
validValues = checkValues(out, documentListURL, updateFrequency);
if (!validValues) {
return;
}
}
if (siteScheduleID == 0) {
if (remove) {
// The user clicked Unregister, but no existing record was found
System.out.println("Unable to remove record for user " + ldapDN);
System.out.println("No matching record found in HARVEST_SITE_SCHEDULE");
out.println("No record found for user " + ldapDN + ".");
out.println("Since you were not registered, no action was taken.");
}
else {
maxValue = getMaxValue(conn,
"HARVEST_SITE_SCHEDULE",
"SITE_SCHEDULE_ID");
siteScheduleID = maxValue + 1;
dbInsert(out, conn, siteScheduleID, contactEmail, documentListURL,
ldapDN, ldapPwd, unit, updateFrequency);
}
}
else {
// Either update or remove an existing record
if (remove) {
dbRemove(out, conn, siteScheduleID, ldapDN);
}
else {
dbUpdate(out, conn, siteScheduleID, contactEmail, documentListURL,
ldapDN, ldapPwd, unit, updateFrequency, dateLastHarvest);
}
}
closeConnection(conn);
}
/**
* Gets a database connection.
*
* @return conn, the Connection object
*/
private Connection getConnection() {
Connection conn = null;
SQLWarning warn;
// Make the database connection
try {
System.out.println("Getting connection to Harvester tables");
conn = DriverManager.getConnection(defaultDB, user, password);
// If a SQLWarning object is available, print its warning(s).
// There may be multiple warnings chained.
warn = conn.getWarnings();
if (warn != null) {
while (warn != null) {
System.out.println("SQLState: " + warn.getSQLState());
System.out.println("Message: " + warn.getMessage());
System.out.println("Vendor: " + warn.getErrorCode());
System.out.println("");
warn = warn.getNextWarning();
}
}
}
catch (SQLException e) {
System.out.println("Database access failed " + e);
}
return conn;
}
/**
* Gets the date of last harvest value from the HARVEST_SITE_SCHEDULE table,
* given a siteScheduleID value (the primary key).
*
* @param conn the connection
* @param siteScheduleID the primary key
* @return dateLastHarvest the string stored in the table, e.g.
* "2004-04-30 00:00:00.0" or ""
*/
private String getDateLastHarvest(Connection conn, int siteScheduleID) {
String dateLastHarvest = "";
String query = "SELECT DATELASTHARVEST FROM HARVEST_SITE_SCHEDULE " +
"WHERE SITE_SCHEDULE_ID=" + siteScheduleID;
Statement stmt;
try {
stmt = conn.createStatement();
ResultSet rs = stmt.executeQuery(query);
while (rs.next()) {
dateLastHarvest = rs.getString("DATELASTHARVEST");
if (rs.wasNull()) {
dateLastHarvest = ""; // Convert null value to empty string
}
}
stmt.close();
}
catch (SQLException ex) {
System.out.println("SQLException: " + ex.getMessage());
}
return dateLastHarvest;
}
/**
* Gets the maximum value of an integer field from a table, given the table
* name and the field name.
*
* @param tableName the database table name
* @param fieldName the field name of the integer field in the table
* @return the maximum integer stored in the fieldName field of tableName
*/
private int getMaxValue(Connection conn, String tableName, String fieldName) {
int maxValue = 0;
int fieldValue;
String query = "SELECT " + fieldName + " FROM " + tableName;
Statement stmt;
try {
stmt = conn.createStatement();
ResultSet rs = stmt.executeQuery(query);
while (rs.next()) {
fieldValue = rs.getInt(fieldName);
maxValue = Math.max(maxValue, fieldValue);
}
stmt.close();
}
catch (SQLException ex) {
System.out.println("SQLException: " + ex.getMessage());
}
return maxValue;
}
/**
* Gets the siteScheduleID value from the HARVEST_SITE_SCHEDULE table, given
* the value of the ldapDN field.
*
* @param conn the database connection
* @param ldapDN the ldap DN string
* @return siteScheduleID, an integer, the primary key
*/
private int getSiteScheduleID(Connection conn, String ldapDN) {
String ldapDNValue; // value of LDAPDN field
String query = "SELECT * FROM HARVEST_SITE_SCHEDULE";
int siteScheduleID = 0;
Statement stmt;
try {
stmt = conn.createStatement();
ResultSet rs = stmt.executeQuery(query);
while (rs.next()) {
ldapDNValue = rs.getString("LDAPDN");
if (ldapDNValue.equalsIgnoreCase(ldapDN)) {
siteScheduleID = rs.getInt("SITE_SCHEDULE_ID");
}
}
stmt.close();
}
catch (SQLException ex) {
System.out.println("SQLException: " + ex.getMessage());
}
return siteScheduleID;
}
/**
* Initializes the servlet. Reads properties and initializes object fields.
*
* @throws ServletException
*/
public void init(ServletConfig config) throws ServletException {
String database;
String dbDriver = "";
String dirPath;
super.init(config);
this.config = config;
this.context = config.getServletContext();
dirPath = context.getRealPath(CONFIG_DIR);
try {
ServletContext context = config.getServletContext();
PropertyService.getInstance(context);
dbDriver = PropertyService.getProperty("database.driver");
defaultDB = PropertyService.getProperty("database.connectionURI");
password = PropertyService.getProperty("database.password");
user = PropertyService.getProperty("database.user");
} catch (ServiceException se) {
System.out.println("Error initializing PropertyService: " + se.getMessage());
} catch (PropertyNotFoundException pnfe) {
System.out.println("Error reading property: " + pnfe.getMessage());
}
// Load the jdbc driver
try {
Class.forName(dbDriver);
} catch (ClassNotFoundException e) {
System.out.println("Can't load driver " + e);
}
}
/**
* Surrounds a string with single quotes.
*
* @param str
* the original string
* @return the quoted string
*/
private String quoteString(String str) {
return "'" + str + "'";
}
/**
* Reports the results of an insert or update to the client.
*
* @param out the PrintWriter
* @param ldapDN the LDAP DN string
* @param contactEmail the email address of the site contact
* @param documentListURL the URL of the harvester document list at the site
* @param updateFrequency the harvest update frequency
* @param unit the unit (e.g. "days", "weeks", "months"
* @param dateNextHarvest the date of the next scheduled harvest
*/
private void reportResults(PrintWriter out,
String ldapDN,
String contactEmail,
String documentListURL,
int updateFrequency,
String unit,
String dateNextHarvest
) {
out.println("Harvester registration updated for " + ldapDN);
out.println(" Email Address: " + contactEmail);
out.println(" Harvest List URL: " + documentListURL);
out.println(" Harvest Frequency: " + updateFrequency);
out.println(" Unit: " + unit);
out.println("");
out.println("Next scheduled harvest date: " + dateNextHarvest);
}
}