Skip to content

Commit 19029cf

Browse files
authored
buf: Remote modules (#1)
* Use remote packages * v2: Use remote modules from Buf
1 parent a965e8c commit 19029cf

File tree

6 files changed

+20
-9647
lines changed

6 files changed

+20
-9647
lines changed

codegen/codegen.go

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -6,13 +6,12 @@ import (
66
"fmt"
77
"io"
88
"os"
9-
)
10-
11-
type Request = CodeGenRequest
129

13-
type Response = CodeGenResponse
10+
pb "buf.build/gen/go/sqlc/sqlc/protocolbuffers/go/protos/plugin"
11+
"google.golang.org/protobuf/proto"
12+
)
1413

15-
type Handler func(context.Context, *Request) (*Response, error)
14+
type Handler func(context.Context, *pb.CodeGenRequest) (*pb.CodeGenResponse, error)
1615

1716
func Run(h Handler) {
1817
if err := run(h); err != nil {
@@ -22,19 +21,19 @@ func Run(h Handler) {
2221
}
2322

2423
func run(h Handler) error {
25-
var req Request
24+
var req pb.CodeGenRequest
2625
reqBlob, err := io.ReadAll(os.Stdin)
2726
if err != nil {
2827
return err
2928
}
30-
if err := req.UnmarshalVT(reqBlob); err != nil {
29+
if err := proto.Unmarshal(reqBlob, &req); err != nil {
3130
return err
3231
}
3332
resp, err := h(context.Background(), &req)
3433
if err != nil {
3534
return err
3635
}
37-
respBlob, err := resp.MarshalVT()
36+
respBlob, err := proto.Marshal(resp)
3837
if err != nil {
3938
return err
4039
}

0 commit comments

Comments
 (0)