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
134 lines
3.0 KiB
Go
134 lines
3.0 KiB
Go
package cmd
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"path/filepath"
|
|
"runtime"
|
|
"runtime/trace"
|
|
|
|
"golang.org/x/sync/errgroup"
|
|
|
|
"github.com/sqlc-dev/sqlc/internal/compiler"
|
|
"github.com/sqlc-dev/sqlc/internal/config"
|
|
"github.com/sqlc-dev/sqlc/internal/opts"
|
|
)
|
|
|
|
type OutputPair struct {
|
|
Gen config.SQLGen
|
|
Plugin *config.Codegen
|
|
|
|
config.SQL
|
|
}
|
|
|
|
type ResultProcessor interface {
|
|
Pairs(context.Context, *config.Config) []OutputPair
|
|
ProcessResult(context.Context, config.CombinedSettings, OutputPair, *compiler.Result) error
|
|
}
|
|
|
|
func Process(ctx context.Context, rp ResultProcessor, dir, filename string, o *Options) error {
|
|
e := o.Env
|
|
stderr := o.Stderr
|
|
|
|
configPath, conf, err := o.ReadConfig(dir, filename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
base := filepath.Base(configPath)
|
|
if err := config.Validate(conf); err != nil {
|
|
fmt.Fprintf(stderr, "error validating %s: %s\n", base, err)
|
|
return err
|
|
}
|
|
|
|
if err := e.Validate(conf); err != nil {
|
|
fmt.Fprintf(stderr, "error validating %s: %s\n", base, err)
|
|
return err
|
|
}
|
|
|
|
return processQuerySets(ctx, rp, conf, dir, o)
|
|
}
|
|
|
|
func processQuerySets(ctx context.Context, rp ResultProcessor, conf *config.Config, dir string, o *Options) error {
|
|
stderr := o.Stderr
|
|
|
|
errored := false
|
|
|
|
pairs := rp.Pairs(ctx, conf)
|
|
grp, gctx := errgroup.WithContext(ctx)
|
|
grp.SetLimit(runtime.GOMAXPROCS(0))
|
|
|
|
stderrs := make([]bytes.Buffer, len(pairs))
|
|
|
|
for i, pair := range pairs {
|
|
sql := pair
|
|
errout := &stderrs[i]
|
|
|
|
grp.Go(func() error {
|
|
combo := config.Combine(*conf, sql.SQL)
|
|
if sql.Plugin != nil {
|
|
combo.Codegen = *sql.Plugin
|
|
}
|
|
|
|
// TODO: This feels like a hack that will bite us later
|
|
joined := make([]string, 0, len(sql.Schema))
|
|
for _, s := range sql.Schema {
|
|
joined = append(joined, filepath.Join(dir, s))
|
|
}
|
|
sql.Schema = joined
|
|
|
|
joined = make([]string, 0, len(sql.Queries))
|
|
for _, q := range sql.Queries {
|
|
joined = append(joined, filepath.Join(dir, q))
|
|
}
|
|
sql.Queries = joined
|
|
|
|
var name, lang string
|
|
parseOpts := opts.Parser{
|
|
Experiment: o.Env.Experiment,
|
|
}
|
|
|
|
switch {
|
|
case sql.Gen.Go != nil:
|
|
name = combo.Go.Package
|
|
lang = "golang"
|
|
|
|
case sql.Plugin != nil:
|
|
lang = fmt.Sprintf("process:%s", sql.Plugin.Plugin)
|
|
name = sql.Plugin.Plugin
|
|
}
|
|
|
|
packageRegion := trace.StartRegion(gctx, "package")
|
|
trace.Logf(gctx, "", "name=%s dir=%s plugin=%s", name, dir, lang)
|
|
|
|
result, failed := parse(gctx, name, dir, sql.SQL, combo, parseOpts, errout)
|
|
if failed {
|
|
packageRegion.End()
|
|
errored = true
|
|
return nil
|
|
}
|
|
if err := rp.ProcessResult(gctx, combo, sql, result); err != nil {
|
|
fmt.Fprintf(errout, "# package %s\n", name)
|
|
fmt.Fprintf(errout, "error generating code: %s\n", err)
|
|
errored = true
|
|
}
|
|
packageRegion.End()
|
|
return nil
|
|
})
|
|
}
|
|
if err := grp.Wait(); err != nil {
|
|
return err
|
|
}
|
|
if errored {
|
|
for i, _ := range stderrs {
|
|
if _, err := io.Copy(stderr, &stderrs[i]); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return fmt.Errorf("errored")
|
|
}
|
|
return nil
|
|
}
|