重构access_token store

添加jsticket store

.rspec

 --format documentation --color spec --drb
+--require spec_helper

lib/weixin_authorize.rb

 require "rest-client"
-
 require "carrierwave"
-require "weixin_authorize/carrierwave/weixin_uploader"
-
 require 'yajl/json_gem'
 
+require "weixin_authorize/carrierwave/weixin_uploader"
 require "weixin_authorize/config"
 require "weixin_authorize/handler"
 require "weixin_authorize/api"
@@ -12,10 +10,18 @@ require "weixin_authorize/client"
 
 module WeixinAuthorize
 
-  # Storage
-  autoload(:Storage,       "weixin_authorize/adapter/storage")
-  autoload(:ClientStorage, "weixin_authorize/adapter/client_storage")
-  autoload(:RedisStorage,  "weixin_authorize/adapter/redis_storage")
+  # token store
+  module Token
+    autoload(:Store,       "weixin_authorize/token/store")
+    autoload(:ObjectStore, "weixin_authorize/token/object_store")
+    autoload(:RedisStore,  "weixin_authorize/token/redis_store")
+  end
+
+  module JsTicket
+    autoload(:Store,       "weixin_authorize/js_ticket/store")
+    autoload(:ObjectStore, "weixin_authorize/js_ticket/object_store")
+    autoload(:RedisStore,  "weixin_authorize/js_ticket/redis_store")
+  end
 
   OK_MSG  = "ok"
   OK_CODE = 0

lib/weixin_authorize/api/data_cube.rb

+# encoding: utf-8
+module WeixinAuthorize
+  module Api
+    module DataCube
+
+    end
+  end
+end

lib/weixin_authorize/client.rb

@@ -15,62 +15,53 @@ module WeixinAuthorize
 
     attr_accessor :app_id, :app_secret, :expired_at # Time.now + expires_in
     attr_accessor :access_token, :redis_key
-    attr_accessor :storage, :jsticket
+    attr_accessor :jsticket, :jsticket_expired_at, :jsticket_redis_key
 
     def initialize(app_id, app_secret, redis_key=nil)
       @app_id     = app_id
       @app_secret = app_secret
-      @expired_at = Time.now.to_i
-      @redis_key  = security_redis_key((redis_key || "weixin_" + app_id))
-      @storage    = Storage.init_with(self)
+      @jsticket_expired_at = @expired_at = Time.now.to_i
+      @redis_key  = security_redis_key(redis_key || "weixin_#{app_id}")
+      @jsticket_redis_key = security_redis_key("js_sdk_#{app_id}")
     end
 
     # return token
     def get_access_token
-      @storage.access_token
+      token_store.access_token
     end
 
     # 检查appid和app_secret是否有效。
     def is_valid?
-      @storage.valid?
+      token_store.valid?
     end
 
-    # TODO: refactor with difference storage
-    def js_ticket
-      raise ValidAccessTokenException if !is_valid?
-      if @jsticket.nil? || @jsticket.result["expired_at"] <= Time.now.to_i
-        @jsticket ||= get_jsticket
+    def token_store
+      Token::Store.init_with(self)
     end
-      @jsticket.result["ticket"]
+
+    def jsticket_store
+      JsTicket::Store.init_with(self)
+    end
+
+    def get_jsticket
+      jsticket_store.jsticket
     end
 
     # 获取js sdk 签名包
     def get_jssign_package(url)
       timestamp = Time.now.to_i
       noncestr = SecureRandom.hex(16)
-      string = "jsapi_ticket=#{js_ticket}&noncestr=#{noncestr}&timestamp=#{timestamp}&url=#{url}";
-      signature = Digest::SHA1.hexdigest(string)
+      str = "jsapi_ticket=#{get_jsticket}&noncestr=#{noncestr}&timestamp=#{timestamp}&url=#{url}";
+      signature = Digest::SHA1.hexdigest(str)
       {
         "appId"     => app_id,    "nonceStr"  => noncestr,
         "timestamp" => timestamp, "url"       => url,
-        "signature" => signature, "rawString" => string
+        "signature" => signature, "rawString" => str
       }
     end
 
