Created
January 20, 2016 23:40
-
-
Save logie17/3e79f010db4c6b4c714d to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
var logLines []string | |
type logMock struct{} | |
func (l *logMock) Write(p []byte) (int, error) { | |
logLines = append(logLines, string(p)) | |
return 0, nil | |
} | |
//CaptureLogOutput A utility to capture the log output of a function and pass it along | |
func CaptureLogOutput(toBeCaptured func(), handleResults func([]string)) { | |
logLines = []string{} | |
currentFlags := log.Flags() | |
log.SetFlags(0) | |
log.SetOutput(&logMock{}) | |
toBeCaptured() | |
log.SetOutput(os.Stderr) | |
log.SetFlags(currentFlags) | |
handleResults(logLines) | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
It would also be handy if the return value of
CaptureLogOutput
were the return value of thefunc
that it encloses. That would allow you to do things like:This may be a can of worms since the return value of CaptureLogOutput then becomes subject to the whims of the code that's being wrapped. So maybe another callback?
Making the callback to test the return value of the wrapped function be the third argument to
CaptureLogOutput
feels sensible since not all code run inCaptureLogOutput
will actually have a return value, so that avoids needing to pass in an empty function as a second argument.