multipool-postgresql-simple (empty) → 0.1.0.0

raw patch · 7 files changed

+148/−0 lines, 7 filesdep +basedep +mtldep +multipoolsetup-changed

Dependencies added: base, mtl, multipool, multipool-postgresql-simple, postgresql-simple, resource-pool, unliftio-core, unordered-containers

Files

• ChangeLog.md +3/−0
• LICENSE +30/−0
• README.md +1/−0
• Setup.hs +2/−0
• multipool-postgresql-simple.cabal +62/−0
• src/Data/MultiPool/PostgreSQLSimple.hs +48/−0
• test/Spec.hs +2/−0

@@ -0,0 +1,3 @@+# Changelog for multipool-postgresql-simple++## Unreleased changes

@@ -0,0 +1,30 @@+Copyright Ian Duncan (c) 2019++All rights reserved.++Redistribution and use in source and binary forms, with or without+modification, are permitted provided that the following conditions are met:++    * Redistributions of source code must retain the above copyright+      notice, this list of conditions and the following disclaimer.++    * Redistributions in binary form must reproduce the above+      copyright notice, this list of conditions and the following+      disclaimer in the documentation and/or other materials provided+      with the distribution.++    * Neither the name of Ian Duncan nor the names of other+      contributors may be used to endorse or promote products derived+      from this software without specific prior written permission.++THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS+"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT+LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR+A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT+OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,+SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT+LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,+DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY+THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT+(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

@@ -0,0 +1,1 @@+# multipool-postgresql-simple

@@ -0,0 +1,2 @@+import Distribution.Simple+main = defaultMain

@@ -0,0 +1,62 @@+cabal-version: 1.12++-- This file has been generated from package.yaml by hpack version 0.31.1.+--+-- see: https://github.com/sol/hpack+--+-- hash: f2bcbf30da1024a0caf8fbeb2fd53d077c9e959a4a278b878d870e5e485bec9c++name:           multipool-postgresql-simple+version:        0.1.0.0+description:    Please see the README on GitHub at <https://github.com/iand675/multipool-postgresql-simple#readme>+homepage:       https://github.com/iand675/multipool-postgresql-simple#readme+bug-reports:    https://github.com/iand675/multipool-postgresql-simple/issues+author:         Ian Duncan+maintainer:     ian@iankduncan.com+copyright:      Ian Duncan+license:        BSD3+license-file:   LICENSE+build-type:     Simple+extra-source-files:+    README.md+    ChangeLog.md++source-repository head+  type: git+  location: https://github.com/iand675/multipool-postgresql-simple++library+  exposed-modules:+      Data.MultiPool.PostgreSQLSimple+  other-modules:+      Paths_multipool_postgresql_simple+  hs-source-dirs:+      src+  build-depends:+      base >=4.7 && <5+    , mtl+    , multipool+    , postgresql-simple+    , resource-pool+    , unliftio-core+    , unordered-containers+  default-language: Haskell2010++test-suite multipool-postgresql-simple-test+  type: exitcode-stdio-1.0+  main-is: Spec.hs+  other-modules:+      Paths_multipool_postgresql_simple+  hs-source-dirs:+      test+  ghc-options: -threaded -rtsopts -with-rtsopts=-N+  build-depends:+      base >=4.7 && <5+    , mtl+    , multipool+    , multipool-postgresql-simple+    , postgresql-simple+    , resource-pool+    , unliftio-core+    , unordered-containers+  default-language: Haskell2010

@@ -0,0 +1,48 @@+{-# LANGUAGE AllowAmbiguousTypes #-}+{-# LANGUAGE DataKinds #-}+{-# LANGUAGE FlexibleInstances #-}+{-# LANGUAGE MultiParamTypeClasses #-}+{-# LANGUAGE OverloadedStrings #-}+{-# LANGUAGE RankNTypes #-}+{-# LANGUAGE TypeFamilies #-}+{-# LANGUAGE FlexibleInstances #-}+{-# LANGUAGE UndecidableInstances #-}+{-# LANGUAGE RecordWildCards #-}++module Data.MultiPool.PostgreSQLSimple+    ( MultiPoolBackend(..)+    ) where++import Control.Exception+import Control.Monad.IO.Unlift+import Control.Monad.Reader+import Data.Pool+import Data.MultiPool+import Data.HashMap.Strict (HashMap)+import qualified Data.HashMap.Strict as HM+import Database.PostgreSQL.Simple++withResource' :: MonadUnliftIO m => Pool a -> (a -> m b) -> m b+withResource' pool action =+  withRunInIO $ \io ->+    liftIO $ withResource pool $ \a ->+      io $ action a++instance MonadUnliftIO m => MultiPoolBackend m Connection where+  type MasterConnection Connection = Connection+  type ReplicaConnection Connection = Connection+  type ReplicaIdentifier Connection = InstanceName Connection++  runWriteAny b m = runWrite b () m+  runWrite b () m = withResource' (multiPoolMaster b) $ runReaderT m++  runReadMaster b () m = runReadAnyMaster b m+  runReadAnyMaster b m = withResource' (multiPoolMaster b) $ runReaderT m+  runReadAny b m = do+    mident <- liftIO $ multiPoolAnyReplicaSelector b b+    case mident of+      Nothing -> runReadAnyMaster b m+      Just ident -> runRead b ident m+  runRead b ident m = case HM.lookup ident (multiPoolReplica b) of+    Nothing -> throw (InstanceDoesNotExist ident)+    Just repl -> withResource' repl $ runReaderT m

@@ -0,0 +1,2 @@+main :: IO ()+main = putStrLn "Test suite not yet implemented"