com.glaf.activiti.util.ProcessUtils.java Source code

Introduction

Here is the source code for com.glaf.activiti.util.ProcessUtils.java

Source

/*
 * 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 com.glaf.activiti.util;

import java.io.InputStream;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

import org.activiti.engine.repository.ProcessDefinition;
import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.core.io.FileSystemResource;

import com.glaf.activiti.service.ActivitiDeployQueryService;
import com.glaf.activiti.service.ActivitiProcessQueryService;
import com.glaf.core.context.ApplicationContext;
import com.glaf.core.context.ContextFactory;
import com.glaf.core.util.FileUtils;

public class ProcessUtils {
    protected final static Log logger = LogFactory.getLog(ProcessUtils.class);

    private static ConcurrentMap<String, ProcessDefinition> cache = new ConcurrentHashMap<String, ProcessDefinition>();

    private ProcessUtils() {

    }

    public static String getFileExt(String fileName) {
        String value = "";
        int start = 0;
        int end = 0;
        if (fileName == null) {
            return null;
        }
        start = fileName.lastIndexOf(46) + 1;
        end = fileName.length();
        value = fileName.substring(start, end);
        if (fileName.lastIndexOf(46) > 0) {
            return value;
        } else {
            return "";
        }
    }

    public static String getImagePath(ProcessDefinition processDefinition) {
        String resourceName = processDefinition.getDiagramResourceName();
        if (resourceName != null) {
            String ext = getFileExt(resourceName);
            String md5_hex = DigestUtils.md5Hex(resourceName);
            String imagePath = processDefinition.getDeploymentId() + "_" + md5_hex + "." + ext;
            return imagePath;
        }
        return null;
    }

    public static byte[] getImage(String processDefinitionId) {
        byte[] bytes = null;
        ProcessDefinition processDefinition = cache.get(processDefinitionId);
        if (processDefinition == null) {
            ActivitiProcessQueryService activitiProcessQueryService = ContextFactory
                    .getBean("activitiProcessQueryService");
            processDefinition = activitiProcessQueryService.getProcessDefinition(processDefinitionId);
        }

        if (processDefinition != null) {
            String resourceName = processDefinition.getDiagramResourceName();
            if (resourceName != null) {
                String filename = ApplicationContext.getAppPath() + "/deploy/bpmn/"
                        + getImagePath(processDefinition);
                FileSystemResource fs = new FileSystemResource(filename);
                if (!fs.exists()) {
                    try {
                        ActivitiDeployQueryService activitiDeployQueryService = ContextFactory
                                .getBean("activitiDeployQueryService");
                        InputStream inputStream = activitiDeployQueryService
                                .getResourceAsStream(processDefinition.getDeploymentId(), resourceName);
                        logger.debug("save:" + filename);
                        FileUtils.save(filename, inputStream);
                    } catch (Exception ex) {
                        ex.printStackTrace();
                    }
                }
                return FileUtils.getBytes(fs.getFile());
            }
        }

        return bytes;
    }

    public static void saveProcessImageToFileSystem(ProcessDefinition processDefinition) {
        logger.debug("@deploymentId:" + processDefinition.getDeploymentId());
        String resourceName = processDefinition.getDiagramResourceName();
        if (resourceName != null) {
            ActivitiDeployQueryService activitiDeployQueryService = ContextFactory
                    .getBean("activitiDeployQueryService");
            InputStream inputStream = activitiDeployQueryService
                    .getResourceAsStream(processDefinition.getDeploymentId(), resourceName);
            logger.debug("@resourceName:" + resourceName);
            String filename = ApplicationContext.getAppPath() + "/deploy/bpmn/" + getImagePath(processDefinition);
            FileSystemResource fs = new FileSystemResource(filename);
            if (!fs.exists()) {
                try {
                    logger.debug("save:" + filename);
                    FileUtils.save(filename, inputStream);
                } catch (Exception ex) {
                    ex.printStackTrace();
                }
            }
        }
    }

}