Refactor Censored dto.

Twinkle942910 · Apr 16, 2018 · 9435a04 · 9435a04
1 parent b8ec93b
commit 9435a04
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/pom.xml b/pom.xml
@@ -46,6 +46,12 @@
             <version>${aho.corasick.version}</version>
         </dependency>
 
+        <dependency>
+            <groupId>com.fasterxml.jackson.core</groupId>
+            <artifactId>jackson-annotations</artifactId>
+            <version>2.9.0</version>
+        </dependency>
+
     </dependencies>
 
     <build>

diff --git a/src/main/java/com/filter/textcorrector/profanity_filtering/ProfanityFilter.java b/src/main/java/com/filter/textcorrector/profanity_filtering/ProfanityFilter.java
@@ -65,7 +65,7 @@ public Censored searchForProfanity(String input) {
         LOGGER.debug("Censored text: " + input);
         LOGGER.debug("Censoring took time: " + (endProccessingTime - startProccessingTime) / (double) 1000000 + " ms");
 
-        return new Censored(input, badWords);
+        return new Censored(input, badWords, !badWords.isEmpty());
     }
 
     public void changeLanguage(Language language){

diff --git a/src/main/java/com/filter/textcorrector/profanity_filtering/model/Censored.java b/src/main/java/com/filter/textcorrector/profanity_filtering/model/Censored.java
@@ -1,5 +1,6 @@
 package com.filter.textcorrector.profanity_filtering.model;
 
+import com.fasterxml.jackson.annotation.JsonProperty;
 import lombok.AllArgsConstructor;
 import lombok.Data;
 import lombok.NoArgsConstructor;
@@ -10,6 +11,12 @@
 @NoArgsConstructor
 @AllArgsConstructor
 public class Censored {
+    @JsonProperty("censored_text")
     private String censoredText;
+
+    @JsonProperty("bad_words")
     private Set<String> badWordList;
+
+    @JsonProperty("contains_profanity")
+    private boolean containsProfanity;
 }