updates to work w/ 0.11 btree option refactoring

4 files changed, 40 insertions, 132 deletions

diff --git a/src/fabric_rpc.erl b/src/fabric_rpc.erl
index 26aebfb5..c9bfb7d2 100644
--- a/src/fabric_rpc.erl
+++ b/src/fabric_rpc.erl
@@ -13,7 +13,6 @@
     offset = nil,
     total_rows,
     reduce_fun = fun couch_db:enum_docs_reduce_to_count/1,
-    stop_fun,
     group_level = 0
 }).
 
@@ -25,21 +24,27 @@ all_docs(DbName, #view_query_args{keys=nil} = QueryArgs) ->
     #view_query_args{
         start_key = StartKey,
         start_docid = StartDocId,
+        end_key = EndKey,
+        end_docid = EndDocId,
         limit = Limit,
         skip = Skip,
         include_docs = IncludeDocs,
-        direction = Dir
+        direction = Dir,
+        inclusive_end = Inclusive
     } = QueryArgs,
-    StartId = if is_binary(StartKey) -> StartKey; true -> StartDocId end,
     {ok, Total} = couch_db:get_doc_count(Db),
     Acc0 = #view_acc{
         db = Db,
         include_docs = IncludeDocs,
         limit = Limit+Skip,
-        total_rows = Total,
-        stop_fun = all_docs_stop_fun(QueryArgs)
+        total_rows = Total
     },
-    Options = [{start_key, StartId}, {dir, Dir}],
+    EndKeyType = if Inclusive -> end_key; true -> end_key_gt end,
+    Options = [
+        {dir, Dir},
+        {start_key, if is_binary(StartKey) -> StartKey; true -> StartDocId end},
+        {EndKeyType, if is_binary(EndKey) -> EndKey; true -> EndDocId end}
+    ],
     {ok, _, Acc} = couch_db:enum_docs(Db, fun view_fold/3, Acc0, Options),
     final_response(Total, Acc#view_acc.offset).
 
@@ -68,17 +73,13 @@ changes(DbName, Args0, StartSeq) ->
 map_view(DbName, DDoc, ViewName, QueryArgs) ->
     {ok, Db} = couch_db:open(DbName, []),
     #view_query_args{
-        start_key = StartKey,
-        start_docid = StartDocId,
         limit = Limit,
         skip = Skip,
         keys = Keys,
         include_docs = IncludeDocs,
-        direction = Dir,
         stale = Stale,
         view_type = ViewType
     } = QueryArgs,
-    Start = {StartKey, StartDocId},
     MinSeq = if Stale == ok -> 0; true -> couch_db:get_update_seq(Db) end,
     Group0 = couch_view_group:design_doc_to_view_group(Db, DDoc),
     {ok, Pid} = gen_server:call(couch_view, {get_group_server, DbName, Group0}),
@@ -90,21 +91,18 @@ map_view(DbName, DDoc, ViewName, QueryArgs) ->
         include_docs = IncludeDocs,
         limit = Limit+Skip,
         total_rows = Total,
-        reduce_fun = fun couch_view:reduce_to_count/1,
-        stop_fun = default_stop_fun(QueryArgs)
+        reduce_fun = fun couch_view:reduce_to_count/1
     },
     case Keys of
     nil ->
-        Options = [{start_key, Start}, {dir, Dir}],
+        Options = couch_httpd_view:make_key_options(QueryArgs),
         {ok, _, Acc} = couch_view:fold(View, fun view_fold/3, Acc0, Options);
     _ ->
         Acc = lists:foldl(fun(Key, AccIn) ->
-            KeyStart = {Key, StartDocId},
-            KeyStop = default_stop_fun(QueryArgs#view_query_args{start_key=Key,
-                end_key=Key}),
-            Options = [{start_key, KeyStart}, {dir, Dir}],
-            {_Go, _, Out} = couch_view:fold(View, fun view_fold/3,
-                AccIn#view_acc{stop_fun = KeyStop}, Options),
+            KeyArgs = QueryArgs#view_query_args{start_key=Key, end_key=Key},
+            Options = couch_httpd_view:make_key_options(KeyArgs),
+            {_Go, _, Out} = couch_view:fold(View, fun view_fold/3, AccIn,
+                Options),
             Out
         end, Acc0, Keys)
     end,
@@ -113,10 +111,6 @@ map_view(DbName, DDoc, ViewName, QueryArgs) ->
 reduce_view(DbName, Group0, ViewName, QueryArgs) ->
     {ok, Db} = couch_db:open(DbName, []),
     #view_query_args{
-        start_key = StartKey,
-        start_docid = StartDocId,
-        end_key = EndKey,
-        end_docid = EndDocId,
         group_level = GroupLevel,
         limit = Limit,
         skip = Skip,
@@ -135,14 +129,16 @@ reduce_view(DbName, Group0, ViewName, QueryArgs) ->
     Options0 = [{key_group_fun, GroupFun}, {dir, Dir}],
     case Keys of
     nil ->
