diff options
author | Alexey Shchepin <alexey@process-one.net> | 2012-11-06 17:58:08 +0200 |
---|---|---|
committer | Evgeniy Khramtsov <ekhramtsov@process-one.net> | 2014-07-10 13:15:15 +0400 |
commit | a4b02c38db131285aa06da9ecad743f219781d6f (patch) | |
tree | 7e78dba5be6939ad5b8b92cb4c6c15a57f9d26a3 /src/gen_mod.erl | |
parent | Preliminary Riak support (diff) |
Updated riak support
Diffstat (limited to 'src/gen_mod.erl')
-rw-r--r-- | src/gen_mod.erl | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/src/gen_mod.erl b/src/gen_mod.erl index 6dac8cde..00a71674 100644 --- a/src/gen_mod.erl +++ b/src/gen_mod.erl @@ -209,22 +209,26 @@ get_opt_host(Host, Opts, Default) -> Val = get_opt(host, Opts, fun iolist_to_binary/1, Default), ejabberd_regexp:greplace(Val, <<"@HOST@">>, Host). --spec db_type(opts()) -> odbc | mnesia. +-spec db_type(opts()) -> odbc | mnesia | riak. db_type(Opts) -> get_opt(db_type, Opts, fun(odbc) -> odbc; (internal) -> mnesia; - (mnesia) -> mnesia end, + (mnesia) -> mnesia; + (riak) -> riak + end, mnesia). --spec db_type(binary(), atom()) -> odbc | mnesia. +-spec db_type(binary(), atom()) -> odbc | mnesia | riak. db_type(Host, Module) -> get_module_opt(Host, Module, db_type, fun(odbc) -> odbc; (internal) -> mnesia; - (mnesia) -> mnesia end, + (mnesia) -> mnesia; + (riak) -> riak + end, mnesia). -spec loaded_modules(binary()) -> [atom()]. |