[Backport] Fix JSON serde for taskStatusPlus (#5469) (#5472)

* Fix JSON serde for taskStatusPlus (#5469)

* Fix JSON serde for taskStatusPlus

* add newline

* fix to statusCode

* fix fobidden api check

* remove debugging code

* Compile fix

Author: jon-wei

api/src/main/java/io/druid/indexer/TaskStatusPlus.java

@@ -25,6 +25,7 @@
 import org.joda.time.DateTime;
 
 import javax.annotation.Nullable;
+import java.util.Objects;
 
 public class TaskStatusPlus
 {
@@ -40,7 +41,7 @@ public TaskStatusPlus(
       @JsonProperty("id") String id,
       @JsonProperty("createdTime") DateTime createdTime,
       @JsonProperty("queueInsertionTime") DateTime queueInsertionTime,
-      @JsonProperty("state") @Nullable TaskState state,
+      @JsonProperty("statusCode") @Nullable TaskState state,
       @JsonProperty("duration") @Nullable Long duration,
       @JsonProperty("location") TaskLocation location
   )
@@ -74,7 +75,8 @@ public DateTime getQueueInsertionTime()
     return queueInsertionTime;
   }
 
-  @JsonProperty
+  @Nullable
+  @JsonProperty("statusCode")
   public TaskState getState()
   {
     return state;
@@ -91,4 +93,40 @@ public TaskLocation getLocation()
   {
     return location;
   }
+
+  @Override
+  public boolean equals(Object o)
+  {
+    if (this == o) {
+      return true;
+    }
+
+    if (o == null || getClass() != o.getClass()) {
+      return false;
+    }
+
+    final TaskStatusPlus that = (TaskStatusPlus) o;
+    if (!id.equals(that.id)) {
+      return false;
+    }
+    if (!createdTime.equals(that.createdTime)) {
+      return false;
+    }
+    if (!queueInsertionTime.equals(that.queueInsertionTime)) {
+      return false;
+    }
+    if (!Objects.equals(state, that.state)) {
+      return false;
+    }
+    if (!Objects.equals(duration, that.duration)) {
+      return false;
+    }
+    return location.equals(that.location);
+  }
+
+  @Override
+  public int hashCode()
+  {
+    return Objects.hash(id, createdTime, queueInsertionTime, state, duration, location);
+  }
 }

api/src/test/java/io/druid/indexer/TaskStatusPlusTest.java

@@ -0,0 +1,87 @@
+/*
+ * Licensed to Metamarkets Group Inc. (Metamarkets) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. Metamarkets licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+package io.druid.indexer;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.core.JsonToken;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.deser.std.StdDeserializer;
+import com.fasterxml.jackson.databind.module.SimpleModule;
+import com.fasterxml.jackson.databind.ser.std.ToStringSerializer;
+import io.druid.java.util.common.DateTimes;
+import org.joda.time.DateTime;
+import org.junit.Assert;
+import org.junit.Test;
+
+import java.io.IOException;
+
+public class TaskStatusPlusTest
+{
+  @Test
+  public void testSerde() throws IOException
+  {
+    final ObjectMapper mapper = new ObjectMapper();
+    mapper.registerModule(
+        new SimpleModule()
+            .addDeserializer(DateTime.class, new DateTimeDeserializer())
+            .addSerializer(DateTime.class, ToStringSerializer.instance)
+    );
+    final TaskStatusPlus status = new TaskStatusPlus(
+        "testId",
+        DateTimes.nowUtc(),
+        DateTimes.nowUtc(),
+        TaskState.RUNNING,
+        1000L,
+        TaskLocation.create("testHost", 1010, -1)
+    );
+    final String json = mapper.writeValueAsString(status);
+    Assert.assertEquals(status, mapper.readValue(json, TaskStatusPlus.class));
+  }
+
+  // Copied from io.druid.jackson.JodaStuff
+  private static class DateTimeDeserializer extends StdDeserializer<DateTime>
+  {
+    public DateTimeDeserializer()
+    {
+      super(DateTime.class);
+    }
+
+    @Override
+    public DateTime deserialize(JsonParser jp, DeserializationContext ctxt)
+        throws IOException, JsonProcessingException
+    {
+      JsonToken t = jp.getCurrentToken();
+      if (t == JsonToken.VALUE_NUMBER_INT) {
+        return DateTimes.utc(jp.getLongValue());
+      }
+      if (t == JsonToken.VALUE_STRING) {
+        String str = jp.getText().trim();
+        if (str.length() == 0) { // [JACKSON-360]
+          return null;
+        }
+        // make sure to preserve time zone information when parsing timestamps
+        return DateTimes.ISO_DATE_OR_TIME_WITH_OFFSET.parse(str);
+      }
+      throw ctxt.mappingException(getValueClass());
+    }
+  }
+}