net.ucanaccess.commands.UpdateCommand Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of ucanaccess Show documentation
Show all versions of ucanaccess Show documentation
An open source pure Java JDBC Driver implementation which allows Java developers and JDBC client programs (e.g., Open Office, Libre Office, Squirrel SQL) to read/write Microsoft Access databases.
/*
Copyright (c) 2012 Marco Amadei.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package net.ucanaccess.commands;
import java.io.IOException;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;
import net.ucanaccess.complex.Attachment;
import net.ucanaccess.complex.SingleValue;
import net.ucanaccess.complex.Version;
import net.ucanaccess.converters.Persist2Jet;
import net.ucanaccess.jdbc.UcanaccessSQLException;
import com.healthmarketscience.jackcess.Column;
import com.healthmarketscience.jackcess.Cursor;
import com.healthmarketscience.jackcess.DataType;
import com.healthmarketscience.jackcess.Table;
import com.healthmarketscience.jackcess.Table.ColumnOrder;
import com.healthmarketscience.jackcess.complex.ComplexValueForeignKey;
public class UpdateCommand extends AbstractCursorCommand {
private List blobColumns;
private String execId;
private IndexSelector indexSelector;
private Object[] modifiedRow;
private Map rowPattern;
private Table table;
private List tableColumns;
private boolean isRollbacking;
public UpdateCommand(Table _table, Map _map, Object[] _modifiedRow, String _execId) {
this.tableColumns = _table.getColumns();
this.indexSelector = new IndexSelector(_table);
this.rowPattern = _map;
this.modifiedRow = _modifiedRow;
this.execId = _execId;
checkBlob(_modifiedRow);
this.table = _table;
}
private void checkBlob(Object[] newRow2) {
for (int i = 0; i < newRow2.length; i++) {
if (newRow2[i] instanceof org.hsqldb.types.BlobData) {
if (blobColumns == null) {
blobColumns = new ArrayList();
}
blobColumns.add(tableColumns.get(i));
}
}
}
@Override
public String getExecId() {
return execId;
}
@Override
public IndexSelector getIndexSelector() {
return indexSelector;
}
@Override
public Map getRowPattern() {
return rowPattern;
}
@Override
public String getTableName() {
return this.table.getName();
}
@Override
public TYPES getType() {
return TYPES.UPDATE;
}
@Override
public IFeedbackAction persist() throws SQLException {
try {
Cursor cur = indexSelector.getCursor();
if (cur.findNextRow(rowPattern)) {
if (this.blobColumns != null) {
for (Column col : this.blobColumns) {
Object val = cur.getCurrentRowValue(col);
modifiedRow[tableColumns.indexOf(col)] = val;
}
}
updateComplex(cur);
persist(cur);
}
} catch (IOException e) {
throw new UcanaccessSQLException(e);
}
return new BlobAction(table, modifiedRow);
}
@Override
public IFeedbackAction persistCurrentRow(Cursor cur) throws IOException, SQLException {
if (this.blobColumns != null) {
for (Column col : this.blobColumns) {
Object val = cur.getCurrentRowValue(col);
modifiedRow[tableColumns.indexOf(col)] = val;
}
}
updateComplex(cur);
persist(cur);
return new BlobAction(table, modifiedRow);
}
private void updateComplex(Cursor cur) throws IOException {
int j = 0;
for (Column cl : this.tableColumns) {
if (cl.getType() == DataType.COMPLEX_TYPE) {
ComplexValueForeignKey rowFk = (ComplexValueForeignKey) cl.getRowValue(cur.getCurrentRow());
if (modifiedRow[j] instanceof Attachment[]) {
rowFk.deleteAllValues();
Attachment[] atcs = (Attachment[]) modifiedRow[j];
for (Attachment atc : atcs) {
rowFk.addAttachment(atc.getUrl(), atc.getName(), atc.getType(), atc.getData(),
atc.getTimeStamp(), atc.getFlags());
}
} else if (modifiedRow[j] instanceof SingleValue[]) {
rowFk.deleteAllValues();
SingleValue[] vs = (SingleValue[]) modifiedRow[j];
for (SingleValue v : vs) {
rowFk.addMultiValue(v.getValue());
}
} else if (modifiedRow[j] instanceof Version[]) {
Version[] vs = (Version[]) modifiedRow[j];
Version v = vs[0];
List oldV = rowFk.getVersions();
String vn = v.getValue();
String vo = oldV.size() > 0 ? oldV.get(0).getValue() : null;
Date upTime = isRollbacking ? new Date() : v.getModifiedDate();
if ((vn != null && vo == null) || (vo != null && vn == null)
|| (vo != null && vn != null && !vo.equals(vn))) {
rowFk.addVersion(vn, upTime);
}
}
}
j++;
}
}
@Override
public IFeedbackAction rollback() throws SQLException {
Persist2Jet p2a = new Persist2Jet();
UpdateCommand urev = new UpdateCommand(this.table, p2a.getRowPattern(this.modifiedRow, this.table),
p2a.getValues(this.getRowPattern(), this.table), this.execId);
urev.isRollbacking = true;
return urev.persist();
}
private void persist(Cursor cur) throws IOException, SQLException {
Object[] mr = this.modifiedRow;
if (table.getDatabase().getColumnOrder().equals(ColumnOrder.DISPLAY)) {
Object[] newRowReorded = new Object[this.modifiedRow.length];
int j = 0;
for (Column cli : table.getColumns()) {
newRowReorded[cli.getColumnIndex()] = this.modifiedRow[j];
j++;
}
mr = newRowReorded;
}
cur.updateCurrentRow(mr);
}
}