Skip to content

Commit

Permalink
remove ownDomain filter
Browse files Browse the repository at this point in the history
  • Loading branch information
stefanwire committed Sep 27, 2023
1 parent 07d56a5 commit 0710f8e
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 21 deletions.
3 changes: 0 additions & 3 deletions charts/mlsstats/templates/cronjob.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -41,9 +41,6 @@ spec:
, "--s3-addressing-style", {{ .Values.config.s3.addressingStyle | quote }}
, "--s3-bucket-name", {{ .Values.config.s3.bucket.name | quote }}
, "--s3-bucket-dir", {{ .Values.config.s3.bucket.directory | quote }}
{{if .Values.config.federationDomain -}}
, "--own-domain", {{ .Values.config.federationDomain | quote -}}
{{- end -}}
]
resources:
env:
Expand Down
12 changes: 1 addition & 11 deletions tools/mlsstats/src/MlsStats/Options.hs
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,6 @@ where

import Amazonka
import Cassandra qualified as C
import Data.Domain
import Data.Text qualified as Text
import Imports
import Options.Applicative
Expand All @@ -34,8 +33,7 @@ import Util.Options

data Opts = Opts
{ cassandraSettings :: CassandraSettings,
s3Settings :: S3Settings,
ownDomain :: Maybe Domain
s3Settings :: S3Settings
}
deriving (Show, Generic)

Expand Down Expand Up @@ -64,14 +62,6 @@ optsParser =
Opts
<$> cassandraSettingsParser
<*> s3SettingsParser
<*> optional
( Domain
<$> strOption
( long "own-domain"
<> metavar "DOMAIN"
<> help "own domain"
)
)

cassandraSettingsParser :: Parser CassandraSettings
cassandraSettingsParser =
Expand Down
13 changes: 6 additions & 7 deletions tools/mlsstats/src/MlsStats/Run.hs
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,7 @@ run o = do
let settings = o.cassandraSettings
galleyTables <- initCas settings.galleyHost settings.galleyPort settings.galleyKeyspace logger'
brigTables <- initCas settings.brigHost settings.brigPort settings.brigKeyspace logger'
runCommand o.s3Settings galleyTables brigTables o.cassandraSettings.pageSize o.ownDomain
runCommand o.s3Settings galleyTables brigTables o.cassandraSettings.pageSize
where
initLogger =
Log.new
Expand All @@ -77,8 +77,8 @@ run o = do
. C.setKeyspace casKeyspace
$ C.defSettings

runCommand :: S3Settings -> ClientState -> ClientState -> Int32 -> Maybe Domain -> IO ()
runCommand s3 galleyTables brigTables queryPageSize mOwnDomain = do
runCommand :: S3Settings -> ClientState -> ClientState -> Int32 -> IO ()
runCommand s3 galleyTables brigTables queryPageSize = do
logger <- newLogger Debug stderr
let service =
setEndpoint (s3.endpoint ^. awsSecure) (s3.endpoint ^. awsHost) (s3.endpoint ^. awsPort) defaultService
Expand All @@ -98,7 +98,7 @@ runCommand s3 galleyTables brigTables queryPageSize mOwnDomain = do
runResourceT $ do
upload "user-client.csv" (userClient brigTables queryPageSize)
upload "conv-group-team-protocol.csv" (convGroupTeamProtocol galleyTables queryPageSize)
upload "domain-user-client-group.csv" (domainUserClientGroup mOwnDomain galleyTables queryPageSize)
upload "domain-user-client-group.csv" (domainUserClientGroup galleyTables queryPageSize)
upload "user-conv.csv" (userConv galleyTables queryPageSize)

userClient :: MonadIO m => ClientState -> Int32 -> ConduitT () ByteString m ()
Expand Down Expand Up @@ -143,14 +143,13 @@ convGroupTeamProtocol cassandra queryPageSize = do
A.String s -> s
_ -> "?"

domainUserClientGroup :: MonadIO m => Maybe Domain -> ClientState -> Int32 -> ConduitT () ByteString m ()
domainUserClientGroup mOwnDomain cassandra queryPageSize = do
domainUserClientGroup :: MonadIO m => ClientState -> Int32 -> ConduitT () ByteString m ()
domainUserClientGroup cassandra queryPageSize = do
yield "user_domain,user,client,group\r\n"
( transPipe
(runClient cassandra)
(paginateC domainUserClientGroupCql (paramsP LocalQuorum () queryPageSize) x1)
.| concat
.| filterC (maybe (const True) (\ownDomain (d, _, _, _) -> ownDomain == d) mOwnDomain)
.| mapC
( \(d, u, c, g) ->
(T.encodeUtf8 (domainText d))
Expand Down

0 comments on commit 0710f8e

Please sign in to comment.