Use ByteString

SeldonIO · Apr 2, 2020 · 61047a5 · 61047a5
1 parent 2d89b97
commit 61047a5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/engine/src/main/java/io/seldon/engine/api/rest/RestClientController.java b/engine/src/main/java/io/seldon/engine/api/rest/RestClientController.java
@@ -16,6 +16,7 @@
 package io.seldon.engine.api.rest;
 
 import com.fasterxml.jackson.databind.ObjectMapper;
+import com.google.protobuf.ByteString;
 import com.google.protobuf.InvalidProtocolBufferException;
 import io.micrometer.core.annotation.Timed;
 import io.opentracing.Span;
@@ -175,7 +176,7 @@ public ResponseEntity<String> predictions_binary(RequestEntity<InputStream> requ
     try {
       ObjectMapper mapper = new ObjectMapper();
       Map<String, Object> protoBody = new HashMap<String, Object>() {{
-        put("binData", toByteArray(requestEntity.getBody()));
+        put("binData", ByteString.copyFrom(toByteArray(requestEntity.getBody())));
       }};
       return _predictions(mapper.writeValueAsString(protoBody));
     } catch (IOException e) {