org.snapscript.agent.profiler.ProfileResultUpdater Maven / Gradle / Ivy

Go to download
package org.snapscript.agent.profiler;

import java.util.Set;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.atomic.AtomicReference;

import org.snapscript.agent.event.ProcessEventChannel;
import org.snapscript.agent.event.ProfileEvent;
import org.snapscript.common.thread.ThreadBuilder;

public class ProfileResultUpdater implements Runnable {

   private final AtomicReference reference;
   private final ProcessEventChannel channel;
   private final ProcessProfiler profiler;
   private final ThreadFactory factory;

   public ProfileResultUpdater(ProcessProfiler profiler, ProcessEventChannel channel) {
      this.reference = new AtomicReference();
      this.factory = new ThreadBuilder();
      this.profiler = profiler;
      this.channel = channel;
   }
   
   public void start(String process) {
      if(reference.compareAndSet(null, process)) {
         Thread thread = factory.newThread(this);
         thread.start();
      }
   }

   @Override
   public void run() {
      long delay = 1000;
      
      while(true) {
         String process = reference.get();
         try {
            Thread.sleep(delay);
            Set results = profiler.lines(2000);
            ProfileEvent event = new ProfileEvent.Builder(process)
               .withResults(results)
               .build();
            
            channel.send(event);
         }catch(Exception e) {
            e.printStackTrace();
         }finally{
            delay = 5000;
         }
      }
   }
}