com.sccl.attech.common.service.BaseService.java Source code

Introduction

Here is the source code for com.sccl.attech.common.service.BaseService.java

Source

/**
 * Copyright &copy; 2012-2013 <a href="https://github.com/sccl/attech">attech</a> All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 */
package com.sccl.attech.common.service;

import java.util.Iterator;
import java.util.List;

import org.apache.commons.lang3.StringUtils;
import org.hibernate.criterion.Criterion;
import org.hibernate.criterion.Junction;
import org.hibernate.criterion.Restrictions;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.common.collect.Lists;
import com.sccl.attech.modules.sys.entity.Role;
import com.sccl.attech.modules.sys.entity.User;

/**
 * Service
 * @author sccl
 * @version 2013-05-15
 */
public abstract class BaseService {

    /**
     * 
     */
    protected Logger logger = LoggerFactory.getLogger(getClass());

    /**
     * ?
     * @param user ?UserUtils.getUser()??
     * @param officeAlias ??dc.createAlias("office", "office");
     * @param userAlias ???
     * @return ?
     */
    protected static Junction dataScopeFilter(User user, String officeAlias, String userAlias) {

        // ????
        List<String> dataScope = Lists.newArrayList();
        Junction junction = Restrictions.disjunction();

        // ???
        if (!user.isAdmin()) {
            for (Role r : user.getRoleList()) {
                if (!dataScope.contains(r.getDataScope()) && StringUtils.isNotBlank(officeAlias)) {
                    boolean isDataScopeAll = false;
                    if (Role.DATA_SCOPE_ALL.equals(r.getDataScope())) {
                        isDataScopeAll = true;
                    } else if (Role.DATA_SCOPE_COMPANY_AND_CHILD.equals(r.getDataScope())) {
                        junction.add(Restrictions.eq("company.id", user.getCompany().getId()));
                        //junction.add(Restrictions.like(officeAlias+".parentIds", "%"+user.getCompany().getParentIds()+","+user.getCompany().getId()+"%"));
                    }
                    //               else if (Role.DATA_SCOPE_COMPANY.equals(r.getDataScope())){
                    //                  junction.add(Restrictions.eq(officeAlias+".id", user.getCompany().getId()));
                    //                  junction.add(Restrictions.and(Restrictions.eq(officeAlias+".id", user.getCompany().getId()),
                    //                        Restrictions.eq(officeAlias+".type", "2"))); // ?
                    //               }
                    else if (Role.DATA_SCOPE_OFFICE_AND_CHILD.equals(r.getDataScope())) {
                        junction.add(Restrictions.eq(officeAlias + ".id", user.getOffice().getId()));
                        junction.add(Restrictions.like(officeAlias + ".parentIds",
                                "%" + user.getOffice().getParentIds() + "," + user.getOffice().getId() + "%"));
                    } else if (Role.DATA_SCOPE_OFFICE.equals(r.getDataScope())) {
                        junction.add(Restrictions.eq(officeAlias + ".id", user.getOffice().getId()));
                    } else if (Role.DATA_SCOPE_CUSTOM.equals(r.getDataScope())) {
                        junction.add(Restrictions.in(officeAlias + ".id", r.getOfficeIdList()));
                    }
                    //else if (Role.DATA_SCOPE_SELF.equals(r.getDataScope())){
                    if (!isDataScopeAll) {
                        if (StringUtils.isNotBlank(userAlias)) {
                            junction.add(Restrictions.eq(userAlias + ".id", user.getId()));
                        } else {
                            junction.add(Restrictions.isNull(officeAlias + ".id"));
                        }
                    } else {
                        // ?????
                        junction = Restrictions.disjunction();
                        break;
                    }

                    dataScope.add(r.getDataScope());
                }
            }
        }

        return junction;
    }

    /**
     * ?
     * @param user ?UserUtils.getUser()??
     * @param officeAlias ??dc.createAlias("office", "office");
     * @param userAlias ???
     * @return ql
     */
    protected static String dataScopeFilterString(User user, String officeAlias, String userAlias) {

        Junction junction = dataScopeFilter(user, officeAlias, userAlias);
        Iterator<Criterion> it = junction.conditions().iterator();

        StringBuilder ql = new StringBuilder();
        if (it.hasNext()) {
            String[] strField = { ".parentIds like ", ".type=", ".id=" };// ????
            ql.append(" and (");
            if (it.hasNext()) {
                String s = it.next().toString();
                for (String field : strField) {
                    s = s.replaceAll(field + "([\\w,%]+)", field + "'$1'");
                }
                ql.append(s);
            }
            while (it.hasNext()) {
                ql.append(" or (");
                String s = it.next().toString();
                for (String field : strField) {
                    s = s.replaceAll(field + "([\\w,%]+)", field + "'$1'");
                }
                ql.append(s).append(")");
            }
            ql.append(")");
        }
        return ql.toString();
    }

    protected List<Long> getIdList(String ids) {
        List<Long> idList = Lists.newArrayList();
        if (StringUtils.isNotBlank(ids)) {
            ids = ids.trim().replace("", ",").replace(" ", ",").replace("", ",");
            String[] arrId = ids.split(",");
            for (String id : arrId) {
                if (id.matches("\\d*")) {
                    idList.add(Long.valueOf(id));
                }
            }
        }
        return idList;
    }
}