org.glowroot.agent.impl.UserProfileScheduler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of glowroot-agent Show documentation
Glowroot Agent
There is a newer version: 0.9.28
/*
 * Copyright 2012-2015 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.glowroot.agent.impl;

import java.lang.management.ManagementFactory;
import java.lang.management.ThreadInfo;
import java.lang.management.ThreadMXBean;
import java.util.concurrent.ScheduledExecutorService;

import org.glowroot.agent.shaded.google.common.annotations.VisibleForTesting;
import org.glowroot.agent.shaded.google.common.collect.ImmutableList;

import org.glowroot.agent.config.ConfigService;
import org.glowroot.agent.model.Transaction;
import org.glowroot.agent.model.Transaction.OverrideSource;
import org.glowroot.common.config.UserRecordingConfig;
import org.glowroot.common.util.ScheduledRunnable;

import static java.util.concurrent.TimeUnit.MILLISECONDS;
import static java.util.concurrent.TimeUnit.NANOSECONDS;

public class UserProfileScheduler {

    private final ScheduledExecutorService scheduledExecutor;
    private final ConfigService configService;

    public UserProfileScheduler(ScheduledExecutorService scheduledExecutor,
            ConfigService configService) {
        this.scheduledExecutor = scheduledExecutor;
        this.configService = configService;
    }

    void maybeScheduleUserProfiling(Transaction transaction, String user) {
        UserRecordingConfig userRecordingConfig = configService.getUserRecordingConfig();
        ImmutableList users = userRecordingConfig.users();
        if (users.isEmpty()) {
            return;
        }
        if (!TransactionCollector.containsIgnoreCase(users, user)) {
            return;
        }
        // for now lumping user recording into slow traces tab
        transaction.setSlowThresholdMillis(0, OverrideSource.USER_RECORDING);

        // schedule the first stack collection for configured interval after transaction start (or
        // immediately, if the transaction's total time already exceeds configured collection
        // interval)
        Integer intervalMillis = userRecordingConfig.profilingIntervalMillis();
        if (intervalMillis == null || intervalMillis <= 0) {
            return;
        }
        ScheduledRunnable userProfileRunnable = new UserProfileRunnable(transaction, configService);
        long initialDelay =
                Math.max(0, intervalMillis - NANOSECONDS.toMillis(transaction.getDurationNanos()));
        userProfileRunnable.scheduleWithFixedDelay(scheduledExecutor, initialDelay, intervalMillis,
                MILLISECONDS);
        transaction.setUserProfileRunnable(userProfileRunnable);
    }

    @VisibleForTesting
    static class UserProfileRunnable extends ScheduledRunnable {

        private final Transaction transaction;
        private final ConfigService configService;

        @VisibleForTesting
        UserProfileRunnable(Transaction transaction, ConfigService configService) {
            this.transaction = transaction;
            this.configService = configService;
        }

        @Override
        public void runInternal() {
            if (transaction.isCompleted()) {
                // there is a small window between trace completion and cancellation of this
                // command,
                // plus, should a stop-the-world gc occur in this small window, even two command
                // executions can fire one right after the other in the small window (assuming the
                // first
                // didn't throw an exception which it does now), since this command is scheduled
                // using
                // ScheduledExecutorService.scheduleWithFixedDelay()
                throw new TerminateSubsequentExecutionsException();
            }
            ThreadMXBean threadBean = ManagementFactory.getThreadMXBean();
            ThreadInfo threadInfo =
                    threadBean.getThreadInfo(transaction.getThreadId(), Integer.MAX_VALUE);
            transaction.captureStackTrace(threadInfo,
                    configService.getAdvancedConfig().maxStackTraceSamplesPerTransaction(),
                    configService.getAdvancedConfig().timerWrapperMethods());
        }
    }
}