Commit 3a866fa3 authored by Przemyslaw Kaminski's avatar Przemyslaw Kaminski

[ngrams API] added VersionedWithCount to ngrams output

parent 76d60bbd
...@@ -78,6 +78,7 @@ module Gargantext.API.Ngrams ...@@ -78,6 +78,7 @@ module Gargantext.API.Ngrams
, Version , Version
, Versioned(..) , Versioned(..)
, VersionedWithCount(..)
, currentVersion , currentVersion
, listNgramsChangedSince , listNgramsChangedSince
) )
...@@ -487,7 +488,7 @@ getTableNgrams :: forall env err m. ...@@ -487,7 +488,7 @@ getTableNgrams :: forall env err m.
-> Maybe MinSize -> Maybe MaxSize -> Maybe MinSize -> Maybe MaxSize
-> Maybe OrderBy -> Maybe OrderBy
-> (NgramsTerm -> Bool) -> (NgramsTerm -> Bool)
-> m (Versioned NgramsTable) -> m (VersionedWithCount NgramsTable)
getTableNgrams _nType nId tabType listId limit_ offset getTableNgrams _nType nId tabType listId limit_ offset
listType minSize maxSize orderBy searchQuery = do listType minSize maxSize orderBy searchQuery = do
...@@ -579,8 +580,7 @@ getTableNgrams _nType nId tabType listId limit_ offset ...@@ -579,8 +580,7 @@ getTableNgrams _nType nId tabType listId limit_ offset
fltr <- tableMap2 & v_data %%~ fmap NgramsTable . setScores (not scoresNeeded) fltr <- tableMap2 & v_data %%~ fmap NgramsTable . setScores (not scoresNeeded)
. filteredNodes . filteredNodes
let fltrCount = length $ fltr ^. v_data . _NgramsTable
printDebug "[getTableNgrams] fltr" $ length $ fltr ^. v_data . _NgramsTable
t2 <- getTime' t2 <- getTime'
tableMap3 <- tableMap2 & v_data %%~ fmap NgramsTable tableMap3 <- tableMap2 & v_data %%~ fmap NgramsTable
...@@ -595,7 +595,7 @@ getTableNgrams _nType nId tabType listId limit_ offset ...@@ -595,7 +595,7 @@ getTableNgrams _nType nId tabType listId limit_ offset
% " sql=" % (if scoresNeeded then "map2" else "map3") % " sql=" % (if scoresNeeded then "map2" else "map3")
% "\n" % "\n"
) t0 t3 t0 t1 t1 t2 t2 t3 ) t0 t3 t0 t1 t1 t2 t2 t3
pure tableMap3 pure $ toVersionedWithCount fltrCount tableMap3
scoresRecomputeTableNgrams :: forall env err m. (RepoCmdM env err m, HasNodeError err, HasConnectionPool env, HasConfig env) => NodeId -> TabType -> ListId -> m Int scoresRecomputeTableNgrams :: forall env err m. (RepoCmdM env err m, HasNodeError err, HasConnectionPool env, HasConfig env) => NodeId -> TabType -> ListId -> m Int
...@@ -661,7 +661,7 @@ type TableNgramsApiGet = Summary " Table Ngrams API Get" ...@@ -661,7 +661,7 @@ type TableNgramsApiGet = Summary " Table Ngrams API Get"
:> QueryParam "maxTermSize" MaxSize :> QueryParam "maxTermSize" MaxSize
:> QueryParam "orderBy" OrderBy :> QueryParam "orderBy" OrderBy
:> QueryParam "search" Text :> QueryParam "search" Text
:> Get '[JSON] (Versioned NgramsTable) :> Get '[JSON] (VersionedWithCount NgramsTable)
type TableNgramsApiPut = Summary " Table Ngrams API Change" type TableNgramsApiPut = Summary " Table Ngrams API Change"
:> QueryParamR "ngramsType" TabType :> QueryParamR "ngramsType" TabType
...@@ -701,7 +701,7 @@ getTableNgramsCorpus :: (RepoCmdM env err m, HasNodeError err, HasConnectionPool ...@@ -701,7 +701,7 @@ getTableNgramsCorpus :: (RepoCmdM env err m, HasNodeError err, HasConnectionPool
-> Maybe MinSize -> Maybe MaxSize -> Maybe MinSize -> Maybe MaxSize
-> Maybe OrderBy -> Maybe OrderBy
-> Maybe Text -- full text search -> Maybe Text -- full text search
-> m (Versioned NgramsTable) -> m (VersionedWithCount NgramsTable)
getTableNgramsCorpus nId tabType listId limit_ offset listType minSize maxSize orderBy mt = getTableNgramsCorpus nId tabType listId limit_ offset listType minSize maxSize orderBy mt =
getTableNgrams NodeCorpus nId tabType listId limit_ offset listType minSize maxSize orderBy searchQuery getTableNgrams NodeCorpus nId tabType listId limit_ offset listType minSize maxSize orderBy searchQuery
where where
...@@ -727,7 +727,7 @@ getTableNgramsDoc :: (RepoCmdM env err m, HasNodeError err, HasConnectionPool en ...@@ -727,7 +727,7 @@ getTableNgramsDoc :: (RepoCmdM env err m, HasNodeError err, HasConnectionPool en
-> Maybe MinSize -> Maybe MaxSize -> Maybe MinSize -> Maybe MaxSize
-> Maybe OrderBy -> Maybe OrderBy
-> Maybe Text -- full text search -> Maybe Text -- full text search
-> m (Versioned NgramsTable) -> m (VersionedWithCount NgramsTable)
getTableNgramsDoc dId tabType listId limit_ offset listType minSize maxSize orderBy _mt = do getTableNgramsDoc dId tabType listId limit_ offset listType minSize maxSize orderBy _mt = do
ns <- selectNodesWithUsername NodeList userMaster ns <- selectNodesWithUsername NodeList userMaster
let ngramsType = ngramsTypeFromTabType tabType let ngramsType = ngramsTypeFromTabType tabType
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment