1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package postgresctl
import (
"context"
"fmt"
"os"
"path/filepath"
"strings"
"github.com/fsnotify/fsnotify"
"github.com/johnandersen777/socialweb-computer-kcp/api/v1alpha1"
"github.com/johnandersen777/socialweb-computer-kcp/internal/kit/fswatch"
)
const (
PidFileName = "postmaster.pid"
pidStatusLine = 7
pidStatusReady = "ready"
)
func (h *Host) PidFile(o v1alpha1.Postgres) string {
return filepath.Join(h.DataDir(o), PidFileName)
}
func (h *Host) Accepting(o v1alpha1.Postgres) bool {
raw, err := os.ReadFile(h.PidFile(o))
if err != nil {
return false
}
lines := strings.Split(string(raw), "\n")
return len(lines) > pidStatusLine && strings.TrimSpace(lines[pidStatusLine]) == pidStatusReady
}
func (h *Host) PidChanges(ctx context.Context, o v1alpha1.Postgres) (<-chan struct{}, error) {
file := filepath.Clean(h.PidFile(o))
w, err := fswatch.Shared().Subscribe(filepath.Dir(file))
if err != nil {
return nil, fmt.Errorf("postgresctl: watching %s: %w", filepath.Dir(file), err)
}
out := make(chan struct{}, 1)
go func() {
defer close(out)
defer w.Close()
for {
select {
case <-ctx.Done():
return
case ev, ok := <-w.Events():
if !ok {
return
}
if filepath.Clean(ev.Name) != file || ev.Op == fsnotify.Chmod {
continue
}
select {
case out <- struct{}{}:
default:
}
case _, ok := <-w.Errors():
if !ok {
return
}
}
}
}()
return out, nil
}
