From bff6899001da0d444ddf97ed2f19a98464a304e5 Mon Sep 17 00:00:00 2001
From: Andrei Zavada <johnhommer@gmail.com>
Date: Fri, 15 Apr 2016 09:04:22 +0300
Subject: [PATCH] decode get_coverage return value

---
 src/riakc_ts.erl | 21 +++++++++++++--------
 1 file changed, 13 insertions(+), 8 deletions(-)

diff --git a/src/riakc_ts.erl b/src/riakc_ts.erl
index d1332ca9..9ba3c4e2 100644
--- a/src/riakc_ts.erl
+++ b/src/riakc_ts.erl
@@ -90,14 +90,19 @@ query_common(Pid, Query, Interpolations, Cover, Options)
 
 
 %% @doc Generate a parallel coverage plan for the specified query
--spec get_coverage(Pid::pid(),
-                   Table::binary(),
-                   QueryText::binary()) -> {ok, Entries::[term()]} | {error, term()} | {'EXIT', any()}.
-get_coverage(Pid, Table, QueryText) ->
-    server_call(Pid,
-                #tscoveragereq{query = #tsinterpolation{base = iolist_to_binary(QueryText)},
-                               replace_cover = undefined,
-                               table = iolist_to_binary(Table)}).
+-spec get_coverage(pid(), table_name(), QueryText::iolist()) ->
+                          {ok, Entries::[term()]} | {error, term()}.
+get_coverage(Pid, Table, Query) ->
+    Message =
+        #tscoveragereq{query = #tsinterpolation{base = iolist_to_binary(Query)},
+                       replace_cover = undefined,
+                       table = iolist_to_binary(Table)},
+    case server_call(Pid, Message) of
+        {ok, Entries} ->
+            {ok, riak_pb_ts_codec:decode_cover_list(Entries)};
+        {error, Reason} ->
+            {error, Reason}
+    end.
 
 
 -spec put(pid(),