Set ETW event name and options for logrus hook (#245)

* ETW name and options functionality

Added functionality to set the event (task) name and options (such as
event or associated event ID) for ETW events created by the hook based
on the logrus.Entry fields.

Signed-off-by: Hamza El-Saawy <hamzaelsaawy@microsoft.com>

* PR: export and comment

Signed-off-by: Hamza El-Saawy <hamzaelsaawy@microsoft.com>
This commit is contained in:
Hamza El-Saawy
2022-04-27 15:00:13 -04:00
committed by GitHub
parent 75ee807f27
commit 87532d1cfe
2 changed files with 117 additions and 13 deletions
+53
View File
@@ -0,0 +1,53 @@
//go:build windows
package etwlogrus
import (
"github.com/sirupsen/logrus"
"github.com/Microsoft/go-winio/pkg/etw"
)
// etw provider
// WithNewETWProvider registers a new ETW provider and sets the hook to log using it.
// The provider will be closed when the hook is closed.
func WithNewETWProvider(n string) HookOpt {
return func(h *Hook) error {
provider, err := etw.NewProvider(n, nil)
if err != nil {
return err
}
h.provider = provider
h.closeProvider = true
return nil
}
}
// WithExistingETWProvider configures the hook to use an existing ETW provider.
// The provider will not be closed when the hook is closed.
func WithExistingETWProvider(p *etw.Provider) HookOpt {
return func(h *Hook) error {
h.provider = p
h.closeProvider = false
return nil
}
}
// WithGetName sets the ETW EventName of an event to the value returned by f
// If the name is empty, the default event name will be used.
func WithGetName(f func(*logrus.Entry) string) HookOpt {
return func(h *Hook) error {
h.getName = f
return nil
}
}
// WithAdditionalEventOpts allows additional ETW event properties (keywords, tags, etc.) to be specified
func WithEventOpts(f func(*logrus.Entry) []etw.EventOpt) HookOpt {
return func(h *Hook) error {
h.getEventsOpts = f
return nil
}
}