001
014
015 package com.liferay.portal.lar.messaging;
016
017 import com.liferay.portal.kernel.exception.PortalException;
018 import com.liferay.portal.kernel.exception.SystemException;
019 import com.liferay.portal.kernel.messaging.BaseMessageStatusMessageListener;
020 import com.liferay.portal.kernel.util.ArrayUtil;
021 import com.liferay.portal.model.CompanyConstants;
022 import com.liferay.portal.model.User;
023 import com.liferay.portal.security.auth.CompanyThreadLocal;
024 import com.liferay.portal.security.auth.PrincipalThreadLocal;
025 import com.liferay.portal.security.permission.PermissionChecker;
026 import com.liferay.portal.security.permission.PermissionCheckerFactoryUtil;
027 import com.liferay.portal.security.permission.PermissionThreadLocal;
028 import com.liferay.portal.service.ServiceContext;
029 import com.liferay.portal.service.ServiceContextThreadLocal;
030 import com.liferay.portal.service.UserLocalServiceUtil;
031 import com.liferay.portal.util.PortalUtil;
032
033 import java.io.Serializable;
034
035 import java.util.HashMap;
036 import java.util.Map;
037
038
041 public abstract class BasePublisherMessageListener
042 extends BaseMessageStatusMessageListener {
043
044 protected void initThreadLocals(
045 long userId, Map<String, String[]> parameterMap)
046 throws PortalException {
047
048 User user = UserLocalServiceUtil.getUserById(userId);
049
050 CompanyThreadLocal.setCompanyId(user.getCompanyId());
051
052 PrincipalThreadLocal.setName(userId);
053
054 PermissionChecker permissionChecker = null;
055
056 try {
057 permissionChecker = PermissionCheckerFactoryUtil.create(user);
058 }
059 catch (Exception e) {
060 throw new SystemException(e);
061 }
062
063 PermissionThreadLocal.setPermissionChecker(permissionChecker);
064
065 ServiceContext serviceContext = new ServiceContext();
066
067 serviceContext.setCompanyId(user.getCompanyId());
068 serviceContext.setPathMain(PortalUtil.getPathMain());
069 serviceContext.setSignedIn(!user.isDefaultUser());
070 serviceContext.setUserId(user.getUserId());
071
072 Map<String, Serializable> attributes =
073 new HashMap<String, Serializable>();
074
075 for (Map.Entry<String, String[]> entry : parameterMap.entrySet()) {
076 String param = entry.getKey();
077 String[] values = entry.getValue();
078
079 if (ArrayUtil.isNotEmpty(values)) {
080 if (values.length == 1) {
081 attributes.put(param, values[0]);
082 }
083 else {
084 attributes.put(param, values);
085 }
086 }
087 }
088
089 serviceContext.setAttributes(attributes);
090
091 ServiceContextThreadLocal.pushServiceContext(serviceContext);
092 }
093
094 protected void resetThreadLocals() {
095 CompanyThreadLocal.setCompanyId(CompanyConstants.SYSTEM);
096 PermissionThreadLocal.setPermissionChecker(null);
097 PrincipalThreadLocal.setName(null);
098 ServiceContextThreadLocal.popServiceContext();
099 }
100
101 }