]> Git — Sourcephile - gargantext.git/blob - src/Data/Gargantext/Database/NodeNode.hs
[CLEAN] Moving CoreNLP in Ngrams folder.
[gargantext.git] / src / Data / Gargantext / Database / NodeNode.hs
1 {-# LANGUAGE TemplateHaskell #-}
2 {-# LANGUAGE FlexibleInstances #-}
3 {-# LANGUAGE MultiParamTypeClasses #-}
4 {-# LANGUAGE FunctionalDependencies #-}
5 {-# LANGUAGE Arrows #-}
6 {-# OPTIONS_GHC -fno-warn-orphans #-}
7
8 module Data.Gargantext.Database.NodeNode where
9
10 import Prelude
11 import Data.Maybe (Maybe)
12 import Data.Profunctor.Product.TH (makeAdaptorAndInstance)
13 import Control.Lens.TH (makeLensesWith, abbreviatedFields)
14 import qualified Database.PostgreSQL.Simple as PGS
15
16 import Opaleye
17
18 import Data.Gargantext.Database.Private (infoGargandb)
19
20 data NodeNodePoly node1_id node2_id score
21 = NodeNode { nodeNode_node1_id :: node1_id
22 , nodeNode_node2_id :: node2_id
23 , nodeNode_score :: score
24 } deriving (Show)
25
26 type NodeNodeWrite = NodeNodePoly (Column PGInt4) (Column PGInt4) (Maybe (Column PGFloat8))
27 type NodeNodeRead = NodeNodePoly (Column PGInt4) (Column PGInt4) (Column PGFloat8)
28
29
30 type NodeNode = NodeNodePoly Int Int (Maybe Double)
31
32 $(makeAdaptorAndInstance "pNodeNode" ''NodeNodePoly)
33 $(makeLensesWith abbreviatedFields ''NodeNodePoly)
34
35
36 nodeNodeTable :: Table NodeNodeWrite NodeNodeRead
37 nodeNodeTable = Table "nodes_nodes" (pNodeNode NodeNode { nodeNode_node1_id = required "node1_id"
38 , nodeNode_node2_id = required "node2_id"
39 , nodeNode_score = optional "score"
40 }
41 )
42
43
44 queryNodeNodeTable :: Query NodeNodeRead
45 queryNodeNodeTable = queryTable nodeNodeTable
46
47
48 -- | not optimized (get all ngrams without filters)
49 nodeNodes :: IO [NodeNode]
50 nodeNodes = do
51 conn <- PGS.connect infoGargandb
52 runQuery conn queryNodeNodeTable
53
54 instance QueryRunnerColumnDefault PGFloat8 (Maybe Double) where
55 queryRunnerColumnDefault = fieldQueryRunnerColumn