Author: [log in to unmask]
Date: Tue Nov 25 01:50:46 2014
New Revision: 1581
Log:
Add a method for inserting a collection in a transaction using a SQL prepared statement.
Modified:
java/trunk/conditions/src/main/java/org/hps/conditions/database/DatabaseConditionsManager.java
Modified: java/trunk/conditions/src/main/java/org/hps/conditions/database/DatabaseConditionsManager.java
=============================================================================
--- java/trunk/conditions/src/main/java/org/hps/conditions/database/DatabaseConditionsManager.java (original)
+++ java/trunk/conditions/src/main/java/org/hps/conditions/database/DatabaseConditionsManager.java Tue Nov 25 01:50:46 2014
@@ -11,17 +11,21 @@
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.sql.Connection;
+import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
+import java.util.Date;
import java.util.Iterator;
import java.util.List;
+import java.util.Map.Entry;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.hps.conditions.api.ConditionsObject;
import org.hps.conditions.api.ConditionsObjectCollection;
+import org.hps.conditions.api.ConditionsObjectException;
import org.hps.conditions.api.ConditionsRecord;
import org.hps.conditions.api.ConditionsRecord.ConditionsRecordCollection;
import org.hps.conditions.api.ConditionsSeries;
@@ -58,19 +62,21 @@
@SuppressWarnings("rawtypes")
public class DatabaseConditionsManager extends ConditionsManagerImplementation {
- static String connectionProperty = "org.hps.conditions.connection.file";
+ protected static final String CONNECTION_PROPERTY = "org.hps.conditions.connection.file";
+ protected static final String DEFAULT_CONFIG = "/org/hps/conditions/config/conditions_dev.xml";
+ protected static final String TEST_RUN_CONFIG = "/org/hps/conditions/config/conditions_database_testrun_2012.xml";
+ protected static final int TEST_RUN_MAX_RUN = 1365;
+
+ protected static Logger logger = LogUtil.create(DatabaseConditionsManager.class);
+
protected String detectorName;
protected List<TableMetaData> tableMetaData;
protected List<ConditionsConverter> converters;
protected File connectionPropertiesFile;
- protected static Logger logger = LogUtil.create(DatabaseConditionsManager.class);
protected ConnectionParameters connectionParameters = new DefaultConnectionParameters();
protected Connection connection;
protected boolean isConnected = false;
protected ConditionsSeriesConverter conditionsSeriesConverter = new ConditionsSeriesConverter(this);
- static final String DEFAULT_CONFIG = "/org/hps/conditions/config/conditions_dev.xml";
- static final String TEST_RUN_CONFIG = "/org/hps/conditions/config/conditions_database_testrun_2012.xml";
- protected static final int TEST_RUN_MAX_RUN = 1365;
protected boolean isInitialized = false;
protected String resourceConfig = null;
protected File fileConfig = null;
@@ -81,6 +87,7 @@
protected TestRunSvtDetectorSetup testRunSvtloader = new TestRunSvtDetectorSetup();
protected SvtDetectorSetup svtLoader = new SvtDetectorSetup();
protected String tag = null;
+ //protected boolean dryRun = false;
/**
* Default connection parameters which will use the SLAC database by default,
@@ -588,7 +595,57 @@
public void setTag(String tag) {
this.tag = tag;
}
-
+
+ public <ObjectType extends ConditionsObject> void insertCollection(ConditionsObjectCollection<ObjectType> collection) throws SQLException {
+ if (collection == null) {
+ throw new IllegalArgumentException("The collection is null.");
+ }
+ if (collection.getObjects().size() == 0) {
+ throw new IllegalArgumentException("The collection is empty.");
+ }
+ if (collection.getTableMetaData() == null) {
+ throw new RuntimeException("The collection does not have table meta data.");
+ }
+ TableMetaData tableMetaData = collection.getTableMetaData();
+ if (collection.getCollectionId() == -1) {
+ try {
+ collection.setCollectionId(this.getNextCollectionID(tableMetaData.getTableName()));
+ } catch (ConditionsObjectException e) {
+ throw new RuntimeException(e);
+ }
+ }
+ logger.fine("inserting collection with ID " + collection.getCollectionId()
+ + " and key " + collection.getTableMetaData().getKey() + " into table " + tableMetaData.getTableName());
+
+ try {
+ connection.setAutoCommit(false);
+ logger.finest("starting insert transaction");
+ String sql = QueryBuilder.buildPreparedInsert(collection.get(0));
+ PreparedStatement preparedStatement =
+ connection.prepareStatement(sql);
+ logger.finest("using prepared statement: " + sql);
+ logger.finest("preparing updates");
+ for (ConditionsObject object : collection.getObjects()) {
+ preparedStatement.setObject(1, object.getCollectionId());
+ int parameterIndex = 2;
+ for (Entry<String,Object> entry : object.getFieldValues().entrySet()) {
+ preparedStatement.setObject(parameterIndex, entry.getValue());
+ ++parameterIndex;
+ }
+ preparedStatement.executeUpdate();
+ }
+ logger.finest("done preparing updates");
+ connection.commit();
+ logger.finest("committed transaction");
+ } catch (Exception e) {
+ logger.warning("rolling back transaction");
+ connection.rollback();
+ logger.warning("transaction was rolled back");
+ } finally {
+ connection.setAutoCommit(true);
+ }
+ }
+
private void setupEcal() {
logger.config("setting up ECAL conditions on detector");
EcalConditions conditions = getCachedConditions(EcalConditions.class, ECAL_CONDITIONS).getCachedData();
@@ -691,11 +748,11 @@
* {@link #setConnectionProperties(File)} or {@link #setConnectionResource(String)}.
*/
private void setupConnectionFromSystemProperty() {
- String systemPropertiesConnectionPath = (String) System.getProperties().get(connectionProperty);
+ String systemPropertiesConnectionPath = (String) System.getProperties().get(CONNECTION_PROPERTY);
if (systemPropertiesConnectionPath != null) {
File f = new File(systemPropertiesConnectionPath);
if (!f.exists()) {
- throw new RuntimeException("Connection properties file from " + connectionProperty + " does not exist.");
+ throw new RuntimeException("Connection properties file from " + CONNECTION_PROPERTY + " does not exist.");
}
this.setConnectionProperties(f);
}
