public static Set<RecordDefinition> edgeRecordDefinitions(final Node<?> node) {
final Set<RecordDefinition> recordDefinitions = new HashSet<>();
for (final Edge<?> edge : node.getEdges()) {
final Object object = edge.getObject();
if (object instanceof Record) {
final Record record = (Record) object;
final RecordDefinition recordDefinition = record.getRecordDefinition();
recordDefinitions.add(recordDefinition);
}
}
return recordDefinitions;
}
private void update(final Record object) throws SQLException {
final RecordDefinition objectType = object.getRecordDefinition();
final PathName typePath = objectType.getPathName();
final RecordDefinition recordDefinition = getRecordDefinition(typePath);
flushIfRequired(recordDefinition);
PreparedStatement statement = this.typeUpdateStatementMap.get(typePath);
if (statement == null) {
final String sql = getUpdateSql(recordDefinition);
try {
statement = this.connection.prepareStatement(sql);
this.typeUpdateStatementMap.put(typePath, statement);
} catch (final SQLException e) {
LOG.error(sql, e);
}
}
int parameterIndex = 1;
final List<FieldDefinition> idFields = recordDefinition.getIdFields();
for (final FieldDefinition fieldDefinition : recordDefinition.getFields()) {
if (!idFields.contains(fieldDefinition)) {
final JdbcFieldDefinition jdbcFieldDefinition = (JdbcFieldDefinition) fieldDefinition;
parameterIndex =
jdbcFieldDefinition.setInsertPreparedStatementValue(statement, parameterIndex, object);
}
}
for (final FieldDefinition idField : idFields) {
final JdbcFieldDefinition jdbcFieldDefinition = (JdbcFieldDefinition) idField;
parameterIndex =
jdbcFieldDefinition.setInsertPreparedStatementValue(statement, parameterIndex, object);
}
statement.addBatch();
Integer batchCount = this.typeUpdateBatchCountMap.get(typePath);
if (batchCount == null) {
batchCount = 1;
this.typeUpdateBatchCountMap.put(typePath, 1);
} else {
batchCount += 1;
this.typeUpdateBatchCountMap.put(typePath, batchCount);
}
if (batchCount >= this.batchSize) {
final String sql = getUpdateSql(recordDefinition);
processCurrentBatch(typePath, sql, statement, this.typeUpdateBatchCountMap);
}
this.recordStore.addStatistic("Update", object);
}
@Override
public synchronized void write(final Record record) {
try {
final RecordDefinition recordDefinition = record.getRecordDefinition();
final RecordStore recordStore = recordDefinition.getRecordStore();
final RecordState state = record.getState();
if (recordStore != this.recordStore) {
if (state != RecordState.DELETED) {
insert(record);
}
} else {
switch (state) {
case NEW:
insert(record);
break;
case MODIFIED:
update(record);
break;
case PERSISTED:
// No action required
break;
case DELETED:
delete(record);
break;
default:
throw new IllegalStateException("State not known");
}
}
} catch (final RuntimeException e) {
throw e;
} catch (final Error e) {
throw e;
} catch (final BatchUpdateException e) {
for (SQLException e1 = e.getNextException(); e1 != null; e1 = e1.getNextException()) {
LOG.error("Unable to write", e1);
}
throw new RuntimeException("Unable to write", e);
} catch (final Exception e) {
throw new RuntimeException("Unable to write", e);
}
}
private void delete(final Record object) throws SQLException {
final RecordDefinition objectType = object.getRecordDefinition();
final PathName typePath = objectType.getPathName();
final RecordDefinition recordDefinition = getRecordDefinition(typePath);
flushIfRequired(recordDefinition);
PreparedStatement statement = this.typeDeleteStatementMap.get(typePath);
if (statement == null) {
final String sql = getDeleteSql(recordDefinition);
try {
statement = this.connection.prepareStatement(sql);
this.typeDeleteStatementMap.put(typePath, statement);
} catch (final SQLException e) {
LOG.error(sql, e);
}
}
int parameterIndex = 1;
final JdbcFieldDefinition idField = (JdbcFieldDefinition) recordDefinition.getIdField();
parameterIndex = idField.setInsertPreparedStatementValue(statement, parameterIndex, object);
statement.addBatch();
Integer batchCount = this.typeDeleteBatchCountMap.get(typePath);
if (batchCount == null) {
batchCount = 1;
this.typeDeleteBatchCountMap.put(typePath, 1);
} else {
batchCount += 1;
this.typeDeleteBatchCountMap.put(typePath, batchCount);
}
this.recordStore.addStatistic("Delete", object);
// TODO this locks code tables which prevents insert
// if (batchCount >= batchSize) {
// final String sql = getDeleteSql(recordDefinition);
// processCurrentBatch(typePath, sql, statement, typeDeleteBatchCountMap,
// getDeleteStatistics());
// }
}