org.jumpmind.symmetric.model.OutgoingBatches Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of symmetric-ds Show documentation
SymmetricDS is an open source database synchronization solution. It is platform-independent, web-enabled, and database-agnostic. SymmetricDS was first built to replicate changes between 'retail store' databases and ad centralized 'corporate' database.
The newest version!
/*
 * Licensed to JumpMind Inc under one or more contributor 
 * license agreements.  See the NOTICE file distributed
 * with this work for additional information regarding 
 * copyright ownership.  JumpMind Inc licenses this file
 * to you under the GNU Lesser General Public License (the
 * "License"); you may not use this file except in compliance
 * with the License. 
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, see           
 * .
 * 
 * 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.jumpmind.symmetric.model;

import java.util.ArrayList;
import java.util.Calendar;
import java.util.Collections;
import java.util.Comparator;
import java.util.Date;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Set;

import org.jumpmind.symmetric.model.OutgoingBatch.Status;

/**
 * 
 */
public class OutgoingBatches {

    List batches = new ArrayList();
    Set activeChannels = new HashSet();
    Set activeChannelIds = new HashSet();

    public OutgoingBatches(List batches) {
        this.batches = batches;
    }

    public OutgoingBatches() {
    }

    public boolean containsBatches() {
        return batches != null && batches.size() > 0;    
    }
    
    public Set getActiveChannels() {
        return activeChannels;
    }

    public void addActiveChannel(NodeChannel nodeChannel) {
        activeChannels.add(nodeChannel);
        activeChannelIds.add(nodeChannel.getChannelId());
    }

    public void setActiveChannels(Set activeChannels) {
        this.activeChannels = activeChannels;
        activeChannelIds = new HashSet();
        for (NodeChannel nodeChannel : activeChannels) {
            activeChannelIds.add(nodeChannel.getChannelId());
        }
    }

    public List getBatches() {
        return batches;
    }

    public void setBatches(List batches) {
        this.batches = batches;
    }

    /**
     * Removes all batches associated with the provided channel from this
     * object.
     * 
     * @param channel
     *            - channel for which corresponding batches are removed
     * @return A list of the batches removed
     */
    public List filterBatchesForChannel(Channel channel) {
        List filtered = getBatchesForChannel(channel);
        batches.removeAll(filtered);
        return filtered;
    }
    
    public int countBatches(boolean includeOnlyErrors) {
        int count = 0;
        if (batches != null) {
            for (OutgoingBatch batch : batches) {
                if (includeOnlyErrors && batch.getStatus() == OutgoingBatch.Status.ER) {
                    count++;
                } else {
                    count++;
                }
            }
        }
        return count;
    }

    public List filterBatchesForChannel(String channelId) {
        List filtered = getBatchesForChannel(channelId);
        batches.removeAll(filtered);
        return filtered;
    }

    public List filterBatchesForChannels(Set channels) {
        List filtered = getBatchesForChannels(channels);
        batches.removeAll(filtered);
        return filtered;
    }
    
    public void removeNonLoadBatches() {
        for (Iterator iterator = batches.iterator(); iterator.hasNext();) {
            OutgoingBatch b = iterator.next();
            if (!b.isLoadFlag()) {
                iterator.remove();
            }
        }
    }
    
    public boolean containsLoadBatches() {
        for (OutgoingBatch b : batches) {
            if (b.isLoadFlag()) {
                return true;
            }
        }
        return false;
    }
    
    public boolean containsBatchesInError() {
        for (OutgoingBatch b : batches) {
            if (b.getStatus() == Status.ER) {
                return true;
            }
        }
        return false;
    }   

    public List getBatchesForChannel(Channel channel) {
        List batchList = new ArrayList();
        if (channel != null) {
            batchList = getBatchesForChannel(channel.getChannelId());
        }
        return batchList;
    }

    public List getBatchesForChannel(String channelId) {
        List batchList = new ArrayList();
        if (channelId != null) {
            for (OutgoingBatch batch : batches) {
                if (channelId.equals(batch.getChannelId())) {
                    batchList.add(batch);
                }
            }
        }
        return batchList;
    }

    public List getBatchesForChannels(Set channelIds) {
        List batchList = new ArrayList();
        if (channelIds != null) {
            for (OutgoingBatch batch : batches) {
                if (channelIds.contains(batch.getChannelId())) {
                    batchList.add(batch);
                }
            }
        }
        return batchList;
    }

    public List getBatchesForChannelWindows(Node targetNode, NodeChannel channel,
            List windows) {
        List keeping = new ArrayList();

        if (windows != null) {
            if (batches != null && batches.size() > 0) {
                if (channel.isEnabled() && inTimeWindow(windows, targetNode.getTimezoneOffset())) {
                    int max = channel.getMaxBatchToSend();
                    int count = 0;
                    for (OutgoingBatch outgoingBatch : batches) {
                        if (channel.getChannelId().equals(outgoingBatch.getChannelId()) && count < max) {
                            keeping.add(outgoingBatch);
                            count++;
                        }
                    }
                }
            }
        }
        return keeping;
    }

    /**
     * If {@link NodeGroupChannelWindow}s are defined for this channel, then
     * check to see if the time (according to the offset passed in) is within on
     * of the configured windows.
     */
    public boolean inTimeWindow(List windows, String timezoneOffset) {
        if (windows != null && windows.size() > 0) {
            for (NodeGroupChannelWindow window : windows) {
                if (window.inTimeWindow(timezoneOffset)) {
                    return true;
                }
            }
            return false;
        } else {
            return true;
        }

    }

    /**
     * Removes all batches that are not associated with an 'activeChannel'.
     * 
     * @return List of batches that were filtered
     */

    public List filterBatchesForInactiveChannels() {
        List filtered = new ArrayList();

        for (OutgoingBatch batch : batches) {
            if (!activeChannelIds.contains(batch.getChannelId())) {
                filtered.add(batch);
            }
        }

        batches.removeAll(filtered);
        return filtered;
    }

    public void sortChannels(List channels) {

        final HashMap errorChannels = new HashMap();
        for (OutgoingBatch batch : batches) {
            if (batch.getStatus().equals(OutgoingBatch.Status.ER)) {
                errorChannels.put(batch.getChannelId(), batch.getLastUpdatedTime());
            }
        }

        Collections.sort(channels, new Comparator() {
            public int compare(NodeChannel b1, NodeChannel b2) {
                boolean isError1 = errorChannels.containsKey(b1.getChannelId());
                boolean isError2 = errorChannels.containsKey(b2.getChannelId());
                if (!isError1 && !isError2) {
                    return b1.getProcessingOrder() < b2.getProcessingOrder() ? -1 : 1;
                } else if (isError1 && isError2) {
                    return errorChannels.get(b1.getChannelId()).compareTo(errorChannels.get(b2.getChannelId()));
                } else if (!isError1 && isError2) {
                    return -1;
                } else {
                    return 1;
                }
            }
        });

        for (NodeChannel nodeChannel : channels) {
            long extractPeriodMillis = nodeChannel.getExtractPeriodMillis();
            Date lastExtractedTime = nodeChannel.getLastExtractedTime();

            if ((extractPeriodMillis < 1) || (lastExtractedTime == null)
                    || (Calendar.getInstance().getTimeInMillis() - lastExtractedTime.getTime() >= extractPeriodMillis)) {
                addActiveChannel(nodeChannel);
            }
        }

        filterBatchesForInactiveChannels();

    }

}