com.liferay.portal.kernel.security.auth.CompanyThreadLocal.java Source code

Introduction

Here is the source code for com.liferay.portal.kernel.security.auth.CompanyThreadLocal.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.kernel.security.auth;

import com.liferay.petra.lang.CentralizedThreadLocal;
import com.liferay.portal.kernel.change.tracking.CTCollectionThreadLocal;
import com.liferay.portal.kernel.log.Log;
import com.liferay.portal.kernel.log.LogFactoryUtil;
import com.liferay.portal.kernel.model.CompanyConstants;
import com.liferay.portal.kernel.model.User;
import com.liferay.portal.kernel.service.UserLocalServiceUtil;
import com.liferay.portal.kernel.util.LocaleThreadLocal;
import com.liferay.portal.kernel.util.TimeZoneThreadLocal;

/**
 * @author Brian Wing Shun Chan
 */
public class CompanyThreadLocal {

    public static Long getCompanyId() {
        Long companyId = _companyId.get();

        if (_log.isDebugEnabled()) {
            _log.debug("getCompanyId " + companyId);
        }

        return companyId;
    }

    public static boolean isDeleteInProcess() {
        return _deleteInProcess.get();
    }

    public static void setCompanyId(Long companyId) {
        if (companyId.equals(_companyId.get())) {
            return;
        }

        if (_log.isDebugEnabled()) {
            _log.debug("setCompanyId " + companyId);
        }

        if (companyId > 0) {
            _companyId.set(companyId);

            try {
                User defaultUser = UserLocalServiceUtil.getDefaultUser(companyId);

                LocaleThreadLocal.setDefaultLocale(defaultUser.getLocale());
                TimeZoneThreadLocal.setDefaultTimeZone(defaultUser.getTimeZone());
            } catch (Exception e) {
                _log.error(e, e);
            }
        } else {
            _companyId.set(CompanyConstants.SYSTEM);

            LocaleThreadLocal.setDefaultLocale(null);
            TimeZoneThreadLocal.setDefaultTimeZone(null);
        }

        CTCollectionThreadLocal.removeCTCollectionId();
    }

    public static void setDeleteInProcess(boolean deleteInProcess) {
        _deleteInProcess.set(deleteInProcess);
    }

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

    private static final ThreadLocal<Long> _companyId = new CentralizedThreadLocal<>(
            CompanyThreadLocal.class + "._companyId", () -> CompanyConstants.SYSTEM);
    private static final ThreadLocal<Boolean> _deleteInProcess = new CentralizedThreadLocal<>(
            CompanyThreadLocal.class + "._deleteInProcess", () -> Boolean.FALSE);

}