com.sitewhere.microservice.security.SystemUserCallable Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sitewhere-microservice Show documentation

SiteWhere Microservice Components Library

There is a newer version: 3.0.13

Show newest version

/*
 * Copyright (c) SiteWhere, LLC. All rights reserved. http://www.sitewhere.com
 *
 * The software in this package is published under the terms of the CPAL v1.0
 * license, a copy of which has been included with this distribution in the
 * LICENSE.txt file.
 */
package com.sitewhere.microservice.security;

import java.util.concurrent.Callable;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.sitewhere.spi.SiteWhereException;
import com.sitewhere.spi.microservice.IFunctionIdentifier;
import com.sitewhere.spi.microservice.IMicroservice;
import com.sitewhere.spi.microservice.IMicroserviceConfiguration;

import io.sitewhere.k8s.crd.tenant.SiteWhereTenant;

/**
 * Allows code to be run in a separate thread along with thread local security
 * credentials for the superuser account. This allows non-authenticated services
 * to interact with GRPC persistence APIs.
 */
public abstract class SystemUserCallable implements Callable {

    /** Static logger instance */
    private static Log LOGGER = LogFactory.getLog(SystemUserCallable.class);

    /** Tenant engine if tenant operation */
    private IMicroservice microservice;

    /** Tenant */
    private SiteWhereTenant tenant;

    public SystemUserCallable(
	    IMicroservice microservice,
	    SiteWhereTenant tenant) {
	this.microservice = microservice;
	this.tenant = tenant;
    }

    /**
     * Implemented in subclasses to specifiy code that should be run as the system
     * user.
     * 
     * @throws SiteWhereException
     */
    public abstract V runAsSystemUser() throws SiteWhereException;

    /*
     * @see java.util.concurrent.Callable#call()
     */
    @Override
    public V call() throws Exception {
	SiteWhereAuthentication previous = UserContext.getCurrentUser();
	try {
	    if (tenant != null) {
		SiteWhereAuthentication system = getMicroservice().getSystemUser()
			.getAuthenticationForTenant(getTenant());
		UserContext.setContext(system);
	    } else {
		SiteWhereAuthentication system = getMicroservice().getSystemUser().getAuthentication();
		UserContext.setContext(system);
	    }
	    return runAsSystemUser();
	} catch (Throwable e) {
	    LOGGER.error("Unhandled exception.", e);
	    throw e;
	} finally {
	    UserContext.setContext(previous);
	}
    }

    protected IMicroservice getMicroservice() {
	return microservice;
    }

    protected SiteWhereTenant getTenant() {
	return tenant;
    }
}