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
78 lines
1.9 KiB
Go
78 lines
1.9 KiB
Go
package opts
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/sqlc-dev/sqlc/internal/plugin"
|
|
)
|
|
|
|
// The ShimOverride struct exists to bridge the gap between the Override struct
|
|
// and the previous Override struct defined in codegen.proto. Eventually these
|
|
// shim structs should be removed in favor of using the existing Override and
|
|
// GoType structs, but it's easier to provide these shim structs to not change
|
|
// the existing, working code.
|
|
type ShimOverride struct {
|
|
DbType string
|
|
Nullable bool
|
|
Column string
|
|
Table *plugin.Identifier
|
|
ColumnName string
|
|
Unsigned bool
|
|
GoType *ShimGoType
|
|
}
|
|
|
|
func shimOverride(req *plugin.GenerateRequest, o *Override) *ShimOverride {
|
|
var column string
|
|
var table plugin.Identifier
|
|
|
|
if o.Column != "" {
|
|
colParts := strings.Split(o.Column, ".")
|
|
switch len(colParts) {
|
|
case 2:
|
|
table.Schema = req.Catalog.DefaultSchema
|
|
table.Name = colParts[0]
|
|
column = colParts[1]
|
|
case 3:
|
|
table.Schema = colParts[0]
|
|
table.Name = colParts[1]
|
|
column = colParts[2]
|
|
case 4:
|
|
table.Catalog = colParts[0]
|
|
table.Schema = colParts[1]
|
|
table.Name = colParts[2]
|
|
column = colParts[3]
|
|
}
|
|
}
|
|
return &ShimOverride{
|
|
DbType: o.DBType,
|
|
Nullable: o.Nullable,
|
|
Unsigned: o.Unsigned,
|
|
Column: o.Column,
|
|
ColumnName: column,
|
|
Table: &table,
|
|
GoType: shimGoType(o),
|
|
}
|
|
}
|
|
|
|
type ShimGoType struct {
|
|
ImportPath string
|
|
Package string
|
|
TypeName string
|
|
BasicType bool
|
|
StructTags map[string]string
|
|
}
|
|
|
|
func shimGoType(o *Override) *ShimGoType {
|
|
// Note that there is a slight mismatch between this and the
|
|
// proto api. The GoType on the override is the unparsed type,
|
|
// which could be a qualified path or an object, as per
|
|
// https://docs.sqlc.dev/en/v1.18.0/reference/config.html#type-overriding
|
|
return &ShimGoType{
|
|
ImportPath: o.GoImportPath,
|
|
Package: o.GoPackage,
|
|
TypeName: o.GoTypeName,
|
|
BasicType: o.GoBasicType,
|
|
StructTags: o.GoStructTags,
|
|
}
|
|
}
|