com.ikon.util.impexp.JcrRepositoryChecker.java Source code

Java tutorial

Introduction

Here is the source code for com.ikon.util.impexp.JcrRepositoryChecker.java

Source

/**
 *  openkm, Open Document Management System (http://www.openkm.com)
 *  Copyright (c) 2006-2013  Paco Avila & Josep Llort
 *
 *  No bytes were intentionally harmed during the development of this application.
 *
 *  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 2 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, write to the Free Software Foundation, Inc.,
 *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

package com.ikon.util.impexp;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.Writer;

import javax.jcr.Node;
import javax.jcr.NodeIterator;
import javax.jcr.Session;

import org.apache.commons.io.IOUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.ikon.bean.Document;
import com.ikon.bean.Folder;
import com.ikon.bean.Mail;
import com.ikon.bean.Note;
import com.ikon.bean.Version;
import com.ikon.core.AccessDeniedException;
import com.ikon.core.Config;
import com.ikon.core.DatabaseException;
import com.ikon.core.PathNotFoundException;
import com.ikon.core.RepositoryException;
import com.ikon.module.DocumentModule;
import com.ikon.module.ModuleManager;
import com.ikon.module.jcr.stuff.JCRUtils;
import com.ikon.module.jcr.stuff.JcrSessionManager;

public class JcrRepositoryChecker {
    private static Logger log = LoggerFactory.getLogger(JcrRepositoryChecker.class);

    private JcrRepositoryChecker() {
    }

    /**
     * Performs a recursive repository document check
     */
    public static ImpExpStats checkDocuments(String token, String fldPath, boolean versions, Writer out,
            InfoDecorator deco) throws PathNotFoundException, AccessDeniedException, RepositoryException,
            IOException, DatabaseException {
        log.debug("checkDocuments({}, {}, {}, {}, {})", new Object[] { token, fldPath, versions, out, deco });
        Session session = null;
        ImpExpStats stats = new ImpExpStats();

        try {
            if (token == null) {
                session = JCRUtils.getSession();
            } else {
                session = JcrSessionManager.getInstance().get(token);
            }

            Node baseNode = session.getRootNode().getNode(fldPath.substring(1));
            stats = checkDocumentsHelper(token, baseNode, versions, out, deco);
        } catch (PathNotFoundException e) {
            log.error(e.getMessage(), e);
            stats.setOk(false);
            throw e;
        } catch (AccessDeniedException e) {
            log.error(e.getMessage(), e);
            stats.setOk(false);
            throw e;
        } catch (FileNotFoundException e) {
            log.error(e.getMessage(), e);
            stats.setOk(false);
            throw e;
        } catch (RepositoryException e) {
            log.error(e.getMessage(), e);
            stats.setOk(false);
            throw e;
        } catch (IOException e) {
            log.error(e.getMessage(), e);
            stats.setOk(false);
            throw e;
        } catch (DatabaseException e) {
            log.error(e.getMessage(), e);
            stats.setOk(false);
            throw e;
        } catch (javax.jcr.RepositoryException e) {
            log.error(e.getMessage(), e);
            stats.setOk(false);
            throw new RepositoryException(e.getMessage(), e);
        } finally {
            if (token == null)
                JCRUtils.logout(session);
        }

        log.debug("checkDocuments: {}", stats);
        return stats;
    }

    /**
     * Performs a recursive repository document check
     */
    private static ImpExpStats checkDocumentsHelper(String token, Node baseNode, boolean versions, Writer out,
            InfoDecorator deco)
            throws FileNotFoundException, PathNotFoundException, AccessDeniedException, RepositoryException,
            IOException, DatabaseException, javax.jcr.PathNotFoundException, javax.jcr.RepositoryException {
        log.debug("checkDocumentsHelper({}, {}, {}, {}, {})",
                new Object[] { token, baseNode, versions, out, deco });
        ImpExpStats stats = new ImpExpStats();

        for (NodeIterator ni = baseNode.getNodes(); ni.hasNext();) {
            Node child = ni.nextNode();

            if (child.isNodeType(Document.TYPE)) {
                ImpExpStats tmp = readDocument(token, child.getPath(), versions, out, deco);
                stats.setDocuments(stats.getDocuments() + tmp.getDocuments());
                stats.setFolders(stats.getFolders() + tmp.getFolders());
                stats.setSize(stats.getSize() + tmp.getSize());
                stats.setOk(stats.isOk() && tmp.isOk());
            } else if (child.isNodeType(Folder.TYPE)) {
                ImpExpStats tmp = readFolder(token, child, versions, out, deco);
                stats.setDocuments(stats.getDocuments() + tmp.getDocuments());
                stats.setFolders(stats.getFolders() + tmp.getFolders());
                stats.setSize(stats.getSize() + tmp.getSize());
                stats.setOk(stats.isOk() && tmp.isOk());
            } else if (child.isNodeType(Mail.TYPE)) {
                ImpExpStats tmp = readFolder(token, child, versions, out, deco);
                stats.setDocuments(stats.getDocuments() + tmp.getDocuments());
                stats.setFolders(stats.getFolders() + tmp.getFolders());
                stats.setSize(stats.getSize() + tmp.getSize());
                stats.setOk(stats.isOk() && tmp.isOk());
            } else if (child.isNodeType(Note.LIST_TYPE)) {
                // Note nodes has no check procedure
            } else {
                log.error("Unknown node type: {} ({})", child.getPrimaryNodeType().getName(), child.getPath());
                stats.setOk(false);
                out.write(deco.print(child.getPath(), 0,
                        "Unknown node type: " + child.getPrimaryNodeType().getName()));
                out.flush();
            }
        }

        log.debug("checkDocumentsHelper: {}", stats);
        return stats;
    }

    /**
     * Read document contents.
     */
    private static ImpExpStats readDocument(String token, String docPath, boolean versions, Writer out,
            InfoDecorator deco) throws PathNotFoundException, AccessDeniedException, RepositoryException,
            DatabaseException, IOException {
        log.debug("readDocument({})", docPath);
        DocumentModule dm = ModuleManager.getDocumentModule();
        File fsPath = new File(Config.NULL_DEVICE);
        ImpExpStats stats = new ImpExpStats();
        Document doc = dm.getProperties(token, docPath);

        try {
            FileOutputStream fos = new FileOutputStream(fsPath);
            InputStream is = dm.getContent(token, docPath, false);
            IOUtils.copy(is, fos);
            is.close();

            if (versions) { // Check version history
                for (Version ver : dm.getVersionHistory(token, docPath)) {
                    is = dm.getContentByVersion(token, docPath, ver.getName());
                    IOUtils.copy(is, fos);
                    IOUtils.closeQuietly(is);
                }
            }

            fos.close();
            out.write(deco.print(docPath, doc.getActualVersion().getSize(), null));
            out.flush();

            // Stats
            stats.setSize(stats.getSize() + doc.getActualVersion().getSize());
            stats.setDocuments(stats.getDocuments() + 1);
        } catch (RepositoryException e) {
            log.error(e.getMessage());
            stats.setOk(false);
            out.write(deco.print(docPath, doc.getActualVersion().getSize(), e.getMessage()));
            out.flush();
        }

        return stats;
    }

    /**
     * Read folder contents. 
     */
    private static ImpExpStats readFolder(String token, Node baseNode, boolean versions, Writer out,
            InfoDecorator deco)
            throws FileNotFoundException, PathNotFoundException, AccessDeniedException, RepositoryException,
            IOException, DatabaseException, javax.jcr.PathNotFoundException, javax.jcr.RepositoryException {
        log.debug("readFolder({})", baseNode.getPath());
        ImpExpStats stats = checkDocumentsHelper(token, baseNode, versions, out, deco);

        // Stats
        stats.setFolders(stats.getFolders() + 1);

        return stats;
    }
}