com.smartystreets.api.us_zipcode.Batch Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of smartystreets-java-sdk Show documentation

A library to help Java developers easily access the SmartyStreets APIs.

There is a newer version: 3.18.3

Show newest version

package com.smartystreets.api.us_zipcode;

import com.smartystreets.api.exceptions.BatchFullException;

import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Vector;

/**
 * This class contains a collection of up to 100 lookups to be sent to the SmartyStreets US ZIP Code API

 *     all at once. This is more efficient than sending them one at a time.
 */
public class Batch {
    public static final int MAX_BATCH_SIZE = 100;
    private Map namedLookups;
    private Vector allLookups;

    public Batch() {
        this.namedLookups = new LinkedHashMap<>();
        this.allLookups = new Vector<>();
    }

    public void add(Lookup lookup) throws BatchFullException {
        if (this.isFull())
            throw new BatchFullException("Batch size cannot exceed " + MAX_BATCH_SIZE);

        String key = lookup.getInputId();

        if (key != null)
            this.namedLookups.put(key, lookup);

        this.allLookups.add(lookup);
    }

    /**
     * Clears the lookups stored in the batch so it can be used again.

     *     This helps avoid the overhead of building a new Batch object for each group of lookups.
     */
    public void clear() {
        this.namedLookups.clear();
        this.allLookups.clear();
    }

    //region [ Helpers ]

    /**
     * @return The number of lookups currently in this batch.
     */
    public int size() {
        return this.allLookups.size();
    }

    public boolean isFull() {
        return (this.size() >= MAX_BATCH_SIZE);
    }

    public Iterator iterator() {
        return this.allLookups.iterator();
    }

    //endregion

    //region [ Getters ]

    public Map getNamedLookups() {
        return this.namedLookups;
    }

    public Vector getAllLookups() {
        return this.allLookups;
    }

    public Lookup get(String inputId) {
        return this.namedLookups.get(inputId);
    }

    public Lookup get(int inputIndex) {
        return this.allLookups.get(inputIndex);
    }

    //endregion
}