81 lines
1.5 KiB
Go
81 lines
1.5 KiB
Go
package embeddedpostgres
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
type syncedLogger struct {
|
|
offset int64
|
|
logger io.Writer
|
|
file *os.File
|
|
}
|
|
|
|
func newSyncedLogger(dir string, logger io.Writer) (*syncedLogger, error) {
|
|
file, err := os.CreateTemp(dir, "embedded_postgres_log")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
s := syncedLogger{
|
|
logger: logger,
|
|
file: file,
|
|
}
|
|
|
|
return &s, nil
|
|
}
|
|
|
|
func (s *syncedLogger) flush() error {
|
|
if s.logger != nil {
|
|
file, err := os.Open(s.file.Name())
|
|
if err != nil {
|
|
return fmt.Errorf("unable to process postgres logs: %s", err)
|
|
}
|
|
|
|
defer func() {
|
|
if err := file.Close(); err != nil {
|
|
panic(err)
|
|
}
|
|
}()
|
|
|
|
if _, err = file.Seek(s.offset, io.SeekStart); err != nil {
|
|
return fmt.Errorf("unable to process postgres logs: %s", err)
|
|
}
|
|
|
|
readBytes, err := io.Copy(s.logger, file)
|
|
if err != nil {
|
|
return fmt.Errorf("unable to process postgres logs: %s", err)
|
|
}
|
|
|
|
s.offset += readBytes
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func readLogsOrTimeout(logger *os.File) (logContent []byte, err error) {
|
|
logContent = []byte("logs could not be read")
|
|
|
|
logContentChan := make(chan []byte, 1)
|
|
errChan := make(chan error, 1)
|
|
|
|
go func() {
|
|
if actualLogContent, err := ioutil.ReadFile(logger.Name()); err == nil {
|
|
logContentChan <- actualLogContent
|
|
} else {
|
|
errChan <- err
|
|
}
|
|
}()
|
|
|
|
select {
|
|
case logContent = <-logContentChan:
|
|
case err = <-errChan:
|
|
case <-time.After(10 * time.Second):
|
|
err = fmt.Errorf("timed out waiting for logs")
|
|
}
|
|
|
|
return logContent, err
|
|
}
|