1 package embeddedpostgres
2
3 import (
4 "fmt"
5 "io"
6 "io/ioutil"
7 "os"
8 "time"
9 )
10
11 type syncedLogger struct {
12 offset int64
13 logger io.Writer
14 file *os.File
15 }
16
17 func newSyncedLogger(dir string, logger io.Writer) (*syncedLogger, error) {
18 file, err := os.CreateTemp(dir, "embedded_postgres_log")
19 if err != nil {
20 return nil, err
21 }
22
23 s := syncedLogger{
24 logger: logger,
25 file: file,
26 }
27
28 return &s, nil
29 }
30
31 func (s *syncedLogger) flush() error {
32 if s.logger != nil {
33 file, err := os.Open(s.file.Name())
34 if err != nil {
35 return fmt.Errorf("unable to process postgres logs: %s", err)
36 }
37
38 defer func() {
39 if err := file.Close(); err != nil {
40 panic(err)
41 }
42 }()
43
44 if _, err = file.Seek(s.offset, io.SeekStart); err != nil {
45 return fmt.Errorf("unable to process postgres logs: %s", err)
46 }
47
48 readBytes, err := io.Copy(s.logger, file)
49 if err != nil {
50 return fmt.Errorf("unable to process postgres logs: %s", err)
51 }
52
53 s.offset += readBytes
54 }
55
56 return nil
57 }
58
59 func readLogsOrTimeout(logger *os.File) (logContent []byte, err error) {
60 logContent = []byte("logs could not be read")
61
62 logContentChan := make(chan []byte, 1)
63 errChan := make(chan error, 1)
64
65 go func() {
66 if actualLogContent, err := ioutil.ReadFile(logger.Name()); err == nil {
67 logContentChan <- actualLogContent
68 } else {
69 errChan <- err
70 }
71 }()
72
73 select {
74 case logContent = <-logContentChan:
75 case err = <-errChan:
76 case <-time.After(10 * time.Second):
77 err = fmt.Errorf("timed out waiting for logs")
78 }
79
80 return logContent, err
81 }
82
View as plain text