Merge pull request kata-containers#1298 from jodh-intel/store-pkg-add…

…-SetLogger store: Add SetLogger API
jcvenegas · Feb 28, 2019 · dbfd965 · dbfd965
2 parents e8a8e0d + f540a80
commit dbfd965
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/virtcontainers/api.go b/virtcontainers/api.go
@@ -48,6 +48,7 @@ func SetLogger(ctx context.Context, logger *logrus.Entry) {
 	virtLog = logger.WithFields(fields)
 
 	deviceApi.SetLogger(virtLog)
+	store.SetLogger(virtLog)
 }
 
 // CreateSandbox is the virtcontainers sandbox creation entry point.

diff --git a/virtcontainers/store/manager.go b/virtcontainers/store/manager.go
@@ -182,6 +182,13 @@ func DeleteAll() {
 
 var storeLog = logrus.WithField("source", "virtcontainers/store")
 
+// SetLogger sets the custom logger to be used by this package. If not called,
+// the package will create its own logger.
+func SetLogger(logger *logrus.Entry) {
+	fields := storeLog.Data
+	storeLog = logger.WithFields(fields)
+}
+
 // Logger returns a logrus logger appropriate for logging Store messages
 func (s *Store) Logger() *logrus.Entry {
 	return storeLog.WithFields(logrus.Fields{