refact(controller): Use session repo factory that takes options

This commit is contained in:
Timothy Messier committed 2022-09-07 17:53:40 +00:00
1 parent 68568828e6
commit febf0b9984
4 files changed
+13 -13

No files matched your search

+2 -2
View File
@@ -336,8 +336,8 @@ func New(ctx context.Context, conf *Config) (*Controller, error) {
c.TargetRepoFn = func() (*target.Repository, error) {
return target.NewRepository(dbase, dbase, c.kms)
}
c.SessionRepoFn = func() (*session.Repository, error) {
return session.NewRepository(dbase, dbase, c.kms)
c.SessionRepoFn = func(opt ...session.Option) (*session.Repository, error) {
return session.NewRepository(dbase, dbase, c.kms, opt...)
}
c.ConnectionRepoFn = func() (*session.ConnectionRepository, error) {
return session.NewConnectionRepository(ctx, dbase, dbase, c.kms)