fix issue #505 : Error messages from server not parsed correctly when using msgpack

Author: lxhoan

src/main/java/org/influxdb/InfluxDBException.java

@@ -1,5 +1,12 @@
 package org.influxdb;
 
+import java.io.InputStream;
+
+import org.msgpack.core.MessagePack;
+import org.msgpack.core.MessageUnpacker;
+import org.msgpack.value.ImmutableMapValue;
+import org.msgpack.value.impl.ImmutableStringValueImpl;
+
 import com.squareup.moshi.JsonAdapter;
 import com.squareup.moshi.Moshi;
 
@@ -168,4 +175,20 @@ public static InfluxDBException buildExceptionForErrorState(final String errorBo
       return new InfluxDBException(errorBody);
     }
   }
+
+  /**
+   * Create corresponding InfluxDBException from the message pack error body.
+   * @param messagePackErrorBody
+   * @return
+   */
+  public static InfluxDBException buildExceptionForErrorState(final InputStream messagePackErrorBody) {
+    try {
+      MessageUnpacker unpacker = MessagePack.newDefaultUnpacker(messagePackErrorBody);
+      ImmutableMapValue mapVal = (ImmutableMapValue) unpacker.unpackValue();
+      return InfluxDBException.buildExceptionFromErrorMessage(
+          mapVal.map().get(new ImmutableStringValueImpl("error")).toString());
+    } catch (Exception e) {
+      return new InfluxDBException(e);
+    }
+  }
 }

src/main/java/org/influxdb/impl/InfluxDBImpl.java

@@ -696,7 +696,11 @@ private <T> T execute(final Call<T> call) {
         return response.body();
       }
       try (ResponseBody errorBody = response.errorBody()) {
-        throw InfluxDBException.buildExceptionForErrorState(errorBody.string());
+        if (messagePack) {
+          throw InfluxDBException.buildExceptionForErrorState(errorBody.byteStream());
+        } else {
+          throw InfluxDBException.buildExceptionForErrorState(errorBody.string());
+        }
       }
     } catch (IOException e) {
       throw new InfluxDBIOException(e);

src/test/java/org/influxdb/InfluxDBExceptionTest.java

@@ -0,0 +1,30 @@
+package org.influxdb;
+
+import org.influxdb.InfluxDBException.DatabaseNotFoundException;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+import org.junit.platform.runner.JUnitPlatform;
+import org.junit.runner.RunWith;
+
+/**
+ * Test cases for InfluxDBException
+ *
+ * @author hoan.le [at] bonitoo.io
+ *
+ */
+
+@RunWith(JUnitPlatform.class)
+public class InfluxDBExceptionTest {
+
+  @Test
+  public void testBuildExceptionForMessagePackErrorState() {
+    DatabaseNotFoundException dbex = (DatabaseNotFoundException) InfluxDBException
+        .buildExceptionForErrorState(InfluxDBExceptionTest.class.getResourceAsStream("msgpack_errorBody.bin"));
+    
+    Assertions.assertEquals("database not found: \"abc\"", dbex.getMessage());
+
+    InfluxDBException ex = InfluxDBException.buildExceptionForErrorState(InfluxDBExceptionTest.class.getResourceAsStream("invalid_msgpack_errorBody.bin"));
+    Assertions.assertTrue(ex.getCause() instanceof ClassCastException);
+    
+  }
+}

src/test/resources/org/influxdb/invalid_msgpack_errorBody.bin

@@ -0,0 +1 @@
+e not found: "abc"

src/test/resources/org/influxdb/msgpack_errorBody.bin

@@ -0,0 +1 @@
+��error�database not found: "abc"