001 /**
002 * Licensed to the Apache Software Foundation (ASF) under one
003 * or more contributor license agreements. See the NOTICE file
004 * distributed with this work for additional information
005 * regarding copyright ownership. The ASF licenses this file
006 * to you under the Apache License, Version 2.0 (the
007 * "License"); you may not use this file except in compliance
008 * with the License. You may obtain a copy of the License at
009 *
010 * http://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing, software
013 * distributed under the License is distributed on an "AS IS" BASIS,
014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
015 * See the License for the specific language governing permissions and
016 * limitations under the License.
017 */
018 package org.apache.oozie.executor.jpa;
019
020 import java.util.Collection;
021 import javax.persistence.EntityManager;
022 import javax.persistence.Query;
023
024 import org.apache.oozie.CoordinatorActionBean;
025 import org.apache.oozie.ErrorCode;
026 import org.apache.oozie.FaultInjection;
027 import org.apache.oozie.client.rest.JsonBean;
028 import org.apache.oozie.sla.SLARegistrationBean;
029 import org.apache.oozie.util.ParamChecker;
030
031 /**
032 * Class for updating and deleting beans in bulk
033 */
034 public class BulkUpdateDeleteJPAExecutor implements JPAExecutor<Void> {
035
036 private Collection<JsonBean> updateList;
037 private Collection<JsonBean> deleteList;
038 private boolean forRerun = true;
039
040 /**
041 * Initialize the JPAExecutor using the update and delete list of JSON beans
042 * @param deleteList
043 * @param updateList
044 */
045 public BulkUpdateDeleteJPAExecutor(Collection<JsonBean> updateList, Collection<JsonBean> deleteList,
046 boolean forRerun) {
047 this.updateList = updateList;
048 this.deleteList = deleteList;
049 this.forRerun = forRerun;
050 }
051
052 public BulkUpdateDeleteJPAExecutor() {
053 }
054
055 /**
056 * Sets the update list for JSON bean
057 *
058 * @param updateList
059 */
060 public void setUpdateList(Collection<JsonBean> updateList) {
061 this.updateList = updateList;
062 }
063
064 /**
065 * Sets the delete list for JSON bean
066 *
067 * @param deleteList
068 */
069 public void setDeleteList(Collection<JsonBean> deleteList) {
070 this.deleteList = deleteList;
071 }
072
073 /**
074 * Sets whether for RerunX command or no. Else it'd be for ChangeX
075 *
076 * @param forRerun
077 */
078 public void setForRerun(boolean forRerun) {
079 this.forRerun = forRerun;
080 }
081
082 /*
083 * (non-Javadoc)
084 *
085 * @see org.apache.oozie.executor.jpa.JPAExecutor#getName()
086 */
087 @Override
088 public String getName() {
089 return "BulkUpdateDeleteJPAExecutor";
090 }
091
092 /*
093 * (non-Javadoc)
094 *
095 * @see org.apache.oozie.executor.jpa.JPAExecutor#execute(javax.persistence.
096 * EntityManager)
097 */
098 @Override
099 public Void execute(EntityManager em) throws JPAExecutorException {
100 try {
101 if (updateList != null) {
102 for (JsonBean entity : updateList) {
103 ParamChecker.notNull(entity, "JsonBean");
104 em.merge(entity);
105 }
106 }
107 // Only used by test cases to check for rollback of transaction
108 FaultInjection.activate("org.apache.oozie.command.SkipCommitFaultInjection");
109 if (deleteList != null) {
110 for (JsonBean entity : deleteList) {
111 ParamChecker.notNull(entity, "JsonBean");
112 if (forRerun) {
113 em.remove(em.merge(entity));
114 }
115 else if (entity instanceof CoordinatorActionBean) {
116 Query g = em.createNamedQuery("DELETE_UNSCHEDULED_ACTION");
117 String coordActionId = ((CoordinatorActionBean) entity).getId();
118 g.setParameter("id", coordActionId);
119 int actionsDeleted = g.executeUpdate();
120 if (actionsDeleted == 0)
121 throw new JPAExecutorException(ErrorCode.E1022, coordActionId);
122 }
123 else {
124 em.remove(em.merge(entity));
125 }
126 }
127 }
128 return null;
129 }
130 catch (JPAExecutorException je) {
131 throw je;
132 }
133 catch (Exception e) {
134 throw new JPAExecutorException(ErrorCode.E0603, e.getMessage(), e);
135 }
136 }
137 }