130 lines
3.5 KiB
Go
130 lines
3.5 KiB
Go
package gitaly
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
|
|
gitalyclient "gitlab.com/gitlab-org/gitaly/v14/client"
|
|
"gitlab.com/gitlab-org/gitaly/v14/proto/go/gitalypb"
|
|
"gitlab.com/gitlab-org/gitaly/v14/streamio"
|
|
)
|
|
|
|
type SmartHTTPClient struct {
|
|
sidechannelRegistry *gitalyclient.SidechannelRegistry
|
|
gitalypb.SmartHTTPServiceClient
|
|
}
|
|
|
|
func (client *SmartHTTPClient) InfoRefsResponseReader(ctx context.Context, repo *gitalypb.Repository, rpc string, gitConfigOptions []string, gitProtocol string) (io.Reader, error) {
|
|
rpcRequest := &gitalypb.InfoRefsRequest{
|
|
Repository: repo,
|
|
GitConfigOptions: gitConfigOptions,
|
|
GitProtocol: gitProtocol,
|
|
}
|
|
|
|
switch rpc {
|
|
case "git-upload-pack":
|
|
stream, err := client.InfoRefsUploadPack(ctx, rpcRequest)
|
|
return infoRefsReader(stream), err
|
|
case "git-receive-pack":
|
|
stream, err := client.InfoRefsReceivePack(ctx, rpcRequest)
|
|
return infoRefsReader(stream), err
|
|
default:
|
|
return nil, fmt.Errorf("InfoRefsResponseWriterTo: Unsupported RPC: %q", rpc)
|
|
}
|
|
}
|
|
|
|
type infoRefsClient interface {
|
|
Recv() (*gitalypb.InfoRefsResponse, error)
|
|
}
|
|
|
|
func infoRefsReader(stream infoRefsClient) io.Reader {
|
|
return streamio.NewReader(func() ([]byte, error) {
|
|
resp, err := stream.Recv()
|
|
return resp.GetData(), err
|
|
})
|
|
}
|
|
|
|
func (client *SmartHTTPClient) ReceivePack(ctx context.Context, repo *gitalypb.Repository, glId string, glUsername string, glRepository string, gitConfigOptions []string, clientRequest io.Reader, clientResponse io.Writer, gitProtocol string) error {
|
|
stream, err := client.PostReceivePack(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
rpcRequest := &gitalypb.PostReceivePackRequest{
|
|
Repository: repo,
|
|
GlId: glId,
|
|
GlUsername: glUsername,
|
|
GlRepository: glRepository,
|
|
GitConfigOptions: gitConfigOptions,
|
|
GitProtocol: gitProtocol,
|
|
}
|
|
|
|
if err := stream.Send(rpcRequest); err != nil {
|
|
return fmt.Errorf("initial request: %v", err)
|
|
}
|
|
|
|
numStreams := 2
|
|
errC := make(chan error, numStreams)
|
|
|
|
go func() {
|
|
rr := streamio.NewReader(func() ([]byte, error) {
|
|
response, err := stream.Recv()
|
|
return response.GetData(), err
|
|
})
|
|
_, err := io.Copy(clientResponse, rr)
|
|
errC <- err
|
|
}()
|
|
|
|
go func() {
|
|
sw := streamio.NewWriter(func(data []byte) error {
|
|
return stream.Send(&gitalypb.PostReceivePackRequest{Data: data})
|
|
})
|
|
_, err := io.Copy(sw, clientRequest)
|
|
stream.CloseSend()
|
|
errC <- err
|
|
}()
|
|
|
|
for i := 0; i < numStreams; i++ {
|
|
if err := <-errC; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (client *SmartHTTPClient) UploadPack(ctx context.Context, repo *gitalypb.Repository, clientRequest io.Reader, clientResponse io.Writer, gitConfigOptions []string, gitProtocol string) error {
|
|
ctx, waiter := client.sidechannelRegistry.Register(ctx, func(conn gitalyclient.SidechannelConn) error {
|
|
if _, err := io.Copy(conn, clientRequest); err != nil {
|
|
return fmt.Errorf("copy request body: %w", err)
|
|
}
|
|
|
|
if err := conn.CloseWrite(); err != nil {
|
|
return fmt.Errorf("close request body: %w", err)
|
|
}
|
|
|
|
if _, err := io.Copy(clientResponse, conn); err != nil {
|
|
return fmt.Errorf("copy response body: %w", err)
|
|
}
|
|
|
|
return nil
|
|
})
|
|
defer waiter.Close()
|
|
|
|
rpcRequest := &gitalypb.PostUploadPackWithSidechannelRequest{
|
|
Repository: repo,
|
|
GitConfigOptions: gitConfigOptions,
|
|
GitProtocol: gitProtocol,
|
|
}
|
|
|
|
if _, err := client.PostUploadPackWithSidechannel(ctx, rpcRequest); err != nil {
|
|
return fmt.Errorf("PostUploadPackWithSidechannel: %w", err)
|
|
}
|
|
|
|
if err := waiter.Close(); err != nil {
|
|
return fmt.Errorf("close sidechannel waiter: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|