com.glaf.activiti.tasklistener.PrepareTaskAssigneeListener.java Source code

Introduction

Here is the source code for com.glaf.activiti.tasklistener.PrepareTaskAssigneeListener.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.tasklistener;

import java.util.Collection;
import java.util.List;
import java.util.Map;

import org.activiti.engine.delegate.DelegateExecution;
import org.activiti.engine.delegate.DelegateTask;
import org.activiti.engine.delegate.Expression;
import org.activiti.engine.delegate.TaskListener;
import org.activiti.engine.impl.context.Context;
import org.activiti.engine.impl.interceptor.CommandContext;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.glaf.core.dao.MyBatisEntityDAO;

public class PrepareTaskAssigneeListener implements TaskListener {
    private static final long serialVersionUID = 1L;

    protected final static Log logger = LogFactory.getLog(PrepareTaskAssigneeListener.class);

    protected Expression statementId;

    protected Expression roleId;

    protected Expression outputAssigneeVar;

    protected Expression outputUsersVar;

    protected Expression outputGroupsVar;

    @Override
    public void notify(DelegateTask delegateTask) {
        logger.debug("----------------------------------------------------");
        logger.debug("---------------PrepareTaskAssigneeListener----------");
        logger.debug("----------------------------------------------------");

        CommandContext commandContext = Context.getCommandContext();
        DelegateExecution execution = delegateTask.getExecution();
        logger.debug("dbSqlsession:" + commandContext.getDbSqlSession());
        logger.debug("sqlsession:" + commandContext.getDbSqlSession().getSqlSession());

        if (execution != null) {
            Map<String, Object> paramMap = new java.util.HashMap<String, Object>();
            paramMap.putAll(execution.getVariables());
            String statement = null;
            if (statementId != null) {
                statement = statementId.getExpressionText();
                logger.debug("statementId:" + statement);
            }
            if (roleId != null) {
                logger.debug("roleId:" + roleId.getExpressionText());
                paramMap.put("roleId", roleId.getExpressionText());
            }

            if (StringUtils.isNotEmpty(statement)) {
                MyBatisEntityDAO entityDAO = new MyBatisEntityDAO(commandContext.getDbSqlSession().getSqlSession());
                List<?> list = entityDAO.getList(statement, paramMap);
                if (list != null && !list.isEmpty()) {
                    Collection<String> users = new java.util.HashSet<String>();
                    Collection<String> groups = new java.util.HashSet<String>();
                    for (Object object : list) {
                        if (object instanceof org.activiti.engine.identity.User) {
                            String actorId = ((org.activiti.engine.identity.User) object).getId();
                            if (!users.contains(actorId)) {
                                users.add(actorId);
                            }
                        } else if (object instanceof org.activiti.engine.identity.Group) {
                            String groupId = ((org.activiti.engine.identity.Group) object).getId();
                            if (!groups.contains(groupId)) {
                                groups.add(groupId);
                            }
                        } else if (object instanceof com.glaf.core.identity.User) {
                            String actorId = ((com.glaf.core.identity.User) object).getActorId();
                            if (!users.contains(actorId)) {
                                users.add(actorId);
                            }
                        } else if (object instanceof com.glaf.core.identity.Role) {
                            String groupId = String.valueOf(((com.glaf.core.identity.Role) object).getRoleId());
                            if (!groups.contains(groupId)) {
                                groups.add(groupId);
                            }
                        }
                    }

                    logger.debug("users:" + users);
                    logger.debug("groups:" + groups);

                    if (users.size() > 0) {
                        if (outputAssigneeVar != null) {
                            Object name = outputAssigneeVar.getValue(delegateTask.getExecution());
                            if (name != null) {
                                execution.setVariable(name.toString(), users.iterator().next());
                            }
                        }

                        if (outputUsersVar != null) {
                            Object name = outputUsersVar.getValue(delegateTask.getExecution());
                            if (name != null) {
                                execution.setVariable(name.toString(), users);
                            }
                        }
                    }

                    if (groups.size() > 0) {
                        if (outputGroupsVar != null) {
                            Object name = outputGroupsVar.getValue(delegateTask.getExecution());
                            if (name != null) {
                                execution.setVariable(name.toString(), groups);
                            }
                        }
                    }
                }
            }
        }
    }

    public void setOutputAssigneeVar(Expression outputAssigneeVar) {
        this.outputAssigneeVar = outputAssigneeVar;
    }

    public void setOutputGroupsVar(Expression outputGroupsVar) {
        this.outputGroupsVar = outputGroupsVar;
    }

    public void setOutputUsersVar(Expression outputUsersVar) {
        this.outputUsersVar = outputUsersVar;
    }

    public void setRoleId(Expression roleId) {
        this.roleId = roleId;
    }

    public void setStatementId(Expression statementId) {
        this.statementId = statementId;
    }

}