96 lines
2 KiB
Ruby
96 lines
2 KiB
Ruby
class CouchRest
|
|
attr_accessor :uri, :uuid_batch_count
|
|
def initialize server = 'http://localhost:5984', uuid_batch_count = 1000
|
|
@uri = server
|
|
@uuid_batch_count = uuid_batch_count
|
|
end
|
|
|
|
# ensure that a database exists
|
|
# creates it if it isn't already there
|
|
# returns it after it's been created
|
|
def self.database! url
|
|
uri = URI.parse url
|
|
path = uri.path
|
|
uri.path = ''
|
|
cr = CouchRest.new(uri.to_s)
|
|
cr.database!(path)
|
|
end
|
|
|
|
def self.database url
|
|
uri = URI.parse url
|
|
path = uri.path
|
|
uri.path = ''
|
|
cr = CouchRest.new(uri.to_s)
|
|
cr.database(path)
|
|
end
|
|
|
|
# list all databases on the server
|
|
def databases
|
|
CouchRest.get "#{@uri}/_all_dbs"
|
|
end
|
|
|
|
def database name
|
|
CouchRest::Database.new(self, name)
|
|
end
|
|
|
|
# creates the database if it doesn't exist
|
|
def database! name
|
|
create_db(name) rescue nil
|
|
database name
|
|
end
|
|
|
|
# get the welcome message
|
|
def info
|
|
CouchRest.get "#{@uri}/"
|
|
end
|
|
|
|
# create a database
|
|
def create_db name
|
|
CouchRest.put "#{@uri}/#{name}"
|
|
database name
|
|
end
|
|
|
|
# restart the couchdb instance
|
|
def restart!
|
|
CouchRest.post "#{@uri}/_restart"
|
|
end
|
|
|
|
def next_uuid count = @uuid_batch_count
|
|
@uuids ||= []
|
|
if @uuids.empty?
|
|
@uuids = CouchRest.post("#{@uri}/_uuids?count=#{count}")["uuids"]
|
|
end
|
|
@uuids.pop
|
|
end
|
|
|
|
class << self
|
|
def put uri, doc = nil
|
|
payload = doc.to_json if doc
|
|
JSON.parse(RestClient.put(uri, payload))
|
|
end
|
|
|
|
def get uri
|
|
JSON.parse(RestClient.get(uri), :max_nesting => false)
|
|
end
|
|
|
|
def post uri, doc = nil
|
|
payload = doc.to_json if doc
|
|
JSON.parse(RestClient.post(uri, payload))
|
|
end
|
|
|
|
def delete uri
|
|
JSON.parse(RestClient.delete(uri))
|
|
end
|
|
|
|
def paramify_url url, params = nil
|
|
if params
|
|
query = params.collect do |k,v|
|
|
v = v.to_json if %w{key startkey endkey}.include?(k.to_s)
|
|
"#{k}=#{CGI.escape(v.to_s)}"
|
|
end.join("&")
|
|
url = "#{url}?#{query}"
|
|
end
|
|
url
|
|
end
|
|
end
|
|
end
|