-        Options = [{start_key, {StartKey,StartDocId}},
-            {end_key, {EndKey,EndDocId}} | Options0],
+        Options0 = couch_httpd_view:make_key_options(QueryArgs),
+        Options = [{key_group_fun, GroupFun} | Options0],
         couch_view:fold_reduce(ReduceView, fun reduce_fold/3, Acc0, Options);
     _ ->
-        
-        [couch_view:fold_reduce(ReduceView, fun reduce_fold/3, Acc0,
-            [{start_key,{K,StartDocId}}, {end_key,{K,EndDocId}} | Options0])
-            || K <- Keys]
+        lists:map(fun(Key) ->
+            KeyArgs = QueryArgs#view_query_args{start_key=Key, end_key=Key},
+            Options0 = couch_httpd_view:make_key_options(KeyArgs),
+            Options = [{key_group_fun, GroupFun} | Options0],
+            couch_view:fold_reduce(ReduceView, fun reduce_fold/3, Acc0, Options)
+        end, Keys)
     end,
     rexi:reply(complete).
 
@@ -231,35 +227,20 @@ view_fold({{Key,Id}, Value}, _Offset, Acc) ->
     #view_acc{
         db = Db,
         limit = Limit,
-        include_docs = IncludeDocs,
-        stop_fun = PassedEnd
+        include_docs = IncludeDocs
     } = Acc,
