Add Douglas.with_stamper

fteem · Apr 5, 2017 · c72f41c · c72f41c
1 parent fe2bbbd
commit c72f41c
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lib/douglas.rb b/lib/douglas.rb
@@ -11,6 +11,14 @@ def the_stamper
  def the_stamper= user
  RequestStore.store[:the_stamper] = user
  end
+
+ def with_stamper user
+ return unless block_given?
+
+ self.the_stamper = user
+ yield
+ self.the_stamper = nil
+ end
  end
 end