71 lines
1.8 KiB
Go
71 lines
1.8 KiB
Go
// Code generated by BobGen psql v0.41.1. DO NOT EDIT.
|
|
// This file is meant to be re-generated in place and/or deleted at any time.
|
|
|
|
package models
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/stephenafamo/bob"
|
|
"github.com/stephenafamo/bob/dialect/psql/dialect"
|
|
"github.com/stephenafamo/bob/orm"
|
|
)
|
|
|
|
var Preload = getPreloaders()
|
|
|
|
type preloaders struct {
|
|
OauthToken oauthTokenPreloader
|
|
Organization organizationPreloader
|
|
User userPreloader
|
|
}
|
|
|
|
func getPreloaders() preloaders {
|
|
return preloaders{
|
|
OauthToken: buildOauthTokenPreloader(),
|
|
Organization: buildOrganizationPreloader(),
|
|
User: buildUserPreloader(),
|
|
}
|
|
}
|
|
|
|
var (
|
|
SelectThenLoad = getThenLoaders[*dialect.SelectQuery]()
|
|
InsertThenLoad = getThenLoaders[*dialect.InsertQuery]()
|
|
UpdateThenLoad = getThenLoaders[*dialect.UpdateQuery]()
|
|
)
|
|
|
|
type thenLoaders[Q orm.Loadable] struct {
|
|
OauthToken oauthTokenThenLoader[Q]
|
|
Organization organizationThenLoader[Q]
|
|
User userThenLoader[Q]
|
|
}
|
|
|
|
func getThenLoaders[Q orm.Loadable]() thenLoaders[Q] {
|
|
return thenLoaders[Q]{
|
|
OauthToken: buildOauthTokenThenLoader[Q](),
|
|
Organization: buildOrganizationThenLoader[Q](),
|
|
User: buildUserThenLoader[Q](),
|
|
}
|
|
}
|
|
|
|
func thenLoadBuilder[Q orm.Loadable, T any](name string, f func(context.Context, bob.Executor, T, ...bob.Mod[*dialect.SelectQuery]) error) func(...bob.Mod[*dialect.SelectQuery]) orm.Loader[Q] {
|
|
return func(queryMods ...bob.Mod[*dialect.SelectQuery]) orm.Loader[Q] {
|
|
return func(ctx context.Context, exec bob.Executor, retrieved any) error {
|
|
loader, isLoader := retrieved.(T)
|
|
if !isLoader {
|
|
return fmt.Errorf("object %T cannot load %q", retrieved, name)
|
|
}
|
|
|
|
err := f(ctx, exec, loader, queryMods...)
|
|
|
|
// Don't cause an issue due to missing relationships
|
|
if errors.Is(err, sql.ErrNoRows) {
|
|
return nil
|
|
}
|
|
|
|
return err
|
|
}
|
|
}
|
|
}
|