it is a naiive thing

main
Bel LaPointe 2023-11-06 05:49:08 -07:00
parent 6d6df1da80
commit 1a586de656
2 changed files with 76 additions and 6 deletions

View File

@ -2,21 +2,88 @@ package replicator
import (
"context"
"io"
"time"
)
type Replicator struct {
Src Driver
Dest Driver
Src Driver
Dest Driver
Interval time.Duration
}
func NewReplicator(src, dest Driver) Replicator {
return Replicator{
Src: src,
Dest: dest,
Src: src,
Dest: dest,
Interval: time.Second,
}
}
func (r Replicator) Stream(ctx context.Context) error {
return io.EOF
last, err := r.getContinuation(ctx)
if err != nil {
return err
}
if last == nil {
if err := r.fullStream(ctx); err != nil {
return err
}
}
return r.deltaStream(ctx)
}
func (r Replicator) getContinuation(ctx context.Context) (Version, error) {
return nil, nil // TODO
}
func (r Replicator) setContinuation(ctx context.Context, continuation Version) error {
return nil // TODO
}
func (r Replicator) deltaStream(ctx context.Context) error {
continuation, err := r.getContinuation(ctx)
if err != nil {
return err
}
for ctx.Err() == nil {
if err := r._deltaStream(ctx, continuation); err != nil {
return err
}
select {
case <-ctx.Done():
case <-time.After(r.Interval):
}
}
return ctx.Err()
}
func (r Replicator) _deltaStream(ctx context.Context, continuation Version) error {
last, err := continuation.AsTime()
if err != nil {
return err
}
keys, after := r.Src.KeysSince(ctx, last)
for keyVersion := range keys {
if t, err := keyVersion.Version.AsTime(); err != nil {
return err
} else if t.After(last) {
last = t
}
valueVersion, err := r.Src.Get(ctx, keyVersion.Key)
if err != nil {
return err
}
if err := r.Dest.Set(ctx, keyVersion.Key, valueVersion.Value, valueVersion.Version); err != nil {
return err
}
r.setContinuation(ctx, TimeAsVersion(last))
}
return *after
}
func (r Replicator) fullStream(ctx context.Context) error {
return r._deltaStream(ctx, nil)
}

View File

@ -23,6 +23,9 @@ func TestReplicatorStream(t *testing.T) {
before: func(t *testing.T, r Replicator) {
r.Src.Set(nil, key, value, version)
},
during: func(t *testing.T, r Replicator) {
time.Sleep(time.Millisecond * 200)
},
after: func(t *testing.T, r Replicator) {
if got, _ := r.Dest.Get(nil, key); !got.Version.Equal(version) || !got.Value.Equal(value) {
t.Error(got)