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
48 lines
833 B
Go
48 lines
833 B
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
|
|
"github.com/sqlc-dev/sqlc/internal/codegen/json"
|
|
"github.com/sqlc-dev/sqlc/internal/plugin"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
func main() {
|
|
if err := run(); err != nil {
|
|
fmt.Fprintf(os.Stderr, "error generating JSON: %s", err)
|
|
os.Exit(2)
|
|
}
|
|
}
|
|
|
|
func run() error {
|
|
var req plugin.GenerateRequest
|
|
reqBlob, err := io.ReadAll(os.Stdin)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := proto.Unmarshal(reqBlob, &req); err != nil {
|
|
return err
|
|
}
|
|
resp, err := json.Generate(context.Background(), &req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
respBlob, err := proto.Marshal(resp)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
w := bufio.NewWriter(os.Stdout)
|
|
if _, err := w.Write(respBlob); err != nil {
|
|
return err
|
|
}
|
|
if err := w.Flush(); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|