Mercurial > prosody-modules
view mod_couchdb/couchdb/mod_couchdb.lua @ 3866:c0df50ce96f0
mod_rest: Handle internal http request errors early and then return
Skips over attempted parsing of the payload which usually failed since
the body is an error string like "connection refused", so this produced
useless errors.
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Sat, 25 Jan 2020 20:22:12 +0100 |
parents | 2469f779b3f7 |
children |
line wrap: on
line source
local http = require "socket.http"; local url = require "socket.url"; local couchapi = module:require("couchdb/couchapi"); local json = module:require("couchdb/json"); local couchdb_url = assert(module:get_option("couchdb_url"), "Option couchdb_url not specified"); local db = couchapi.db(couchdb_url); local function couchdb_get(key) local a,b = db:doc(key):get() print(json.encode(a)); if b == 404 then return nil; end if b == 200 then b = nil; end return a.payload,b; end local function couchdb_put(key, value) local a,b = db:doc(key):get(); return db:doc(key):put({ payload = value, _rev = a and a._rev }); end local st = require "util.stanza"; local handlers = {}; handlers.accounts = { get = function(self, user) return couchdb_get(self.host.."/"..user.."/account"); end; set = function(self, user, data) return couchdb_put(self.host.."/"..user.."/account", data); end; }; handlers.vcard = { get = function(self, user) return couchdb_get(self.host.."/"..user.."/vcard"); end; set = function(self, user, data) return couchdb_put(self.host.."/"..user.."/vcard", data); end; }; handlers.private = { get = function(self, user) return couchdb_get(self.host.."/"..user.."/private"); end; set = function(self, user, data) return couchdb_put(self.host.."/"..user.."/private", data); end; }; handlers.roster = { get = function(self, user) return couchdb_get(self.host.."/"..user.."/roster"); end; set = function(self, user, data) return couchdb_put(self.host.."/"..user.."/roster", data); end; }; ----------------------------- local driver = {}; function driver:open(datastore, typ) local handler = handlers[datastore]; if not handler then return nil; end local host = module.host; --local cache_key = host.." "..datastore; --if self.ds_cache[cache_key] then return self.ds_cache[cache_key]; end local instance = setmetatable({ host = host, datastore = datastore }, { __index = handler }); --for key,val in pairs(handler) do -- instance[key] = val; --end --if instance.init then instance:init(); end --self.ds_cache[cache_key] = instance; return instance; end module:provides("storage", driver);