From f26ad32b18809d81c9c01d6b007736eb0299d5f6 Mon Sep 17 00:00:00 2001
From: Adam Kocoloski <adam@cloudant.com>
Date: Mon, 14 Jun 2010 16:06:05 -0400
Subject: atoms are not ejson keys, use bins for http compatibility

---
 src/dbs_cache.erl  | 17 +++++++++++------
 src/partitions.erl | 16 ++++++++--------
 2 files changed, 19 insertions(+), 14 deletions(-)

(limited to 'src')

diff --git a/src/dbs_cache.erl b/src/dbs_cache.erl
index 96319802..f4faab3a 100644
--- a/src/dbs_cache.erl
+++ b/src/dbs_cache.erl
@@ -66,21 +66,26 @@ cache_dbs() ->
     end.
 
 cache_map(Id, Props) ->
-    Map = couch_util:get_value(map, Props),
-    lists:foreach(fun({[{node,Node},{b,Beg},{e,End}]}) ->
+    Map = couch_util:get_value(<<"map">>, Props),
+    lists:foreach(fun({[{<<"node">>,Node},{<<"b">>,Beg},{<<"e">>,End}]}) ->
         Part = #shard{
             name = partitions:shard_name(Beg, Id),
             dbname = Id,
-            node = Node,
+            node = to_atom(Node),
             range = [Beg,End]
         },
         ets:insert(partitions, Part)
     end, Map).
 
 cache_nodes(Id, Props) ->
-    Nodes = couch_util:get_value(nodes, Props),
-    lists:foreach(fun({[{order,Order},{node, Node},{options,Opts}]}) ->
-        ets:insert(memnodes, {Id, {Order, Node, Opts}})
+    Nodes = couch_util:get_value(<<"nodes">>, Props),
+    lists:foreach(fun({[{<<"order">>,Order},{<<"node">>, Node},{<<"options">>,Opts}]}) ->
+        ets:insert(memnodes, {Id, {Order, to_atom(Node), Opts}})
     end, Nodes).
 
+to_atom(Node) when is_binary(Node) ->
+    list_to_atom(binary_to_list(Node));
+to_atom(Node) when is_atom(Node) ->
+    Node.
+
 %{ok, ets:insert(dbs_cache, {Id, Props})};
diff --git a/src/partitions.erl b/src/partitions.erl
index 4fba0011..3e024264 100644
--- a/src/partitions.erl
+++ b/src/partitions.erl
@@ -37,10 +37,10 @@ hash(Item) ->
 install_fullmap(DbName, Fullmap, FullNodes, Options) ->
     {N,Q} = db_init_constants(Options),
     Doc = {[{<<"_id">>,DbName},
-            {map, jsonify(map, Fullmap)},
-            {nodes, jsonify(nodes, FullNodes)},
-            {n,N},
-            {q,Q}]},
+            {<<"map">>, jsonify(<<"map">>, Fullmap)},
+            {<<"nodes">>, jsonify(<<"nodes">>, FullNodes)},
+            {<<"n">>,N},
+            {<<"q">>,Q}]},
     write_db_doc(Doc).
 
 for_key(DbName, Key) ->
@@ -153,13 +153,13 @@ make_map(AllNodes, [Node|RestNodes], [Part|RestParts], Acc) ->
     % add a node/part combo to the Acc
     make_map(AllNodes, RestNodes, RestParts, [{Node,Part}|Acc]).
 
-jsonify(map, Map) ->
+jsonify(<<"map">>, Map) ->
     lists:map(fun(#shard{node=Node, range=[Beg,End]}) ->
-        {[{node, Node}, {b, Beg}, {e, End}]}
+        {[{<<"node">>, Node}, {<<"b">>, Beg}, {<<"e">>, End}]}
     end, Map);
-jsonify(nodes, Nodes) ->
+jsonify(<<"nodes">>, Nodes) ->
     lists:map(fun({Order, Node, Options}) ->
-        {[{order, Order}, {node, Node}, {options, Options}]}
+        {[{<<"order">>, Order}, {<<"node">>, Node}, {<<"options">>, Options}]}
     end, Nodes).
 
 write_db_doc(EDoc) ->
-- 
cgit v1.2.3