apm.common.service.DataService.java Source code

Introduction

Here is the source code for apm.common.service.DataService.java

Source

/**
 * Copyright © 2012-2013 Zaric All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 */
package apm.common.service;

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

import apm.common.utils.StringUtils;
import apm.modules.sys.entity.Role;
import apm.modules.sys.entity.User;

import org.hibernate.criterion.Criterion;
import org.hibernate.criterion.Junction;
import org.hibernate.criterion.Restrictions;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.transaction.annotation.Transactional;

import com.google.common.collect.Lists;

@Transactional(readOnly = true)
public abstract class DataService {

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

    /**
     * ?
     * @param dc Hibernate
     * @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()) {
                String DATA_SCOPE = r.getDataScope();
                if (!dataScope.contains(DATA_SCOPE) && StringUtils.isNotBlank(officeAlias)) {
                    boolean isDataScopeAll = false;
                    if (Role.DATA_SCOPE_ALL.equals(DATA_SCOPE)) {
                        isDataScopeAll = true;
                    } else if (Role.DATA_SCOPE_COMPANY_AND_CHILD.equals(DATA_SCOPE)) { //(?)??
                        junction.add(Restrictions.eq(officeAlias + ".id", user.getCompany().getId()));
                        junction.add(Restrictions.like(officeAlias + ".parentIds",
                                user.getCompany().getParentIds() + user.getCompany().getId() + ",%"));
                    } else if (Role.DATA_SCOPE_COMPANY.equals(DATA_SCOPE)) {
                        junction.add(Restrictions.eq(officeAlias + ".id", user.getCompany().getId()));
                        junction.add(Restrictions.and(
                                Restrictions.eq(officeAlias + ".parent.id", user.getCompany().getId()),
                                Restrictions.eq(officeAlias + ".type", "2"))); // ?
                    } else if (Role.DATA_SCOPE_OFFICE_AND_CHILD.equals(DATA_SCOPE)) {
                        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(DATA_SCOPE)) {
                        junction.add(Restrictions.eq(officeAlias + ".id", user.getOffice().getId()));
                    } else if (Role.DATA_SCOPE_CUSTOM.equals(DATA_SCOPE)) {
                        junction.add(Restrictions.in(officeAlias + ".id", r.getOfficeIdList()));
                    }
                    //else if (Role.DATA_SCOPE_SELF.equals(DATA_SCOPE)){
                    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(DATA_SCOPE);
                }
            }
        }
        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();
        ql.append(" and (");
        if (it.hasNext()) {
            ql.append(it.next());
        }
        String[] strField = { ".parentIds like ", ".type=" }; // ????
        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();
    }
}