lib/websocket_rails/configuration.rb in websocket-rails-0.5.0 vs lib/websocket_rails/configuration.rb in websocket-rails-0.6.0

@@ -7,9 +7,17 @@
 
     def user_identifier=(identifier)
       @user_identifier = identifier
     end
 
+    def user_class
+      @user_class ||= User
+    end
+
+    def user_class=(klass)
+      @user_class = klass
+    end
+
     def keep_subscribers_when_private?
       @keep_subscribers_when_private ||= false
     end
 
     def keep_subscribers_when_private=(value)