class QM::MongoStore
Public Class Methods
new(opts = {})
click to toggle source
# File lib/qm/defs-mongo.rb, line 103 def initialize(opts = {}) # This constructor function needs documentation. @settings = opts end
Public Instance Methods
close()
click to toggle source
# File lib/qm/defs-mongo.rb, line 17 def close() # This function needs documentation. @api_db.connection.close if @api_db.respond_to?('connection') @log_db.connection.close if @log_db.respond_to?('connection') return end
connect_api_store(opts = {})
click to toggle source
# File lib/qm/defs-mongo.rb, line 24 def connect_api_store(opts = {}) # This function needs documentation. if (opts.has_key?(:mongo)) then @api_db ||= Mongo::MongoClient.from_uri(opts[:mongo]).db @api_db.collection('avars').ensure_index({ box: Mongo::ASCENDING, key: Mongo::ASCENDING }, { unique: true }) @api_db.collection('avars').ensure_index('exp_date', { expireAfterSeconds: 0 }) end return @api_db end
connect_log_store(opts = {})
click to toggle source
# File lib/qm/defs-mongo.rb, line 41 def connect_log_store(opts = {}) # This function needs documentation. if (opts.has_key?(:mongo)) then @log_db ||= Mongo::MongoClient.from_uri(opts[:mongo]).db end return @log_db end
get_avar(params)
click to toggle source
# File lib/qm/defs-mongo.rb, line 49 def get_avar(params) # This helper function retrieves an avar's representation if it # exists, and it also updates the "expiration date" of the avar in # the database so that data still being used for computations will # not be removed. x = @api_db.collection('avars').find_and_modify({ fields: { _id: 0, body: 1 }, query: { box: params[0], key: params[1] }, update: { # NOTE: The hash below must use `=>` (not `:`) in JRuby, as # of version 1.7.18, but QM won't be supporting JRuby anyway # until (a.) JRuby 9000 is stable and (b.) I understand Puma. '$set': { exp_date: Time.now + @settings.avar_ttl } }, upsert: false }) return (x.nil?) ? '{}' : x['body'] end
get_list(params)
click to toggle source
# File lib/qm/defs-mongo.rb, line 76 def get_list(params) # This helper function retrieves a list of "key" properties for avars # in the database that have a "status" property, because those are # assumed to represent task descriptions. The function returns the # list as a stringified JSON array in which order is not important. opts = { fields: { _id: 0, key: 1 } } query = { box: params[0], exp_date: { '$gt': Time.now }, status: params[1] } x = [] @api_db.collection('avars').find(query, opts).each do |doc| # This block appends each task's key to a running list, but the # the order in which the keys are added is *not* sorted. x.push(doc['key']) end return (x.length == 0) ? '[]' : x.to_json end
log(request)
click to toggle source
# File lib/qm/defs-mongo.rb, line 108 def log(request) # This helper function inserts a new document into MongoDB after each # request. Eventually, this function will be replaced by one that # delegates to a custom `log` function like the Node.js version. @log_db.collection('traffic').insert({ host: request.host, method: request.request_method, timestamp: Time.now, url: request.fullpath }, { w: 0 }) return end
set_avar(params)
click to toggle source
# File lib/qm/defs-mongo.rb, line 123 def set_avar(params) # This helper function writes an avar to the database by "upserting" # a Mongo document that represents it. doc = { body: params.last, box: params[0], exp_date: Time.now + @settings.avar_ttl, key: params[1] } doc['status'] = params[2] if params.length == 4 opts = { multi: false, upsert: true, w: 1 } query = { box: params[0], key: params[1] } @api_db.collection('avars').update(query, doc, opts) return end