Source

riak / src / jiak_util.erl

Andy Gross 546dc86 














Andy Gross 3b3ec2d 
Bryan Fink 86b709c 
Andy Gross 3b3ec2d 
Bryan Fink 86b709c 
Andy Gross 3b3ec2d 
Bryan Fink 0b02326 
Andy Gross 3b3ec2d 
Andy Gross c5d03df 
Bryan Fink 86b709c 
Andy Gross c5d03df 
Andy Gross 3b3ec2d 
Bryan Fink 86b709c 




Andy Gross 3b3ec2d 

Bryan Fink 0b02326 
Andy Gross 3b3ec2d 
Bryan Fink 0b02326 
Bryan Fink 6ba6bb1 
Bryan Fink 0b02326 
Bryan Fink 6ba6bb1 
Bryan Fink 86b709c 

Andy Gross 3b3ec2d 

Bryan Fink 86b709c 


Bryan Fink 0b02326 
Andy Gross 3b3ec2d 

Bryan Fink 0b02326 
Andy Gross 3b3ec2d 
Bryan Fink 0b02326 



Andy Gross c5d03df 
Andy Gross dadc0d2 






Bryan Fink 0b02326 

Andy Gross dadc0d2 






Bryan Fink 6ba6bb1 



















Bryan Fink 0b02326 















Andy Gross c5d03df 


Bryan Fink 0b02326 
%% This file is provided to you under the Apache License,
%% Version 2.0 (the "License"); you may not use this file
%% except in compliance with the License.  You may obtain
%% a copy of the License at

%%   http://www.apache.org/licenses/LICENSE-2.0

%% Unless required by applicable law or agreed to in writing,
%% software distributed under the License is distributed on an
%% "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
%% KIND, either express or implied.  See the License for the
%% specific language governing permissions and limitations
%% under the License.    

%% @doc Utilities for jiak_resource and jiak_object.
-module(jiak_util).
-export([jiak_default_props/0,
         jiak_module_for_bucket/1, 
         extract_bucket_props/1,
         get_jiak_module/1, 
         bucket_from_reqdata/1]).

-include_lib("eunit/include/eunit.hrl").
-include("jiak.hrl").

%% @private
jiak_default_props() ->
    [{allowed_fields, ?JIAK_SCHEMA_WILDCARD},
     {required_fields, []},
     {read_mask, ?JIAK_SCHEMA_WILDCARD},
     {write_mask, ?JIAK_SCHEMA_WILDCARD}].

%% @private
jiak_module_for_bucket(BucketName) when is_binary(BucketName) ->
    BucketProps = riak_bucket:get_bucket(BucketName),
    case proplists:lookup(bucket_mod, BucketProps) of
        {bucket_mod, Module} when Module /= undefined ->
            Module;
        _ ->
            jiak_default:new(
              extract_bucket_props(BucketProps))
    end.

extract_bucket_props(BucketProps) ->
    [ {Prop, proplists:get_value(Prop, BucketProps, Default)}
      || {Prop, Default} <- jiak_default_props() ].

%% @private
get_jiak_module(ReqData) ->
    jiak_module_for_bucket(bucket_from_reqdata(ReqData)).

%% @spec bucket_from_reqdata(webmachine:wrq()) -> binary()
%% @doc Extract the bucket name, as a binary, from the request URI.
bucket_from_reqdata(RD) ->
    list_to_binary(mochiweb_util:unquote(wrq:path_info(bucket, RD))).

dynamic_bucket_test() ->
    riak_ring_manager:start_link(test),
    riak_eventer:start_link(test),    
    BucketProps = [{allowed_fields, [<<"test">>]},
                   {required_fields, []},
                   {read_mask, [<<"test">>]},
                   {write_mask, [<<"test">>]}],
    riak_bucket:set_bucket(<<"dynamic_bucket_test">>, BucketProps),
    Mod = jiak_module_for_bucket(<<"dynamic_bucket_test">>),
    ?assertEqual([<<"test">>], Mod:allowed_fields()),
    ?assertEqual([], Mod:required_fields()),
    ?assertEqual([<<"test">>], Mod:read_mask()),
    ?assertEqual([<<"test">>], Mod:write_mask()),
    riak_ring_manager:stop(),
    riak_eventer:stop().

%% just like dynamic_bucket_test, but ensuring that
%% {bucket_mod, undefined} gets treated as if bucket_mod
%% were not set.
dynamic_bucket_undefined_test() ->
    riak_ring_manager:start_link(test),
    riak_eventer:start_link(test),    
    BucketProps = [{bucket_mod, undefined},
                   {allowed_fields, [<<"test">>]},
                   {required_fields, []},
                   {read_mask, [<<"test">>]},
                   {write_mask, [<<"test">>]}],
    riak_bucket:set_bucket(<<"dynamic_bucket_test">>, BucketProps),
    Mod = jiak_module_for_bucket(<<"dynamic_bucket_test">>),
    ?assertEqual([<<"test">>], Mod:allowed_fields()),
    ?assertEqual([], Mod:required_fields()),
    ?assertEqual([<<"test">>], Mod:read_mask()),
    ?assertEqual([<<"test">>], Mod:write_mask()),
    riak_ring_manager:stop(),
    riak_eventer:stop().

module_bucket_test() ->
    riak_ring_manager:start_link(test),
    riak_eventer:start_link(test),    
    BucketProps = [{bucket_mod, jiak_example}],
    riak_bucket:set_bucket(<<"module_bucket_test">>, BucketProps),
    Mod = jiak_module_for_bucket(<<"module_bucket_test">>),
    ?assertEqual([<<"foo">>,<<"bar">>,<<"baz">>,<<"quux">>],
                 Mod:allowed_fields()),
    ?assertEqual([<<"foo">>], Mod:required_fields()),
    ?assertEqual([<<"foo">>,<<"bar">>], Mod:read_mask()),
    ?assertEqual([<<"foo">>,<<"baz">>], Mod:write_mask()),
    riak_ring_manager:stop(),
    riak_eventer:stop().
    

bucket_from_uri_test() ->
    PI = dict:store(bucket, "foo", dict:new()),
    RD0 = wrq:create('PUT', "1.1", "/jiak/foo", mochiweb_headers:empty()),
    RD = wrq:load_dispatch_data(PI, none, none, none, none, RD0),
    ?assertEqual(<<"foo">>, bucket_from_reqdata(RD)).
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.