| /* Copyright (c) 2001-2010, The HSQL Development Group |
| * All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions are met: |
| * |
| * Redistributions of source code must retain the above copyright notice, this |
| * list of conditions and the following disclaimer. |
| * |
| * Redistributions in binary form must reproduce the above copyright notice, |
| * this list of conditions and the following disclaimer in the documentation |
| * and/or other materials provided with the distribution. |
| * |
| * Neither the name of the HSQL Development Group nor the names of its |
| * contributors may be used to endorse or promote products derived from this |
| * software without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
| * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
| * ARE DISCLAIMED. IN NO EVENT SHALL HSQL DEVELOPMENT GROUP, HSQLDB.ORG, |
| * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, |
| * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
| * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; |
| * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND |
| * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
| * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| |
| package org.hsqldb; |
| |
| import org.hsqldb.error.Error; |
| import org.hsqldb.error.ErrorCode; |
| import org.hsqldb.lib.ArrayUtil; |
| import org.hsqldb.navigator.RowSetNavigatorData; |
| import org.hsqldb.navigator.RowSetNavigatorDataChange; |
| import org.hsqldb.persist.PersistentStore; |
| import org.hsqldb.result.Result; |
| import org.hsqldb.result.ResultConstants; |
| import org.hsqldb.types.Type; |
| |
| public class StatementResultUpdate extends StatementDML { |
| |
| int actionType; |
| Type[] types; |
| Result result; |
| |
| StatementResultUpdate() { |
| |
| super(); |
| |
| isTransactionStatement = true; |
| } |
| |
| public String describe(Session session) { |
| return ""; |
| } |
| |
| public Result execute(Session session) { |
| |
| try { |
| return getResult(session); |
| } catch (Throwable e) { |
| return Result.newErrorResult(e, null); |
| } |
| } |
| |
| Result getResult(Session session) { |
| |
| checkAccessRights(session); |
| |
| Object[] args = session.sessionContext.dynamicArguments; |
| Row row; |
| PersistentStore store = baseTable.getRowStore(session); |
| |
| switch (actionType) { |
| |
| case ResultConstants.UPDATE_CURSOR : { |
| row = getRow(session, args); |
| |
| /** |
| * @todo - in 2PL mode isDeleted() always returns false. |
| * While write lock prevents delete by other transactions, |
| * same-transaction deletes are not caught |
| */ |
| if (row == null || row.isDeleted(session, store)) { |
| throw Error.error(ErrorCode.X_24521); |
| } |
| |
| RowSetNavigatorDataChange list = |
| new RowSetNavigatorDataChange(); |
| Object[] data = |
| (Object[]) ArrayUtil.duplicateArray(row.getData()); |
| boolean[] columnCheck = baseTable.getNewColumnCheckList(); |
| |
| for (int i = 0; i < baseColumnMap.length; i++) { |
| if (types[i] == Type.SQL_ALL_TYPES) { |
| continue; |
| } |
| |
| data[baseColumnMap[i]] = args[i]; |
| columnCheck[baseColumnMap[i]] = true; |
| } |
| |
| int[] colMap = ArrayUtil.booleanArrayToIntIndexes(columnCheck); |
| |
| list.addRow(session, row, data, baseTable.getColumnTypes(), |
| colMap); |
| update(session, baseTable, list); |
| |
| break; |
| } |
| case ResultConstants.DELETE_CURSOR : { |
| row = getRow(session, args); |
| |
| if (row == null || row.isDeleted(session, store)) { |
| throw Error.error(ErrorCode.X_24521); |
| } |
| |
| RowSetNavigatorDataChange navigator = |
| new RowSetNavigatorDataChange(); |
| |
| navigator.addRow(row); |
| delete(session, baseTable, navigator); |
| |
| break; |
| } |
| case ResultConstants.INSERT_CURSOR : { |
| Object[] data = baseTable.getNewRowData(session); |
| |
| for (int i = 0; i < data.length; i++) { |
| data[baseColumnMap[i]] = args[i]; |
| } |
| |
| return insertSingleRow(session, store, data); |
| } |
| } |
| |
| return Result.updateOneResult; |
| } |
| |
| Row getRow(Session session, Object[] args) { |
| |
| int rowIdIndex = result.metaData.getColumnCount(); |
| Long rowId = (Long) args[rowIdIndex]; |
| PersistentStore store = baseTable.getRowStore(session); |
| Row row = null; |
| |
| if (rowIdIndex + 2 == result.metaData.getExtendedColumnCount()) { |
| Object[] data = |
| ((RowSetNavigatorData) result.getNavigator()).getData( |
| rowId.longValue()); |
| |
| if (data != null) { |
| row = (Row) data[rowIdIndex + 1]; |
| } |
| } else { |
| int id = (int) rowId.longValue(); |
| |
| row = (Row) store.get(id, false); |
| } |
| |
| this.result = null; |
| |
| return row; |
| } |
| |
| void setRowActionProperties(Result result, int action, Table table, |
| Type[] types, int[] columnMap) { |
| |
| this.result = result; |
| this.actionType = action; |
| this.baseTable = table; |
| this.types = types; |
| this.baseColumnMap = columnMap; |
| } |
| |
| void checkAccessRights(Session session) { |
| |
| switch (type) { |
| |
| case StatementTypes.CALL : { |
| break; |
| } |
| case StatementTypes.INSERT : { |
| session.getGrantee().checkInsert(targetTable, |
| insertCheckColumns); |
| |
| break; |
| } |
| case StatementTypes.SELECT_CURSOR : |
| break; |
| |
| case StatementTypes.DELETE_WHERE : { |
| session.getGrantee().checkDelete(targetTable); |
| |
| break; |
| } |
| case StatementTypes.UPDATE_WHERE : { |
| session.getGrantee().checkUpdate(targetTable, |
| updateCheckColumns); |
| |
| break; |
| } |
| case StatementTypes.MERGE : { |
| session.getGrantee().checkInsert(targetTable, |
| insertCheckColumns); |
| session.getGrantee().checkUpdate(targetTable, |
| updateCheckColumns); |
| |
| break; |
| } |
| } |
| } |
| } |