Merge pull request #9 from grove693/feature/small-refactorings

Created TransactionType enum + added fluent methods on Transaction object

Author: ilvalerione

src/main/java/dev/inspector/agent/model/Transaction.java

@@ -16,7 +16,7 @@ public class Transaction extends Context implements Transportable {
 
     private static Logger LOGGER = LoggerFactory.getLogger(Transaction.class);
     private String model = "transaction";
-    private String type = "request";
+    private TransactionType type;
     private String name;
     private String hash = System.currentTimeMillis() + "" + (int) (Math.random() * 100);
     private User user;
@@ -28,23 +28,38 @@ public class Transaction extends Context implements Transportable {
     public Transaction(String name) {
         this.name = name;
         this.timestamp = TimesUtils.getTimestamp();
+        type = TransactionType.GENERAL;
     }
-    public void withUser(User user){
+    public Transaction withUser(User user){
         this.user = user;
+
+        return this;
+    }
+
+    public Transaction withType(TransactionType type) {
+        this.type = type;
+
+        return this;
     }
 
-    public void setResult(String result){
+    public Transaction setResult(String result){
         this.result = result;
+
+        return this;
     }
 
-    public void end(){
+    public Transaction end(){
         BigDecimal end = TimesUtils.getTimestamp();
         this.end(end.subtract(this.timestamp).multiply(BigDecimal.valueOf(1000)));
+
+        return this;
     }
 
-    public void end(BigDecimal duration){
+    public Transaction end(BigDecimal duration){
         this.durationInMillis = duration.longValue();
         this.memoryPeak = this.getMemoryPeak();
+
+        return this;
     }
 
     public static long getMemoryPeak() {
@@ -75,7 +90,7 @@ public JSONObject toTransport() {
             .put("model", this.model)
             .put("hash", this.hash)
             .put("name", this.name)
-            .put("type", this.type)
+            .put("type", this.type.getType())
             .put("timestamp", this.timestamp)
             .put("duration", this.durationInMillis)
             .put("result", this.result)

src/main/java/dev/inspector/agent/model/TransactionType.java

@@ -0,0 +1,18 @@
+package dev.inspector.agent.model;
+
+public enum TransactionType {
+
+    GENERAL("transaction"),
+    REQUEST("request"),
+    SCHEDULER("scheduler");
+
+    private final String type;
+
+    TransactionType(String type) {
+        this.type = type;
+    }
+
+    public String getType() {
+        return type;
+    }
+}