Update example instances (fixes issue #23)

mightybyte · Dec 3, 2014 · 6524235 · 6524235
1 parent 03547ef
commit 6524235
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/example/Site.hs b/example/Site.hs
@@ -8,7 +8,7 @@ module Main where
 ------------------------------------------------------------------------------
 import           Control.Applicative
 import           Control.Monad.Trans
-import           Control.Monad.Trans.Reader
+import           Control.Monad.Reader
 import           Control.Monad.State
 import           Data.ByteString (ByteString)
 import           Control.Lens
@@ -40,6 +40,7 @@ makeLenses ''App
 
 instance HasPostgres (Handler b App) where
     getPostgresState = with db get
+    setLocalPostgresState s = local (set (db . snapletValue) s)
 
 ------------------------------------------------------------------------------
 -- | The application's routes.

diff --git a/src/Snap/Snaplet/PostgresqlSimple.hs b/src/Snap/Snaplet/PostgresqlSimple.hs
@@ -39,6 +39,7 @@ Optionally, if you find yourself doing many database queries, you can eliminate
 
 > instance HasPostgres (Handler b App) where
 >   getPostgresState = with db get
+>   setLocalPostgresState s = local (set (db . snapletValue) s)
 
 With this code, our postHandler example no longer requires the 'with' function: