com.mycollab.module.user.service.mybatis.RoleServiceDBImpl.java Source code

Introduction

Here is the source code for com.mycollab.module.user.service.mybatis.RoleServiceDBImpl.java

Source

/**
 * This file is part of mycollab-services.
 *
 * mycollab-services 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 3 of the License, or
 * (at your option) any later version.
 *
 * mycollab-services 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 mycollab-services.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.mycollab.module.user.service.mybatis;

import com.mycollab.db.persistence.ICrudGenericDAO;
import com.mycollab.db.persistence.ISearchableDAO;
import com.mycollab.db.persistence.service.DefaultService;
import com.mycollab.module.user.dao.RoleMapper;
import com.mycollab.module.user.dao.RoleMapperExt;
import com.mycollab.module.user.dao.RolePermissionMapper;
import com.mycollab.module.user.domain.*;
import com.mycollab.module.user.domain.criteria.RoleSearchCriteria;
import com.mycollab.module.user.service.RoleService;
import com.mycollab.security.PermissionMap;
import org.apache.commons.collections.CollectionUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.List;

/**
 * @author MyCollab Ltd.
 * @since 1.0
 */
@Service
public class RoleServiceDBImpl extends DefaultService<Integer, Role, RoleSearchCriteria> implements RoleService {

    @Autowired
    private RoleMapper roleMapper;
    @Autowired
    private RoleMapperExt roleMapperExt;

    @Autowired
    private RolePermissionMapper rolePermissionMapper;

    @Override
    public ICrudGenericDAO<Integer, Role> getCrudMapper() {
        return roleMapper;
    }

    @Override
    public ISearchableDAO<RoleSearchCriteria> getSearchMapper() {
        return roleMapperExt;
    }

    @Override
    public Integer saveWithSession(Role record, String username) {
        if (Boolean.TRUE.equals(record.getIsdefault())) {
            setAllRoleNotDefault(record.getSaccountid());
        }
        return super.saveWithSession(record, username);
    }

    private void setAllRoleNotDefault(Integer sAccountId) {
        Role updateRecord = new Role();
        updateRecord.setIsdefault(Boolean.FALSE);
        RoleExample ex = new RoleExample();
        ex.createCriteria().andSaccountidEqualTo(sAccountId);
        roleMapper.updateByExampleSelective(updateRecord, ex);
    }

    @Override
    public Integer updateWithSession(Role record, String username) {
        if (Boolean.TRUE.equals(record.getIsdefault())) {
            setAllRoleNotDefault(record.getSaccountid());
        }
        return super.updateWithSession(record, username);
    }

    @Override
    public void savePermission(Integer roleId, PermissionMap permissionMap, Integer accountid) {
        String perVal = permissionMap.toJsonString();

        RolePermissionExample ex = new RolePermissionExample();
        ex.createCriteria().andRoleidEqualTo(roleId);

        RolePermission rolePer = new RolePermission();
        rolePer.setRoleid(roleId);
        rolePer.setRoleval(perVal);

        Long data = rolePermissionMapper.countByExample(ex);
        if (data > 0) {
            rolePermissionMapper.updateByExampleSelective(rolePer, ex);
        } else {
            rolePermissionMapper.insert(rolePer);
        }
    }

    @Override
    public SimpleRole findById(Integer roleId, Integer sAccountId) {
        return roleMapperExt.findById(roleId);
    }

    @Override
    public Integer getDefaultRoleId(Integer sAccountId) {
        RoleExample ex = new RoleExample();
        ex.createCriteria().andIsdefaultEqualTo(Boolean.TRUE).andSaccountidEqualTo(sAccountId);
        List<Role> roles = roleMapper.selectByExample(ex);
        if (CollectionUtils.isNotEmpty(roles)) {
            return roles.get(0).getId();
        } else {
            ex = new RoleExample();
            ex.createCriteria().andRolenameEqualTo(SimpleRole.GUEST).andSaccountidEqualTo(sAccountId);
            roles = roleMapper.selectByExample(ex);
            return CollectionUtils.isNotEmpty(roles) ? roles.get(0).getId() : null;
        }
    }
}