-    private
-
-      def get_jsticket
-        ticket = http_get("/ticket/getticket", {type: 1})
-        ticket.result["expired_at"] = ticket.result["expires_in"] + Time.now.to_i
-        ticket
-      end
-
-      def access_token_param
-        {access_token: get_access_token}
-      end
-
+    # 暴露出：http_get,http_post两个方法，方便第三方开发者扩展未开发的微信API。
     def http_get(url, headers={}, endpoint="plain")
-        puts "get jsticket" if url.include?("getticket")
       headers = headers.merge(access_token_param)
       WeixinAuthorize.http_get_without_token(url, headers, endpoint)
     end
@@ -80,6 +71,12 @@ module WeixinAuthorize
       WeixinAuthorize.http_post_without_token(url, payload, headers, endpoint)
     end
 
+    private
+
+      def access_token_param
+        {access_token: get_access_token}
+      end
+
       def security_redis_key(key)
         Digest::MD5.hexdigest(key.to_s).upcase
       end

lib/weixin_authorize/js_ticket/object_store.rb

+module WeixinAuthorize
+  module JsTicket
+    class ObjectStore < Store
+
+      def jsticket_expired?
+        # 如果当前token过期时间小于现在的时间，则重新获取一次
+        client.jsticket_expired_at <= Time.now.to_i
+      end
+
+      def jsticket
+        super
+        client.jsticket
+      end
+
+      def refresh_jsticket
+        super
+      end
+
+    end
+  end
+end

lib/weixin_authorize/js_ticket/redis_store.rb

+module WeixinAuthorize
+  module JsTicket
+    class RedisStore < Store
+      JSTICKET = "jsticket"
+      EXPIRED_AT = "expired_at"
+
+      def jsticket_expired?
+        weixin_redis.hvals(client.jsticket_redis_key).empty?
+      end
+
+      def refresh_jsticket
+        super
+        weixin_redis.hmset(
+          client.jsticket_redis_key,
+          JSTICKET,
+          client.jsticket,
+          EXPIRED_AT,
+          client.jsticket_expired_at
+        )
+        weixin_redis.expireat(
+          client.jsticket_redis_key,
+          client.jsticket_expired_at.to_i
+        )
+      end
+
+      def jsticket
+        super
+        client.jsticket = weixin_redis.hget(client.jsticket_redis_key, JSTICKET)
+        client.jsticket_expired_at = weixin_redis.hget(
+          client.jsticket_redis_key,
+          EXPIRED_AT
+        )
+        client.jsticket
+      end
+
+      def weixin_redis
+        WeixinAuthorize.weixin_redis
+      end
+    end
+  end
+end

lib/weixin_authorize/js_ticket/store.rb

+# encoding: utf-8
+module WeixinAuthorize
+  module JsTicket
+    class Store
+
+      attr_accessor :client
+
+      def initialize(client)
+        @client = client
+      end
+
+      def self.init_with(client)
+        if WeixinAuthorize.weixin_redis.nil?
+          ObjectStore.new(client)
+        else
+          RedisStore.new(client)
+        end
+      end
+
+      def jsticket_expired?
+        raise NotImplementedError, "Subclasses must implement a jsticket_expired? method"
+      end
+
+      def refresh_jsticket
+        set_jsticket
+      end
+
+      def jsticket
+        refresh_jsticket if jsticket_expired?
+      end
+
+      def set_jsticket
+        result = client.http_get("/ticket/getticket", {type: 1}).result
+        client.jsticket = result["ticket"]
+        client.jsticket_expired_at = result["expires_in"] + Time.now.to_i
+      end
+
+    end
+  end
+end

