content: /*****************************************************************
   *   Licensed to the Apache Software Foundation (ASF) under one
   *  or more contributor license agreements.  See the NOTICE file
   *  distributed with this work for additional information
   *  regarding copyright ownership.  The ASF licenses this file
   *  to you 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
   *
   *    https://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 org.apache.cayenne.access.translator.batch;
  
  import org.apache.cayenne.access.sqlbuilder.SQLBuilder;
  import org.apache.cayenne.access.sqlbuilder.UpdateBuilder;
  import org.apache.cayenne.access.translator.DbAttributeBinding;
  import org.apache.cayenne.access.types.ExtendedType;
  import org.apache.cayenne.dba.DbAdapter;
  import org.apache.cayenne.map.DbAttribute;
  import org.apache.cayenne.query.BatchQueryRow;
  import org.apache.cayenne.query.UpdateBatchQuery;
  
  /**
   * @since 4.2
   */
  public class UpdateBatchTranslator extends BaseBatchTranslator<UpdateBatchQuery> implements BatchTranslator {
  
      public UpdateBatchTranslator(UpdateBatchQuery query, DbAdapter adapter) {
          super(query, adapter);
      }
  
      @Override
      public String getSql() {
          UpdateBatchQuery query = context.getQuery();
  
          UpdateBuilder updateBuilder = SQLBuilder.update(context.getRootDbEntity());
          for (DbAttribute attr : query.getUpdatedAttributes()) {
              updateBuilder.set(SQLBuilder
                      .column(attr.getName()).attribute(attr)
                      .eq(SQLBuilder.value(1).attribute(attr))
              );
          }
          updateBuilder.where(buildQualifier(query.getQualifierAttributes()));
  
          return doTranslate(updateBuilder);
      }
  
      @Override
      protected boolean isNullAttribute(DbAttribute attribute) {
          return context.getQuery().isNull(attribute);
      }
  
      @Override
      public DbAttributeBinding[] updateBindings(BatchQueryRow row) {
          UpdateBatchQuery updateBatch = context.getQuery();
  
          int i = 0;
          int j = 0;
          for(; i < updateBatch.getUpdatedAttributes().size(); i++) {
              Object value = row.getValue(i);
              ExtendedType<?> extendedType = value == null
                  ? context.getAdapter().getExtendedTypes().getDefaultType()
                  : context.getAdapter().getExtendedTypes().getRegisteredType(value.getClass());
              bindings[j].include(++j, value, extendedType);
          }
  
          for(DbAttribute attribute : updateBatch.getQualifierAttributes()) {
              if(updateBatch.isNull(attribute)) {
                  i++;
                  continue;
              }
              Object value = row.getValue(i);
              ExtendedType<?> extendedType = context.getAdapter().getExtendedTypes().getRegisteredType(value.getClass());
              bindings[j].include(++j, value, extendedType);
              i++;
          }
          return bindings;
      }
  }
  