channellistmaker.listmaker.EPGListMaker.java Source code

Introduction

Here is the source code for channellistmaker.listmaker.EPGListMaker.java

Source

/*
 * Copyright (C) 2016 normal
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU 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 General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package channellistmaker.listmaker;

import channellistmaker.listmaker.fileseeker.FileSeeker;
import java.io.File;
import java.util.Collections;
import java.util.List;
import org.apache.commons.io.filefilter.FileFilterUtils;
import org.apache.commons.io.filefilter.IOFileFilter;
import org.w3c.dom.Document;
import java.lang.invoke.MethodHandles;
import java.nio.charset.Charset;
import java.util.HashSet;
import java.util.Set;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

/**
 * ???xml??????
 *
 * @author dosdiaopfhj
 */
public class EPGListMaker {

    private static final Log LOG;

    static {
        final Class<?> myClass = MethodHandles.lookup().lookupClass();
        LOG = LogFactory.getLog(myClass);
    }
    private final FileSeeker seeker;

    private static final IOFileFilter SUFFIX_1 = FileFilterUtils.suffixFileFilter("xml");
    private static final IOFileFilter SUFFIX_2 = FileFilterUtils.suffixFileFilter("Xml");
    private static final IOFileFilter SUFFIX_3 = FileFilterUtils.suffixFileFilter("xMl");
    private static final IOFileFilter SUFFIX_4 = FileFilterUtils.suffixFileFilter("xmL");
    private static final IOFileFilter SUFFIX_5 = FileFilterUtils.suffixFileFilter("XMl");
    private static final IOFileFilter SUFFIX_6 = FileFilterUtils.suffixFileFilter("xML");
    private static final IOFileFilter SUFFIX_7 = FileFilterUtils.suffixFileFilter("XmL");
    private static final IOFileFilter SUFFIX_8 = FileFilterUtils.suffixFileFilter("XML");

    private final Charset charset;

    /**
     * @param sourceDir ?
     * @param charset ??
     */
    public EPGListMaker(File sourceDir, Charset charset) {
        this.charset = charset;
        seeker = new FileSeeker(sourceDir,
                FileFilterUtils.or(SUFFIX_1, SUFFIX_2, SUFFIX_3, SUFFIX_4, SUFFIX_5, SUFFIX_6, SUFFIX_7, SUFFIX_8));
        seeker.setRecursive(false);
    }

    /**
     * EPG XML????????
     *
     * @return ???????Document????????
     */
    public synchronized Set<Document> seek() {
        Set<Document> EPGs = Collections.synchronizedSet(new HashSet<Document>());
        List<File> FL = this.seeker.seek();
        for (File F : FL) {
            Document d = new XMLLoader(charset).Load(F);
            if (d != null) {
                LOG.info("EPG?????? EPG FILE = " + F.toString());
                EPGs.add(d);
            } else {
                LOG.warn(
                        "EPG?????????????? EPG FILE = "
                                + F.toString());
            }
        }
        return Collections.unmodifiableSet(EPGs);
    }
}