31 lines
883 B
Go
31 lines
883 B
Go
package commonspace
|
|
|
|
import (
|
|
"context"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/commonspace/remotediff"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/commonspace/spacesyncproto"
|
|
)
|
|
|
|
type RpcHandler interface {
|
|
HeadSync(ctx context.Context, req *spacesyncproto.HeadSyncRequest) (*spacesyncproto.HeadSyncResponse, error)
|
|
Stream(stream spacesyncproto.DRPCSpace_StreamStream) error
|
|
}
|
|
|
|
type rpcHandler struct {
|
|
s *space
|
|
}
|
|
|
|
func (r *rpcHandler) HeadSync(ctx context.Context, req *spacesyncproto.HeadSyncRequest) (*spacesyncproto.HeadSyncResponse, error) {
|
|
return remotediff.HandlerRangeRequest(ctx, r.s.diff, req)
|
|
}
|
|
|
|
func (r *rpcHandler) Stream(stream spacesyncproto.DRPCSpace_StreamStream) (err error) {
|
|
err = r.s.SyncService().StreamPool().AddAndReadStream(stream)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
<-stream.Context().Done()
|
|
return
|
|
}
|