Refactoring.

structurizr · Jan 25, 2024 · eb3ddda · eb3ddda
1 parent e0141d1
commit eb3ddda
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/structurizr-inspection/src/main/java/com/structurizr/inspection/Inspection.java b/structurizr-inspection/src/main/java/com/structurizr/inspection/Inspection.java
@@ -10,7 +10,7 @@ protected Inspection(Inspector inspector) {
         this.inspector = inspector;
     }
 
-    protected abstract String  getType();
+    protected abstract String getType();
 
     public Inspector getInspector() {
         return inspector;
@@ -25,7 +25,7 @@ protected Violation noViolation() {
     }
 
     protected Violation violation(String description) {
-        return new Violation(getType(), description);
+        return new Violation(this, description);
     }
 
 }
diff --git a/structurizr-inspection/src/main/java/com/structurizr/inspection/Violation.java b/structurizr-inspection/src/main/java/com/structurizr/inspection/Violation.java
@@ -2,17 +2,17 @@
 
 public final class Violation {
 
-    private final String type;
+    private Inspection inspection;
     private Severity severity;
     private final String message;
 
-    Violation(String type, String message) {
-        this.type = type;
+    Violation(Inspection inspection, String message) {
+        this.inspection = inspection;
         this.message = message;
     }
 
     public String getType() {
-        return type;
+        return inspection.getType();
     }
 
     public Severity getSeverity() {
@@ -25,7 +25,7 @@ public String getMessage() {
 
     @Override
     public String toString() {
-        return type + " | " + severity + " | " + message;
+        return inspection.getType() + " | " + severity + " | " + message;
     }
 
     public Violation withSeverity(Severity severity) {