Fix routing_key type

lpgauth · Feb 18, 2025 · e9171c5 · e9171c5
1 parent 2a1f341
commit e9171c5
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/include/marina.hrl b/include/marina.hrl
@@ -69,14 +69,15 @@
     page_size => pos_integer(),
     paging_state => binary(),
     pid => pid(),
-    routing_key => binary(),
+    routing_key => routing_key(),
     skip_metadata => boolean(),
     timeout => pos_integer(),
     values => values()
 }.
 
 -type result() :: #result {}.
 -type result_metadata() :: #result_metadata {}.
+-type routing_key() :: integer() | binary().
 -type statement_id() :: binary().
 -type stream() :: 0..32768.
 -type value() :: binary().

diff --git a/src/marina_ring.erl b/src/marina_ring.erl
@@ -24,7 +24,7 @@ build(Nodes) ->
     Ranges = ranges(lists:usort(lists:flatten(Ring))),
     marina_compiler:ring_utils(Ranges).
 
--spec lookup(binary()) ->
+-spec lookup(routing_key()) ->
     atom().
 
 lookup(Key) ->

diff --git a/src/marina_token.erl b/src/marina_token.erl
@@ -6,7 +6,7 @@
 ]).
 
 %% public
--spec m3p(integer() | binary()) ->
+-spec m3p(routing_key()) ->
     integer().
 
 m3p(Key) when is_integer(Key) ->