0f4df23290
Some checks failed
kotlin / test (push) Has been skipped
python / test (push) Has been skipped
typescript / test (push) Has been skipped
go / build darwin/amd64 (push) Has been cancelled
go / build linux/amd64 (push) Has been cancelled
go / build windows/amd64 (push) Has been cancelled
go / build darwin/arm64 (push) Has been cancelled
go / build linux/arm64 (push) Has been cancelled
go / build windows/arm64 (push) Has been cancelled
go / test (push) Has been cancelled
go / vuln_check (push) Has been cancelled
53 lines
1.5 KiB
Go
53 lines
1.5 KiB
Go
package ext
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
|
|
"github.com/sqlc-dev/sqlc/internal/plugin"
|
|
)
|
|
|
|
type Handler interface {
|
|
Generate(context.Context, *plugin.GenerateRequest) (*plugin.GenerateResponse, error)
|
|
|
|
Invoke(ctx context.Context, method string, args any, reply any, opts ...grpc.CallOption) error
|
|
NewStream(ctx context.Context, desc *grpc.StreamDesc, method string, opts ...grpc.CallOption) (grpc.ClientStream, error)
|
|
}
|
|
|
|
type wrapper struct {
|
|
fn func(context.Context, *plugin.GenerateRequest) (*plugin.GenerateResponse, error)
|
|
}
|
|
|
|
func (w *wrapper) Generate(ctx context.Context, req *plugin.GenerateRequest) (*plugin.GenerateResponse, error) {
|
|
return w.fn(ctx, req)
|
|
}
|
|
|
|
func (w *wrapper) Invoke(ctx context.Context, method string, args any, reply any, opts ...grpc.CallOption) error {
|
|
req, ok := args.(*plugin.GenerateRequest)
|
|
if !ok {
|
|
return fmt.Errorf("args isn't a GenerateRequest")
|
|
}
|
|
resp, ok := reply.(*plugin.GenerateResponse)
|
|
if !ok {
|
|
return fmt.Errorf("reply isn't a GenerateResponse")
|
|
}
|
|
res, err := w.Generate(ctx, req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
resp.Files = res.Files
|
|
return nil
|
|
}
|
|
|
|
func (w *wrapper) NewStream(ctx context.Context, desc *grpc.StreamDesc, method string, opts ...grpc.CallOption) (grpc.ClientStream, error) {
|
|
return nil, status.Error(codes.Unimplemented, "")
|
|
}
|
|
|
|
func HandleFunc(fn func(context.Context, *plugin.GenerateRequest) (*plugin.GenerateResponse, error)) Handler {
|
|
return &wrapper{fn}
|
|
}
|