sqlboiler/templates/08_relationship_one_to_one_eager.tpl

94 lines
2.5 KiB
Smarty
Raw Permalink Normal View History

2016-09-18 20:13:11 +02:00
{{- if .Table.IsJoinTable -}}
{{- else -}}
{{- $dot := . -}}
2016-09-19 01:49:18 +02:00
{{- range .Table.ToOneRelationships -}}
{{- $txt := txtsFromOneToOne $dot.Tables $dot.Table . -}}
2016-09-18 20:13:11 +02:00
{{- $varNameSingular := $dot.Table.Name | singular | camelCase -}}
{{- $arg := printf "maybe%s" $txt.LocalTable.NameGo -}}
{{- $slice := printf "%sSlice" $txt.LocalTable.NameGo}}
// Load{{$txt.Function.Name}} allows an eager lookup of values, cached into the
2016-08-23 06:53:37 +02:00
// loaded structs of the objects.
2016-09-18 20:13:11 +02:00
func ({{$varNameSingular}}L) Load{{$txt.Function.Name}}(e boil.Executor, singular bool, {{$arg}} interface{}) error {
var slice []*{{$txt.LocalTable.NameGo}}
var object *{{$txt.LocalTable.NameGo}}
2016-08-16 09:30:08 +02:00
count := 1
if singular {
2016-09-18 20:13:11 +02:00
object = {{$arg}}.(*{{$txt.LocalTable.NameGo}})
} else {
slice = *{{$arg}}.(*{{$slice}})
count = len(slice)
}
2016-08-16 09:30:08 +02:00
args := make([]interface{}, count)
if singular {
if object.R == nil {
object.R = &{{$varNameSingular}}R{}
}
2016-09-18 20:13:11 +02:00
args[0] = object.{{$txt.LocalTable.ColumnNameGo}}
} else {
for i, obj := range slice {
if obj.R == nil {
obj.R = &{{$varNameSingular}}R{}
}
2016-09-18 20:13:11 +02:00
args[i] = obj.{{$txt.LocalTable.ColumnNameGo}}
}
}
2016-08-16 09:30:08 +02:00
query := fmt.Sprintf(
2016-09-19 01:02:08 +02:00
"select * from {{.ForeignTable | $dot.SchemaTable}} where {{.ForeignColumn | $dot.Quotes}} in (%s)",
strmangle.Placeholders(dialect.IndexPlaceholders, count, 1, 1),
)
2016-08-16 09:30:08 +02:00
if boil.DebugMode {
fmt.Fprintf(boil.DebugWriter, "%s\n%v\n", query, args)
}
results, err := e.Query(query, args...)
if err != nil {
2016-09-18 20:13:11 +02:00
return errors.Wrap(err, "failed to eager load {{$txt.ForeignTable.NameGo}}")
}
defer results.Close()
2016-08-16 09:30:08 +02:00
2016-09-18 20:13:11 +02:00
var resultSlice []*{{$txt.ForeignTable.NameGo}}
2016-09-15 05:59:55 +02:00
if err = queries.Bind(results, &resultSlice); err != nil {
2016-09-18 20:13:11 +02:00
return errors.Wrap(err, "failed to bind eager loaded slice {{$txt.ForeignTable.NameGo}}")
}
2016-08-16 09:30:08 +02:00
{{if not $dot.NoHooks -}}
2016-09-21 06:05:26 +02:00
if len({{$varNameSingular}}AfterSelectHooks) != 0 {
for _, obj := range resultSlice {
if err := obj.doAfterSelectHooks(e); err != nil {
return err
}
}
}
{{- end}}
if len(resultSlice) == 0 {
return nil
}
if singular {
2016-09-18 20:13:11 +02:00
object.R.{{$txt.Function.Name}} = resultSlice[0]
return nil
}
2016-08-16 09:30:08 +02:00
for _, local := range slice {
for _, foreign := range resultSlice {
2016-09-19 01:02:08 +02:00
{{if $txt.Function.UsesBytes -}}
2016-09-18 20:13:11 +02:00
if 0 == bytes.Compare(local.{{$txt.Function.LocalAssignment}}, foreign.{{$txt.Function.ForeignAssignment}}) {
{{else -}}
2016-09-18 20:13:11 +02:00
if local.{{$txt.Function.LocalAssignment}} == foreign.{{$txt.Function.ForeignAssignment}} {
{{end -}}
2016-09-18 20:13:11 +02:00
local.R.{{$txt.Function.Name}} = foreign
break
}
}
}
2016-08-16 09:30:08 +02:00
return nil
2016-08-16 09:30:08 +02:00
}
2016-09-18 20:13:11 +02:00
{{end -}}{{/* range */}}
{{end}}{{/* join table */}}