com.liferay.portal.security.permission.DoAsUserThread.java Source code

Introduction

Here is the source code for com.liferay.portal.security.permission.DoAsUserThread.java

Source

/**
 * Copyright (c) 2000-present Liferay, Inc. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 2.1 of the License, or (at your option)
 * any later version.
 *
 * This library 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 Lesser General Public License for more
 * details.
 */

package com.liferay.portal.security.permission;

import com.liferay.portal.kernel.log.Log;
import com.liferay.portal.kernel.log.LogFactoryUtil;
import com.liferay.portal.kernel.model.User;
import com.liferay.portal.kernel.security.auth.CompanyThreadLocal;
import com.liferay.portal.kernel.security.auth.PrincipalThreadLocal;
import com.liferay.portal.kernel.security.permission.PermissionChecker;
import com.liferay.portal.kernel.security.permission.PermissionCheckerFactoryUtil;
import com.liferay.portal.kernel.security.permission.PermissionThreadLocal;
import com.liferay.portal.kernel.service.UserLocalServiceUtil;

/**
 * @author Brian Wing Shun Chan
 */
public abstract class DoAsUserThread extends Thread {

    public DoAsUserThread(long userId) {
        this(userId, 1);
    }

    public DoAsUserThread(long userId, int retries) {
        _userId = userId;
        _retries = retries;
    }

    public boolean isSuccess() {
        return _success;
    }

    @Override
    public void run() {
        for (int i = 0; i < _retries; i++) {
            try {
                User user = UserLocalServiceUtil.getUserById(_userId);

                CompanyThreadLocal.setCompanyId(user.getCompanyId());

                PrincipalThreadLocal.setName(_userId);

                PermissionChecker permissionChecker = PermissionCheckerFactoryUtil.create(user);

                PermissionThreadLocal.setPermissionChecker(permissionChecker);

                doRun();

                _success = true;

                return;
            } catch (Exception e) {
                _log.error(e, e);
            } finally {
                PrincipalThreadLocal.setName(null);
                PermissionThreadLocal.setPermissionChecker(null);
            }
        }
    }

    protected abstract void doRun() throws Exception;

    private static final Log _log = LogFactoryUtil.getLog(DoAsUserThread.class);

    private final int _retries;
    private boolean _success;
    private final long _userId;

}