-    case PassedEnd(Key, Id) of
-    true ->
-        {stop, Acc};
-    false ->
-        Doc = if not IncludeDocs -> undefined; true ->
-            case couch_db:open_doc(Db, Id, []) of
-            {not_found, deleted} ->
-                null;
-            {not_found, missing} ->
-                undefined;
-            {ok, Doc0} ->
-                couch_doc:to_json_obj(Doc0, [])
-            end
-        end,
-        rexi:sync_reply(#view_row{key=Key, id=Id, value=Value, doc=Doc}),
-        {ok, Acc#view_acc{limit=Limit-1}}
-    end.
-
-all_docs_stop_fun(#view_query_args{direction=fwd, end_key=EndKey}) ->
-    fun(ViewKey, _) ->
-        couch_db_updater:less_docid(EndKey, ViewKey)
-    end;
-all_docs_stop_fun(#view_query_args{direction=rev, end_key=EndKey}) ->
-    fun(ViewKey, _) ->
-        couch_db_updater:less_docid(ViewKey, EndKey)
-    end.
+    Doc = if not IncludeDocs -> undefined; true ->
+        case couch_db:open_doc(Db, Id, []) of
+        {not_found, deleted} ->
+            null;
+        {not_found, missing} ->
+            undefined;
+        {ok, Doc0} ->
+            couch_doc:to_json_obj(Doc0, [])
+        end
+    end,
+    rexi:sync_reply(#view_row{key=Key, id=Id, value=Value, doc=Doc}),
+    {ok, Acc#view_acc{limit=Limit-1}}.
 
 final_response(Total, nil) ->
     case rexi:sync_reply({total_and_offset, Total, Total}) of ok ->
@@ -268,33 +249,6 @@ final_response(Total, nil) ->
 final_response(_Total, _Offset) ->
     rexi:reply(complete).
 
-default_stop_fun(#view_query_args{direction=fwd, inclusive_end=true} = Args) ->
-    #view_query_args{end_key=EndKey, end_docid=EndDocId} = Args,
-    fun(ViewKey, ViewId) ->
-        couch_view:less_json([EndKey, EndDocId], [ViewKey, ViewId])
-    end;
-default_stop_fun(#view_query_args{direction=fwd} = Args) ->
-    #view_query_args{end_key=EndKey, end_docid=EndDocId} = Args,
-    fun
-        (ViewKey, _ViewId) when ViewKey == EndKey ->
-            true;
-        (ViewKey, ViewId) ->
-            couch_view:less_json([EndKey, EndDocId], [ViewKey, ViewId])
-    end;
-default_stop_fun(#view_query_args{direction=rev, inclusive_end=true} = Args) ->
-    #view_query_args{end_key=EndKey, end_docid=EndDocId} = Args,
-    fun(ViewKey, ViewId) ->
-        couch_view:less_json([ViewKey, ViewId], [EndKey, EndDocId])
-    end;
-default_stop_fun(#view_query_args{direction=rev} = Args) ->
-    #view_query_args{end_key=EndKey, end_docid=EndDocId} = Args,
-    fun
-        (ViewKey, _ViewId) when ViewKey == EndKey ->
-            true;
-        (ViewKey, ViewId) ->
-            couch_view:less_json([ViewKey, ViewId], [EndKey, EndDocId])
-    end.
-
 group_rows_fun(exact) ->
     fun({Key1,_}, {Key2,_}) -> Key1 == Key2 end;
 group_rows_fun(0) ->
diff --git a/src/fabric_view.erl b/src/fabric_view.erl
index 2432ab40..99cf1671 100644
--- a/src/fabric_view.erl
+++ b/src/fabric_view.erl
@@ -151,20 +151,11 @@ get_next_row(#collector{reducer = RedSrc} = St) when RedSrc =/= undefined ->
         get_next_row(St#collector{keys=RestKeys})
     end;
 get_next_row(State) ->
-    #collector{
-        rows = [Row|Rest],
-        counters = Counters0,
-        stop_fun = Stop
-    } = State,
+    #collector{rows = [Row|Rest], counters = Counters0} = State,
     Worker = Row#view_row.worker,
     Counters1 = fabric_dict:update_counter(Worker, -1, Counters0),
     NewState = maybe_resume_worker(Worker, State#collector{counters=Counters1}),
-    case Stop(Row) of
-    true ->
-        throw(complete);
-    false ->
-        {Row, NewState#collector{rows = Rest}}
-    end.
+    {Row, NewState#collector{rows = Rest}}.
 
 find_next_key(nil, Dir, RowDict) ->
     case lists:sort(sort_fun(Dir), dict:fetch_keys(RowDict)) of
diff --git a/src/fabric_view_all_docs.erl b/src/fabric_view_all_docs.erl
index 99834286..196d6837 100644
--- a/src/fabric_view_all_docs.erl
+++ b/src/fabric_view_all_docs.erl
@@ -19,7 +19,6 @@ go(DbName, #view_query_args{keys=nil} = QueryArgs, Callback, Acc0) ->
         counters = fabric_dict:init(Workers, 0),
         skip = Skip,
         limit = Limit,
-        stop_fun = stop_fun(QueryArgs),
         user_acc = Acc0
     },
     try fabric_util:receive_loop(Workers, #shard.ref, fun handle_message/3,
@@ -120,15 +119,6 @@ handle_message(complete, Worker, State) ->
     fabric_view:maybe_send_row(State#collector{counters = Counters}).
 
 
-stop_fun(#view_query_args{direction=fwd, end_key=EndKey}) ->
-    fun(#view_row{id=Id}) ->
-        couch_db_updater:less_docid(EndKey, Id)
-    end;
-stop_fun(#view_query_args{direction=rev, end_key=EndKey}) ->
-    fun(#view_row{id=Id}) ->
-        couch_db_updater:less_docid(Id, EndKey)
-    end.
-
 merge_row(fwd, Row, Rows) ->
     lists:keymerge(#view_row.id, [Row], Rows);
 merge_row(rev, Row, Rows) ->
diff --git a/src/fabric_view_map.erl b/src/fabric_view_map.erl
index c314586c..b686b5fc 100644
--- a/src/fabric_view_map.erl
+++ b/src/fabric_view_map.erl
@@ -22,7 +22,6 @@ go(DbName, DDoc, View, Args, Callback, Acc0) ->
         counters = fabric_dict:init(Workers, 0),
         skip = Skip,
         limit = Limit,
-        stop_fun = stop_fun(Args),
         keys = fabric_view:keydict(Keys),
         sorted = Args#view_query_args.sorted,
         user_acc = Acc0
@@ -120,32 +119,6 @@ handle_message(complete, Worker, State) ->
     Counters = fabric_dict:update_counter(Worker, 1, State#collector.counters),
     fabric_view:maybe_send_row(State#collector{counters = Counters}).
 
-stop_fun(#view_query_args{} = QueryArgs) ->
-    #view_query_args{
-        direction = Dir,
-        inclusive_end = Inclusive,
-        end_key = EndKey,
-        end_docid = EndDocId
-    } = QueryArgs,
-    stop_fun(Dir, Inclusive, EndKey, EndDocId).
-
-stop_fun(fwd, true, EndKey, EndDocId) ->
-    fun(#view_row{key=Key, id=Id}) ->
-        couch_view:less_json([EndKey, EndDocId], [Key, Id])
-    end;
-stop_fun(fwd, false, EndKey, EndDocId) ->
-    fun(#view_row{key=K}) when K==EndKey -> true; (#view_row{key=Key, id=Id}) ->
-        couch_view:less_json([EndKey, EndDocId], [Key, Id])
-    end;
-stop_fun(rev, true, EndKey, EndDocId) ->
-    fun(#view_row{key=Key, id=Id}) ->
-        couch_view:less_json([Key, Id], [EndKey, EndDocId])
-    end;
-stop_fun(rev, false, EndKey, EndDocId) ->
-    fun(#view_row{key=K}) when K==EndKey -> true; (#view_row{key=Key, id=Id}) ->
-        couch_view:less_json([Key, Id], [EndKey, EndDocId])
-    end.
-
 merge_row(fwd, undefined, Row, Rows) ->
     lists:merge(fun(#view_row{key=KeyA, id=IdA}, #view_row{key=KeyB, id=IdB}) ->
         couch_view:less_json([KeyA, IdA], [KeyB, IdB])