Рыба проекта. Минимальная функциональность
This commit is contained in:
@@ -0,0 +1,102 @@
|
||||
package pipeline
|
||||
|
||||
import (
|
||||
"context"
|
||||
"io"
|
||||
|
||||
"git.tswf.io/infra/go-synapse-backupper/pkg/domain"
|
||||
"git.tswf.io/infra/go-synapse-backupper/pkg/domain/crypto"
|
||||
"git.tswf.io/infra/go-synapse-backupper/pkg/domain/pgdump"
|
||||
)
|
||||
|
||||
// Option configures a Runner.
|
||||
type Option func(*runner)
|
||||
|
||||
// WithDumper replaces the default pg_dump dumper.
|
||||
func WithDumper(dumper pgdump.Dumper) Option {
|
||||
return func(r *runner) {
|
||||
r.dumper = dumper
|
||||
}
|
||||
}
|
||||
|
||||
// WithEncryptor replaces the default encryptor.
|
||||
func WithEncryptor(encryptor crypto.Encryptor) Option {
|
||||
return func(r *runner) {
|
||||
r.encryptor = encryptor
|
||||
}
|
||||
}
|
||||
|
||||
// Runner orchestrates the dump → encrypt → sink pipeline.
|
||||
type runner struct {
|
||||
dumper pgdump.Dumper
|
||||
encryptor crypto.Encryptor
|
||||
}
|
||||
|
||||
// NewRunner creates a pipeline runner with the given functional options.
|
||||
func NewRunner(options ...Option) *runner {
|
||||
r := &runner{}
|
||||
for _, option := range options {
|
||||
option(r)
|
||||
}
|
||||
return r
|
||||
}
|
||||
|
||||
// Run executes the full backup pipeline: pg_dump → encrypt → sink.
|
||||
func (r *runner) Run(
|
||||
ctx context.Context,
|
||||
pgDumpOpts pgdump.Options,
|
||||
recipients []crypto.RecipientPub,
|
||||
sink domain.Sink,
|
||||
rand io.Reader,
|
||||
) (retErr error) {
|
||||
tx, err := sink.Begin(pgDumpOpts.Key)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer func() {
|
||||
if retErr != nil {
|
||||
_ = tx.Abort()
|
||||
}
|
||||
}()
|
||||
|
||||
dumpCtx, dumpCancel := context.WithCancel(ctx)
|
||||
defer dumpCancel()
|
||||
|
||||
pipeR, pipeW := io.Pipe()
|
||||
|
||||
dumpErrCh := make(chan error, 1)
|
||||
go func() {
|
||||
dumpErrCh <- r.dumper.Dump(dumpCtx, pgDumpOpts, pipeW)
|
||||
}()
|
||||
|
||||
encryptErrCh := make(chan error, 1)
|
||||
go func() {
|
||||
encryptErrCh <- r.encryptor.Encrypt(pipeR, recipients, tx, rand)
|
||||
}()
|
||||
|
||||
select {
|
||||
case derr := <-dumpErrCh:
|
||||
if derr != nil {
|
||||
_ = pipeR.CloseWithError(derr)
|
||||
_ = <-encryptErrCh
|
||||
return derr
|
||||
}
|
||||
eerr := <-encryptErrCh
|
||||
if eerr != nil {
|
||||
return eerr
|
||||
}
|
||||
return tx.Commit()
|
||||
case eerr := <-encryptErrCh:
|
||||
if eerr != nil {
|
||||
dumpCancel()
|
||||
_ = pipeR.CloseWithError(eerr)
|
||||
_ = <-dumpErrCh
|
||||
return eerr
|
||||
}
|
||||
derr := <-dumpErrCh
|
||||
if derr != nil {
|
||||
return derr
|
||||
}
|
||||
return tx.Commit()
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user