Question

我正在使用内部数据库编写应用程序。我正在使用MongoId。它只是一个只有一个条目的数据库。我正在这个数据库中保存一个令牌

api.rb

    def get_wink_token
      retrieve_token.present? ? retrieve_token : new_token
    end

    def new_token
      RestClient.get "#{ENV['DOMAIN']}/oauth2/authorize?response_type=code&client_id=#{ENV['CLIENT_ID']}&redirect_uri=#{ENV['REDIRECT_URI']}"
      token_credentials = RestClient.post "#{ENV['DOMAIN']}/oauth2/token", credentials, headers
      access_token = JSON.parse(token_credentials)['data']['access_token']
      TokenDb.any_in(:name => 'Token').destroy_all
      TokenDb.create(name:'Token', token:access_token)
      access_token
    end

    def retrieve_token
      TokenDb.where(:name => 'Token').present? ? TokenDb.where(:name => 'Token').first[:token] : nil
    end

通过new_token获取令牌的过程运行正常。我的问题是我在执行TokenDb.where时遇到了崩溃。现在产生了崩溃。

TokenDB类的定义如下：

class TokenDb
  include Mongoid::Document
  include Mongoid::Timestamps
  include Mongoid::Attributes::Dynamic

  field :name,      type:String
  field :token,     type:String

  validates_presence_of :name, :token
end

我要做的是检查数据库tokenDb是否有一个名为Token的条目并检索数据，如果没有，我生成一个新密钥

2017-03-13 23:18:08 - NoMethodError - undefined method `each' for nil   /Users/sebastien/.rvm/gems/ruby-2.3.1/gems/activesupport-5.0.1/lib/active_support/core_ext/object/blank.rb:22:in `present?'
    /Users/sebastien/smarthome/models/credentials.rb:23:in `retrieve_token'

问题发生在TokenDb.where ...行

有什么想法吗？

Answer 1

您可以简单地使用find_by并在不存在的情况下解救该异常：

def get_wink_token
  TokenDb.find_by(name: 'token').token
rescue Mongoid::Errors::DocumentNotFound
  create_token
end

def create_token
  RestClient.get "#{ENV['DOMAIN']}/oauth2/authorize?response_type=code&client_id=#{ENV['CLIENT_ID']}&redirect_uri=#{ENV['REDIRECT_URI']}"
  token_credentials = RestClient.post "#{ENV['DOMAIN']}/oauth2/token", credentials, headers
  access_token = JSON.parse(token_credentials)['data']['access_token']
  TokenDb.any_in(:name => 'Token').destroy_all
  TokenDb.create(name: 'Token', token: access_token)
  access_token
end

虽然仅仅为此使用数据库是过度的......

Mongoid访问问题

1 个答案: