package fr.ifremer.adagio.synchro.service;
   
   /*
    * #%L
    * SIH-Adagio :: Synchronization
    * $Id:$
    * $HeadURL:$
    * %%
    * Copyright (C) 2012 - 2014 Ifremer
   * %%
   * This program is free software: you can redistribute it and/or modify
   * it under the terms of the GNU Affero General Public License as published by
   * the Free Software Foundation, either version 3 of the License, or
   * (at your option) any later version.
   * 
   * This program is distributed in the hope that it will be useful,
   * but WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   * GNU General Public License for more details.
   * 
   * You should have received a copy of the GNU Affero General Public License
   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
   * #L%
   */
  
  import java.util.List;
  import java.util.Map;
  
  import org.apache.commons.collections.CollectionUtils;
  import org.apache.commons.collections4.MapUtils;
  
  import com.google.common.collect.Lists;
  import com.google.common.collect.Maps;
  
  import fr.ifremer.adagio.synchro.intercept.SynchroWriteBuffer;
  
  public class SynchroPendingOperationBuffer implements SynchroWriteBuffer {
  
  	private List<List<Object>> pks;
  
  	private Map<Integer, Integer> remoteIdsMap;
  
  	private Map<Integer, Map<String, Integer>> missingRemoteIdByColumnIndex = Maps.newHashMap();
  
  	private final String tableName;
  
  	public SynchroPendingOperationBuffer(String tableName) {
  		this.tableName = tableName;
  	}
  
  	public List<List<Object>> getPks() {
  		return pks;
  	}
  
  	public void setPks(List<List<Object>> pks) {
  		this.pks = pks;
  	}
  
  	public void addPks(List<List<Object>> updatedPks) {
  		if (pks == null) {
  			pks = Lists.newArrayList();
  		}
  		pks.addAll(updatedPks);
  	}
  
  	public String getTableName() {
  		return tableName;
  	}
  
  	public Map<Integer, Integer> getRemoteIdsMap() {
  		return remoteIdsMap;
  	}
  
  	public void addRemoteIdsMap(Map<Integer, Integer> updatedRemoteIds) {
  		if (remoteIdsMap == null) {
  			remoteIdsMap = updatedRemoteIds;
  		}
  		remoteIdsMap.putAll(updatedRemoteIds);
  	}
  
  	public boolean isEmpty() {
  		return MapUtils.isEmpty(remoteIdsMap) && CollectionUtils.isEmpty(pks);
  	}
  
  	@Override
  	public void addMissingRemoteId(String tablename, int columnIndex, String pkStr, Integer remoteId) {
  		Map<String, Integer> missingRemoteIds = missingRemoteIdByColumnIndex.get(columnIndex);
  		if (missingRemoteIds == null) {
  			missingRemoteIds = Maps.newHashMap();
  			missingRemoteIdByColumnIndex.put(columnIndex, missingRemoteIds);
  		}
  		missingRemoteIds.put(pkStr, remoteId);
  	}
  }