build source

vendor/github.com/fergusstrange/embedded-postgres/logging.go

@@ -0,0 +1,81 @@
+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
+}