Sha256: 6d0aa9c4010da1f46f3ec8d451324b90d92bc53ca1c51168f6fe2888ae527ae5

Contents?: true

Size: 1.88 KB

Versions: 19

Compression:

Stored size: 1.88 KB

Contents

raise "Couch terminus not supported without couchrest gem" unless Puppet.features.couchdb?

require 'couchrest'
class Puppet::Indirector::Couch < Puppet::Indirector::Terminus

  # The CouchRest database instance. One database instance per Puppet runtime
  # should be sufficient.
  #
  def self.db; @db ||= CouchRest.database! Puppet[:couchdb_url] end
  def db; self.class.db end

  def find(request)
    attributes_of get(request)
  end

  # Create or update the couchdb document with the request's data hash.
  #
  def save(request)
    raise ArgumentError, "PUT does not accept options" unless request.options.empty?
    update(request) || create(request)
  end

  private

  # RKH:TODO: Do not depend on error handling, check if the document exists
  # first. (Does couchrest support this?)
  #
  def get(request)
    db.get(id_for(request))
  rescue RestClient::ResourceNotFound
    Puppet.debug "No couchdb document with id: #{id_for(request)}"
    return nil
  end

  def update(request)
    doc = get request
    return unless doc
    doc.merge!(hash_from(request))
    doc.save
    true
  end

  def create(request)
    db.save_doc hash_from(request)
  end

  # The attributes hash that is serialized to CouchDB as JSON. It includes
  # metadata that is used to help aggregate data in couchdb. Add
  # model-specific attributes in subclasses.
  #
  def hash_from(request)
    {
      "_id"         => id_for(request),
      "puppet_type" => document_type_for(request)
    }
  end

  # The couchdb response stripped of metadata, used to instantiate the model
  # instance that is returned by save.
  #
  def attributes_of(response)
    response && response.reject{|k,v| k =~ /^(_rev|puppet_)/ }
  end

  def document_type_for(request)
    request.indirection_name
  end

  # The id used to store the object in couchdb. Implemented in subclasses.
  #
  def id_for(request)
    raise NotImplementedError
  end

end

Version data entries

19 entries across 19 versions & 1 rubygems

Version Path
puppet-2.6.18 lib/puppet/indirector/couch.rb
puppet-2.6.17 lib/puppet/indirector/couch.rb
puppet-2.6.16 lib/puppet/indirector/couch.rb
puppet-2.6.15 lib/puppet/indirector/couch.rb
puppet-2.6.14 lib/puppet/indirector/couch.rb
puppet-2.6.13 lib/puppet/indirector/couch.rb
puppet-2.6.12 lib/puppet/indirector/couch.rb
puppet-2.6.11 lib/puppet/indirector/couch.rb
puppet-2.6.10 lib/puppet/indirector/couch.rb
puppet-2.6.9 lib/puppet/indirector/couch.rb
puppet-2.6.8 lib/puppet/indirector/couch.rb
puppet-2.6.7 lib/puppet/indirector/couch.rb
puppet-2.6.6 lib/puppet/indirector/couch.rb
puppet-2.6.5 lib/puppet/indirector/couch.rb
puppet-2.6.4 lib/puppet/indirector/couch.rb
puppet-2.6.3 lib/puppet/indirector/couch.rb
puppet-2.6.2 lib/puppet/indirector/couch.rb
puppet-2.6.1 lib/puppet/indirector/couch.rb
puppet-2.6.0 lib/puppet/indirector/couch.rb