org.snapscript.studio.agent.event.BeginEventMarshaller Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of snap-all Show documentation
Dynamic scripting for the JVM
There is a newer version: 1.4.6
package org.snapscript.studio.agent.event;

import static org.snapscript.studio.agent.event.ProcessEventType.START;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;

import org.snapscript.studio.agent.ProcessMode;
import org.snapscript.studio.agent.core.ExecuteStatus;

public class BeginEventMarshaller implements ProcessEventMarshaller {

   @Override
   public BeginEvent fromMessage(MessageEnvelope message) throws IOException {
      byte[] array = message.getData();
      int length = message.getLength();
      int offset = message.getOffset();
      ByteArrayInputStream buffer = new ByteArrayInputStream(array, offset, length);
      DataInputStream input = new DataInputStream(buffer);
      String process = input.readUTF();
      String system = input.readUTF();  
      String project = input.readUTF(); 
      String resource = input.readUTF();      
      String status = input.readUTF();
      String mode = input.readUTF();
      long totalMemory = input.readLong();
      long usedMemory = input.readLong();
      int threads = input.readInt();
      long duration = input.readLong();
      
      return new BeginEvent.Builder(process)
         .withMode(ProcessMode.resolveMode(mode))
         .withDuration(duration)
         .withSystem(system)
         .withProject(project)
         .withResource(resource)
         .withStatus(ExecuteStatus.resolveStatus(status))
         .withTotalMemory(totalMemory)
         .withUsedMemory(usedMemory)
         .withThreads(threads)
         .build();
   }

   @Override
   public MessageEnvelope toMessage(BeginEvent event) throws IOException {
      ByteArrayOutputStream buffer = new ByteArrayOutputStream();
      DataOutputStream output = new DataOutputStream(buffer);      
      String process = event.getProcess();
      String system = event.getSystem();
      String project = event.getProject();
      String resource = event.getResource();
      ExecuteStatus status = event.getStatus();
      ProcessMode mode = event.getMode();
      long totalMemory = event.getTotalMemory();
      long usedMemory = event.getUsedMemory();
      int threads = event.getThreads();
      long duration = event.getDuration();
      
      output.writeUTF(process);
      output.writeUTF(system);
      output.writeUTF(project);
      output.writeUTF(resource);
      output.writeUTF(status.name());
      output.writeUTF(mode.name());
      output.writeLong(totalMemory);
      output.writeLong(usedMemory);
      output.writeInt(threads);
      output.writeLong(duration);
      output.flush();
      
      byte[] array = buffer.toByteArray();
      return new MessageEnvelope(START.code, array, 0, array.length);
   }
}