norbert.mynemo.dataimport.scraping.input.CkMappingFile.java Source code

Introduction

Here is the source code for norbert.mynemo.dataimport.scraping.input.CkMappingFile.java

Source

/*
 * Copyright 2015 Norbert
 *
 * 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
 *
 * http://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 norbert.mynemo.dataimport.scraping.input;

import static com.google.common.base.Preconditions.checkArgument;

import java.io.File;
import java.io.IOException;
import java.util.Iterator;

import norbert.mynemo.dataimport.scraping.CkMapping;

import org.apache.commons.csv.CSVParser;
import org.apache.commons.csv.CSVRecord;

import com.google.common.collect.UnmodifiableIterator;

/**
 * This class represents a file containing mappings between CK movie ids and IMDb movie ids. The
 * format of this file is defined in the {@link CkMapping} class.
 */
public class CkMappingFile implements Iterable<CkMapping> {

    private final class MappingIterator extends UnmodifiableIterator<CkMapping> {

        private Iterator<CSVRecord> iterator = parser.iterator();

        @Override
        public boolean hasNext() {
            boolean result = iterator.hasNext();

            if (!result) {
                try {
                    parser.close();
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            }
            return result;
        }

        @Override
        public CkMapping next() {
            return CkMapping.createMapping(iterator.next());
        }
    }

    /**
     * Returns <code>true</code> if the given file can be parsed, <code>false</code> otherwise.
     */
    public static boolean canParse(String filepath) throws IOException {
        checkArgument(new File(filepath).exists(), "The file must exists.");

        try (CSVParser parser = CkMapping.createParser(filepath)) {
            for (CSVRecord record : parser) {
                return CkMapping.isValid(record);
            }
        }

        // everything seems right
        return true;
    }

    private final CSVParser parser;

    public CkMappingFile(String filepath) throws IOException {
        checkArgument(new File(filepath).exists(), "The file must exist.");

        parser = CkMapping.createParser(filepath);
    }

    @Override
    public Iterator<CkMapping> iterator() {
        return new MappingIterator();
    }
}