org.codelibs.fess.crawler.service.impl.DBDataServiceImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fess-crawler-db Show documentation
The newest version!
/*
 * Copyright 2012-2016 CodeLibs Project and the Others.
 *
 * 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 org.codelibs.fess.crawler.service.impl;

import java.util.ArrayList;
import java.util.List;

import javax.annotation.Resource;

import org.codelibs.fess.crawler.Constants;
import org.codelibs.fess.crawler.db.exbhv.AccessResultBhv;
import org.codelibs.fess.crawler.db.exbhv.AccessResultDataBhv;
import org.codelibs.fess.crawler.db.exentity.AccessResult;
import org.codelibs.fess.crawler.db.exentity.AccessResultData;
import org.codelibs.fess.crawler.dbflute.cbean.result.ListResultBean;
import org.codelibs.fess.crawler.exception.CrawlerSystemException;
import org.codelibs.fess.crawler.service.DataService;
import org.codelibs.fess.crawler.util.AccessResultCallback;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * @author shinsuke
 *
 */
public class DBDataServiceImpl implements DataService {
    private static final Logger logger = LoggerFactory
            .getLogger(DBDataServiceImpl.class);

    @Resource
    protected AccessResultBhv accessResultBhv;

    @Resource
    protected AccessResultDataBhv accessResultDataBhv;

    /*
     * (non-Javadoc)
     *
     * @see org.codelibs.fess.crawler.service.DataService#store(org.codelibs.fess.crawler.entity.AccessResult)
     */
    @Override
    public void store(final AccessResult accessResult) {
        if (accessResult == null) {
            throw new CrawlerSystemException("AccessResult is null.");
        }

        accessResultBhv
                .insert(accessResult);

        AccessResultData accessResultData = accessResult.getAccessResultData();
        if (accessResultData == null) {
            accessResultData = new AccessResultData();
            accessResultData.setTransformerName(Constants.NO_TRANSFORMER);
        }
        accessResultData.setId(accessResult.getId());
        accessResultDataBhv
                .insert(accessResultData);
    }

    /*
     * (non-Javadoc)
     *
     * @see org.codelibs.fess.crawler.service.DataService#getCount(java.lang.String)
     */
    @Override
    public int getCount(final String sessionId) {
        return accessResultBhv.selectCount(cb -> {
            cb.query().setSessionId_Equal(sessionId);
        });
    }

    /*
     * (non-Javadoc)
     *
     * @see org.codelibs.fess.crawler.service.DataService#delete(java.lang.String)
     */
    @Override
    public void delete(final String sessionId) {
        int count = accessResultDataBhv.deleteBySessionId(sessionId);

        if (logger.isDebugEnabled()) {
            logger.debug("Deleted accessResultData: " + count);
        }

        count = accessResultBhv.deleteBySessionId(sessionId);

        if (logger.isDebugEnabled()) {
            logger.debug("Deleted accessResult: " + count);
        }
    }

    /*
     * (non-Javadoc)
     *
     * @see org.codelibs.fess.crawler.service.DataService#deleteAll()
     */
    @Override
    public void deleteAll() {
        int count = accessResultDataBhv.deleteAll();

        if (logger.isDebugEnabled()) {
            logger.debug("Deleted accessResultData: " + count);
        }

        count = accessResultBhv.deleteAll();

        if (logger.isDebugEnabled()) {
            logger.debug("Deleted accessResult: " + count);
        }
    }

    /*
     * (non-Javadoc)
     *
     * @see org.codelibs.fess.crawler.service.DataService#getAccessResult(java.lang.String, java.lang.String)
     */
    @Override
    public AccessResult getAccessResult(final String sessionId, final String url) {
        final ListResultBean list = accessResultBhv
                .selectList(cb -> {
                    cb.setupSelect_AccessResultDataAsOne();
                    cb.query().setSessionId_Equal(sessionId);
                    cb.query().setUrl_Equal(url);
                });
        if (list.isEmpty()) {
            return null;
        }
        return list.get(0);
    }

    /*
     * (non-Javadoc)
     *
     * @see org.codelibs.fess.crawler.service.DataService#getAccessResultList(java.lang.String, boolean)
     */
    @Override
    public List getAccessResultList(final String url,
            final boolean hasData) {
        final List list = accessResultBhv
                .selectList(cb -> {
                    if (hasData) {
                        cb.setupSelect_AccessResultDataAsOne();
                    }
                    cb.query().setUrl_Equal(url);
                    cb.query().addOrderBy_CreateTime_Desc();
                });

        final List accessResultList = new ArrayList<>();
        accessResultList.addAll(list);
        return accessResultList;
    }

    /*
     * (non-Javadoc)
     *
     * @see org.codelibs.fess.crawler.service.DataService#iterate(java.lang.String, org.codelibs.fess.crawler.util.AccessResultCallback)
     */
    @Override
    public void iterate(final String sessionId,
            final AccessResultCallback accessResultCallback) {
        accessResultBhv.selectCursor(cb -> {
            cb.setupSelect_AccessResultDataAsOne();
            cb.query().setSessionId_Equal(sessionId);
            cb.query().addOrderBy_CreateTime_Asc();
        }, entity -> accessResultCallback.iterate(entity));
    }

    /*
     * (non-Javadoc)
     *
     * @see org.codelibs.fess.crawler.service.DataService#update(org.codelibs.fess.crawler.entity.AccessResult)
     */
    @Override
    public void update(final AccessResult accessResult) {
        accessResultBhv
                .update(accessResult);

        if (accessResult.getAccessResultData() != null) {
            accessResultDataBhv
                    .update(accessResult
                            .getAccessResultData());
        }
    }

    /*
     * (non-Javadoc)
     *
     * @see org.codelibs.fess.crawler.service.DataService#update(java.util.List)
     */
    @Override
    public void update(final List accessResultList) {
        final List arList = new ArrayList<>();
        final List ardList = new ArrayList<>();

        for (final AccessResult accessResult : accessResultList) {
            arList.add(accessResult);
            if (accessResult.getAccessResultData() != null) {
                ardList.add(accessResult
                        .getAccessResultData());
            }
        }
        accessResultBhv.batchUpdate(arList);
        if (!ardList.isEmpty()) {
            accessResultDataBhv.batchUpdate(ardList);
        }

        for (final AccessResult accessResult : accessResultList) {
            update(accessResult);
        }
    }
}