/*
DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
Copyright (c) 2016 Payara Foundation. All rights reserved.
The contents of this file are subject to the terms of the Common Development
and Distribution License("CDDL") (collectively, the "License"). You
may not use this file except in compliance with the License. You can
obtain a copy of the License at
https://glassfish.dev.java.net/public/CDDL+GPL_1_1.html
or packager/legal/LICENSE.txt. See the License for the specific
language governing permissions and limitations under the License.
When distributing the software, include this License Header Notice in each
file and include the License file at packager/legal/LICENSE.txt.
*/
package fish.payara.nucleus.requesttracing.admin;
import com.sun.enterprise.config.serverbeans.Config;
import com.sun.enterprise.config.serverbeans.Domain;
import com.sun.enterprise.util.LocalStringManagerImpl;
import com.sun.enterprise.util.SystemPropertyConstants;
import fish.payara.nucleus.requesttracing.RequestTracingService;
import fish.payara.nucleus.requesttracing.configuration.RequestTracingServiceConfiguration;
import org.glassfish.api.ActionReport;
import org.glassfish.api.I18n;
import org.glassfish.api.Param;
import org.glassfish.api.admin.*;
import org.glassfish.config.support.CommandTarget;
import org.glassfish.config.support.TargetType;
import org.glassfish.hk2.api.PerLookup;
import org.glassfish.internal.api.Target;
import org.jvnet.hk2.annotations.Service;
import org.jvnet.hk2.config.ConfigSupport;
import org.jvnet.hk2.config.SingleConfigCode;
import org.jvnet.hk2.config.TransactionFailure;
import javax.inject.Inject;
import java.beans.PropertyVetoException;
import java.util.Properties;
import java.util.concurrent.TimeUnit;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.glassfish.hk2.api.ServiceLocator;
/**
* Admin command to enable/disable all request tracing services defined in
* domain.xml.
*
* @author mertcaliskan
*/
@ExecuteOn({RuntimeType.DAS, RuntimeType.INSTANCE})
@TargetType(value = {CommandTarget.DAS, CommandTarget.STANDALONE_INSTANCE, CommandTarget.CLUSTER, CommandTarget.CLUSTERED_INSTANCE, CommandTarget.CONFIG})
@Service(name = "requesttracing-configure")
@CommandLock(CommandLock.LockType.NONE)
@PerLookup
@I18n("requesttracing.configure")
@RestEndpoints({
@RestEndpoint(configBean = Domain.class,
opType = RestEndpoint.OpType.POST,
path = "requesttracing-configure",
description = "Enables/Disables Request Tracing Service")
})
public class RequestTracingConfigurer implements AdminCommand {
final private static LocalStringManagerImpl strings = new LocalStringManagerImpl(RequestTracingConfigurer.class);
@Inject
ServerEnvironment server;
@Inject
RequestTracingService service;
@Inject
protected Logger logger;
@Inject
protected Target targetUtil;
@Param(name = "target", optional = true, defaultValue = SystemPropertyConstants.DAS_SERVER_NAME)
String target;
@Param(name = "enabled", optional = false)
private Boolean enabled;
@Param(name = "dynamic", optional = true, defaultValue = "false")
private Boolean dynamic;
@Param(name = "thresholdUnit", optional = true, defaultValue = "SECONDS")
private String unit;
@Param(name = "thresholdValue", optional = true, defaultValue = "30")
private String value;
@Inject
ServiceLocator serviceLocator;
CommandRunner.CommandInvocation inv;
@Override
public void execute(AdminCommandContext context) {
final AdminCommandContext theContext = context;
final ActionReport actionReport = context.getActionReport();
Properties extraProperties = actionReport.getExtraProperties();
if (extraProperties == null) {
extraProperties = new Properties();
actionReport.setExtraProperties(extraProperties);
}
if (!validate(actionReport)) {
return;
}
Config config = targetUtil.getConfig(target);
final RequestTracingServiceConfiguration requestTracingServiceConfiguration = config.getExtensionByType(RequestTracingServiceConfiguration.class);
if (requestTracingServiceConfiguration != null) {
try {
ConfigSupport.apply(new SingleConfigCode<RequestTracingServiceConfiguration>() {
@Override
public Object run(final RequestTracingServiceConfiguration requestTracingServiceConfigurationProxy) throws
PropertyVetoException, TransactionFailure {
if (enabled != null) {
requestTracingServiceConfigurationProxy.enabled(enabled.toString());
}
if (unit != null) {
requestTracingServiceConfigurationProxy.setThresholdUnit(unit);
}
if (value != null) {
requestTracingServiceConfigurationProxy.setThresholdValue(value);
}
actionReport.setActionExitCode(ActionReport.ExitCode.SUCCESS);
return requestTracingServiceConfigurationProxy;
}
}, requestTracingServiceConfiguration);
} catch (TransactionFailure ex) {
logger.log(Level.WARNING, "Exception during command ", ex);
actionReport.setMessage(ex.getCause().getMessage());
actionReport.setActionExitCode(ActionReport.ExitCode.FAILURE);
return;
}
}
if (dynamic) {
if (server.isDas()) {
if (targetUtil.getConfig(target).isDas()) {
service.getExecutionOptions().setEnabled(enabled);
if (value != null) {
service.getExecutionOptions().setThresholdValue(Long.valueOf(value));
actionReport.appendMessage(strings.getLocalString("requesttracing.configure.thresholdvalue.success",
"Request Tracing Service Threshold Value is set to {0}.", value) + "\n");
}
if (unit != null) {
service.getExecutionOptions().setThresholdUnit(TimeUnit.valueOf(unit));
actionReport.appendMessage(strings.getLocalString("requesttracing.configure.thresholdunit.success",
"Request Tracing Service Threshold Unit is set to {0}.", unit) + "\n");
}
}
} else {
service.getExecutionOptions().setEnabled(enabled);
if (value != null) {
service.getExecutionOptions().setThresholdValue(Long.valueOf(value));
actionReport.appendMessage(strings.getLocalString("requesttracing.configure.thresholdvalue.success",
"Request Tracing Service Threshold Value is set to {0}.", value) + "\n");
}
if (unit != null) {
service.getExecutionOptions().setThresholdUnit(TimeUnit.valueOf(unit));
actionReport.appendMessage(strings.getLocalString("requesttracing.configure.thresholdunit.success",
"Request Tracing Service Threshold Unit is set to {0}.", unit) + "\n");
}
}
}
}
private boolean validate(ActionReport actionReport) {
boolean result = false;
if (value != null) {
try {
int thresholdValue = Integer.parseInt(value);
if (thresholdValue <= 0 || thresholdValue > Short.MAX_VALUE * 2) {
actionReport.failure(logger, "Threshold Value must be greater than zero or less than " + Short.MAX_VALUE * 2 + 1);
return result;
}
} catch (NumberFormatException nfe) {
actionReport.failure(logger, "Threshold Value is not a valid integer", nfe);
return result;
}
}
if (unit != null) {
try {
if (!unit.equals("NANOSECONDS")
&& !unit.equals("MICROSECONDS")
&& !unit.equals("MILLISECONDS")
&& !unit.equals("SECONDS")
&& !unit.equals("MINUTES")
&& !unit.equals("HOURS")
&& !unit.equals("DAYS")) {
actionReport.failure(logger, unit + " is an invalid time unit");
return result;
}
} catch (IllegalArgumentException iaf) {
actionReport.failure(logger, unit + " is an invalid time unit", iaf);
return result;
}
}
return true;
}
}