com.qwazr.crawler.common.CrawlerServiceImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of qwazr-crawlers Show documentation
Web and File Crawlers for QWAZR
The newest version!
/*
 * Copyright 2015-2020 Emmanuel Keller / QWAZR
 * 
 * Licensed 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 com.qwazr.crawler.common;

import com.qwazr.server.AbstractServiceImpl;
import com.qwazr.server.ServerException;
import java.util.LinkedHashMap;
import java.util.function.IntConsumer;
import java.util.logging.Logger;
import javax.ws.rs.NotFoundException;

public abstract class CrawlerServiceImpl<
        SESSION extends CrawlSessionBase,
        THREAD extends CrawlThread,
        MANAGER extends CrawlManager,
        DEFINITION extends CrawlDefinition,
        STATUS extends CrawlSessionStatus,
        ITEM extends CrawlItem
        > extends AbstractServiceImpl implements CrawlerServiceInterface {

    protected final Logger logger;

    protected final MANAGER crawlManager;

    protected CrawlerServiceImpl(Logger logger, MANAGER crawlManager) {
        this.logger = logger;
        this.crawlManager = crawlManager;
    }

    @Override
    public LinkedHashMap getSessions(final String wildcardPattern,
                                                     final Integer start,
                                                     final Integer rows,
                                                     final IntConsumer totalConsumer) {
        try {
            return crawlManager.getSessions(wildcardPattern, start == null ? 0 : start, rows == null ? 10 : rows, totalConsumer);
        } catch (Exception e) {
            throw ServerException.getJsonException(logger, e);
        }
    }

    @Override
    public STATUS getSessionStatus(final String sessionName) {
        try {
            final STATUS status = crawlManager.getSessionStatus(sessionName);
            if (status != null)
                return status;
            throw new NotFoundException("Session not found");
        } catch (Exception e) {
            throw ServerException.getJsonException(logger, e);
        }
    }

    @Override
    public DEFINITION getSessionDefinition(final String sessionName) {
        try {
            final DEFINITION definition = crawlManager.getSessionDefinition(sessionName);
            if (definition != null)
                return definition;
            throw new NotFoundException("Definition not found");
        } catch (Exception e) {
            throw ServerException.getJsonException(logger, e);
        }
    }

    @Override
    public void stopSession(final String sessionName, final String reason) {
        try {
            crawlManager.abortSession(sessionName, reason);
        } catch (Exception e) {
            throw ServerException.getTextException(logger, e);
        }
    }

    @Override
    public void removeSession(final String sessionName) {
        try {
            crawlManager.removeSession(sessionName);
        } catch (Exception e) {
            throw ServerException.getTextException(logger, e);
        }
    }

    @Override
    public STATUS runSession(final String sessionName, final DEFINITION crawlDefinition) {
        try {
            return crawlManager.runSession(sessionName, crawlDefinition);
        } catch (Exception e) {
            throw ServerException.getJsonException(logger, e);
        }
    }

}