download request: fix setting of log level for plugin manager
Signed-off-by: Steve Kriss <steve@heptio.com>pull/799/head
parent
3234124afe
commit
cf7c8587f0
|
@ -742,6 +742,7 @@ func (s *server) runControllers(config *api.Config) error {
|
||||||
s.sharedInformerFactory.Ark().V1().Backups(),
|
s.sharedInformerFactory.Ark().V1().Backups(),
|
||||||
s.pluginRegistry,
|
s.pluginRegistry,
|
||||||
s.logger,
|
s.logger,
|
||||||
|
s.logLevel,
|
||||||
)
|
)
|
||||||
wg.Add(1)
|
wg.Add(1)
|
||||||
go func() {
|
go func() {
|
||||||
|
|
|
@ -50,8 +50,7 @@ type downloadRequestController struct {
|
||||||
createSignedURL cloudprovider.CreateSignedURLFunc
|
createSignedURL cloudprovider.CreateSignedURLFunc
|
||||||
backupLocationLister listers.BackupStorageLocationLister
|
backupLocationLister listers.BackupStorageLocationLister
|
||||||
backupLister listers.BackupLister
|
backupLister listers.BackupLister
|
||||||
pluginRegistry plugin.Registry
|
newPluginManager func(logrus.FieldLogger) plugin.Manager
|
||||||
newPluginManager func(logger logrus.FieldLogger, logLevel logrus.Level, pluginRegistry plugin.Registry) plugin.Manager
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewDownloadRequestController creates a new DownloadRequestController.
|
// NewDownloadRequestController creates a new DownloadRequestController.
|
||||||
|
@ -63,6 +62,7 @@ func NewDownloadRequestController(
|
||||||
backupInformer informers.BackupInformer,
|
backupInformer informers.BackupInformer,
|
||||||
pluginRegistry plugin.Registry,
|
pluginRegistry plugin.Registry,
|
||||||
logger logrus.FieldLogger,
|
logger logrus.FieldLogger,
|
||||||
|
logLevel logrus.Level,
|
||||||
) Interface {
|
) Interface {
|
||||||
c := &downloadRequestController{
|
c := &downloadRequestController{
|
||||||
genericController: newGenericController("downloadrequest", logger),
|
genericController: newGenericController("downloadrequest", logger),
|
||||||
|
@ -74,8 +74,10 @@ func NewDownloadRequestController(
|
||||||
|
|
||||||
// use variables to refer to these functions so they can be
|
// use variables to refer to these functions so they can be
|
||||||
// replaced with fakes for testing.
|
// replaced with fakes for testing.
|
||||||
createSignedURL: cloudprovider.CreateSignedURL,
|
createSignedURL: cloudprovider.CreateSignedURL,
|
||||||
newPluginManager: plugin.NewManager,
|
newPluginManager: func(logger logrus.FieldLogger) plugin.Manager {
|
||||||
|
return plugin.NewManager(logger, logLevel, pluginRegistry)
|
||||||
|
},
|
||||||
|
|
||||||
clock: &clock.RealClock{},
|
clock: &clock.RealClock{},
|
||||||
}
|
}
|
||||||
|
@ -143,7 +145,7 @@ const signedURLTTL = 10 * time.Minute
|
||||||
|
|
||||||
// generatePreSignedURL generates a pre-signed URL for downloadRequest, changes the phase to
|
// generatePreSignedURL generates a pre-signed URL for downloadRequest, changes the phase to
|
||||||
// Processed, and persists the changes to storage.
|
// Processed, and persists the changes to storage.
|
||||||
func (c *downloadRequestController) generatePreSignedURL(downloadRequest *v1.DownloadRequest, log *logrus.Entry) error {
|
func (c *downloadRequestController) generatePreSignedURL(downloadRequest *v1.DownloadRequest, log logrus.FieldLogger) error {
|
||||||
update := downloadRequest.DeepCopy()
|
update := downloadRequest.DeepCopy()
|
||||||
|
|
||||||
var (
|
var (
|
||||||
|
@ -173,7 +175,7 @@ func (c *downloadRequestController) generatePreSignedURL(downloadRequest *v1.Dow
|
||||||
return errors.WithStack(err)
|
return errors.WithStack(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
pluginManager := c.newPluginManager(log, log.Level, c.pluginRegistry)
|
pluginManager := c.newPluginManager(log)
|
||||||
defer pluginManager.CleanupClients()
|
defer pluginManager.CleanupClients()
|
||||||
|
|
||||||
objectStore, err := getObjectStoreForLocation(backupLocation, pluginManager)
|
objectStore, err := getObjectStoreForLocation(backupLocation, pluginManager)
|
||||||
|
|
|
@ -61,6 +61,7 @@ func newDownloadRequestTestHarness(t *testing.T) *downloadRequestTestHarness {
|
||||||
informerFactory.Ark().V1().Backups(),
|
informerFactory.Ark().V1().Backups(),
|
||||||
nil,
|
nil,
|
||||||
arktest.NewLogger(),
|
arktest.NewLogger(),
|
||||||
|
logrus.InfoLevel,
|
||||||
).(*downloadRequestController)
|
).(*downloadRequestController)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -69,9 +70,7 @@ func newDownloadRequestTestHarness(t *testing.T) *downloadRequestTestHarness {
|
||||||
|
|
||||||
controller.clock = clock.NewFakeClock(clockTime)
|
controller.clock = clock.NewFakeClock(clockTime)
|
||||||
|
|
||||||
controller.newPluginManager = func(_ logrus.FieldLogger, _ logrus.Level, _ plugin.Registry) plugin.Manager {
|
controller.newPluginManager = func(_ logrus.FieldLogger) plugin.Manager { return pluginManager }
|
||||||
return pluginManager
|
|
||||||
}
|
|
||||||
|
|
||||||
pluginManager.On("CleanupClients").Return()
|
pluginManager.On("CleanupClients").Return()
|
||||||
objectStore.On("Init", mock.Anything).Return(nil)
|
objectStore.On("Init", mock.Anything).Return(nil)
|
||||||
|
|
Loading…
Reference in New Issue