lib/weixin_authorize/adapter/client_storage.rb → lib/weixin_authorize/token/object_store.rb

 # encoding: utf-8
 module WeixinAuthorize
-  class ClientStorage < Storage
+  module Token
+    class ObjectStore < Store
 
       def valid?
         super
@@ -20,4 +21,5 @@ module WeixinAuthorize
         client.access_token
       end
     end
+  end
 end

lib/weixin_authorize/adapter/redis_storage.rb → lib/weixin_authorize/token/redis_store.rb

 # encoding: utf-8
 module WeixinAuthorize
-
-  class RedisStorage < Storage
+  module Token
+    class RedisStore < Store
 
       def valid?
        weixin_redis.del(client.redis_key)
@@ -29,6 +29,7 @@ module WeixinAuthorize
       def weixin_redis
         WeixinAuthorize.weixin_redis
       end
-
     end
+  end
+
 end

lib/weixin_authorize/adapter/storage.rb → lib/weixin_authorize/token/store.rb

 # encoding: utf-8
 module WeixinAuthorize
-
-  class Storage
+  module Token
+    class Store
 
       attr_accessor :client
 
@@ -11,9 +11,9 @@ module WeixinAuthorize
 
       def self.init_with(client)
         if WeixinAuthorize.weixin_redis.nil?
-        ClientStorage.new(client)
+          ObjectStore.new(client)
         else
-        RedisStorage.new(client)
+          RedisStore.new(client)
         end
       end
 
@@ -25,7 +25,7 @@ module WeixinAuthorize
         auth_result = http_get_access_token
         auth = false
         if auth_result.is_ok?
-        set_access_token_for_client(auth_result.result)
+          set_access_token(auth_result.result)
           auth = true
         end
         {"valid" => auth, "handler" => auth_result}
@@ -33,7 +33,7 @@ module WeixinAuthorize
 
       def refresh_token
         handle_valid_exception
-      set_access_token_for_client
+        set_access_token
       end
 
       def access_token
@@ -44,7 +44,7 @@ module WeixinAuthorize
         raise NotImplementedError, "Subclasses must implement a token_expired? method"
       end
 
-    def set_access_token_for_client(access_token_infos=nil)
+      def set_access_token(access_token_infos=nil)
         token_infos = access_token_infos || http_get_access_token.result
         client.access_token = token_infos["access_token"]
         client.expired_at   = Time.now.to_i + token_infos["expires_in"].to_i
@@ -67,7 +67,6 @@ module WeixinAuthorize
             raise ValidAccessTokenException, result_handler.full_error_message
           end
         end
-
     end
-
+  end
 end

spec/2_fetch_jsticket_spec.rb

+describe WeixinAuthorize::Client do
+  describe "#get jsticket" do
+    it "return the same jsticket in the same thing twice" do
+      js_ticket_1 = $client.get_jsticket
+      sleep 5
+      js_ticket_2 = $client.get_jsticket
+      expect(js_ticket_1).to eq(js_ticket_2)
+    end
+  end
+end

spec/spec_helper.rb

@@ -43,7 +43,7 @@ ENV["APPSECRET"]="1a941cd88cb4579ba98ec06b6813af03"
 ENV["OPENID"]="o9k6BuB0kydAcPTc7sPxppB1GQqA"
 
 # Comment to test for ClientStorage
-redis = Redis.new(:host => "127.0.0.1",:port => "6379")
+redis = Redis.new(host: "127.0.0.1", port: "6379", db: 15)
 
 namespace = "weixin_test:weixin_authorize"
 
@@ -58,6 +58,7 @@ WeixinAuthorize.configure do |config|
 end
 
 $client = WeixinAuthorize::Client.new(ENV["APPID"], ENV["APPSECRET"])
+
 RSpec.configure do |config|
 # The settings below are suggested to provide a good initial experience
 # with RSpec, but feel free to customize to your heart's content.