tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

concurrent preloads

Browse Source
This commit is contained in:
Jake Burkhead 2016-06-28 10:49:42 -07:00
parent 3324ab2063
commit 9757e285ad
No known key found for this signature in database
GPG Key ID: 78B1126DB558F08F
1 changed files with 18 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
callback_query_preload.go
View File

@ -5,6 +5,7 @@ import (
"fmt" "fmt"
"reflect" "reflect"
"strings" "strings"
"sync"
) )
// preloadCallback used to preload associations // preloadCallback used to preload associations
@ -16,7 +17,9 @@ func preloadCallback(scope *Scope) {
var ( var (
preloadedMap = map[string]bool{} preloadedMap = map[string]bool{}
fields = scope.Fields() fields = scope.Fields()
wg = &sync.WaitGroup{}
) )
defer wg.Wait()
for _, preload := range scope.Search.preload { for _, preload := range scope.Search.preload {
var ( var (
@ -45,16 +48,18 @@ func preloadCallback(scope *Scope) {
continue continue
} }
wg.Add(1)
switch field.Relationship.Kind { switch field.Relationship.Kind {
case "has_one": case "has_one":
currentScope.handleHasOnePreload(field, currentPreloadConditions) go currentScope.handleHasOnePreload(field, currentPreloadConditions, wg)
case "has_many": case "has_many":
currentScope.handleHasManyPreload(field, currentPreloadConditions) go currentScope.handleHasManyPreload(field, currentPreloadConditions, wg)
case "belongs_to": case "belongs_to":
currentScope.handleBelongsToPreload(field, currentPreloadConditions) go currentScope.handleBelongsToPreload(field, currentPreloadConditions, wg)
case "many_to_many": case "many_to_many":
currentScope.handleManyToManyPreload(field, currentPreloadConditions) go currentScope.handleManyToManyPreload(field, currentPreloadConditions, wg)
default: default:
wg.Done()
scope.Err(errors.New("unsupported relation")) scope.Err(errors.New("unsupported relation"))
} }
@ -77,6 +82,7 @@ func preloadCallback(scope *Scope) {
} }
} }
} }
} }
func (scope *Scope) generatePreloadDBWithConditions(conditions []interface{}) (*DB, []interface{}) { func (scope *Scope) generatePreloadDBWithConditions(conditions []interface{}) (*DB, []interface{}) {
@ -97,7 +103,8 @@ func (scope *Scope) generatePreloadDBWithConditions(conditions []interface{}) (*
} }
// handleHasOnePreload used to preload has one associations // handleHasOnePreload used to preload has one associations
func (scope *Scope) handleHasOnePreload(field *Field, conditions []interface{}) { func (scope *Scope) handleHasOnePreload(field *Field, conditions []interface{}, wg *sync.WaitGroup) {
defer wg.Done()
relation := field.Relationship relation := field.Relationship
// get relations's primary keys // get relations's primary keys
@ -146,7 +153,8 @@ func (scope *Scope) handleHasOnePreload(field *Field, conditions []interface{})
} }
// handleHasManyPreload used to preload has many associations // handleHasManyPreload used to preload has many associations
func (scope *Scope) handleHasManyPreload(field *Field, conditions []interface{}) { func (scope *Scope) handleHasManyPreload(field *Field, conditions []interface{}, wg *sync.WaitGroup) {
defer wg.Done()
relation := field.Relationship relation := field.Relationship
// get relations's primary keys // get relations's primary keys
@ -197,7 +205,8 @@ func (scope *Scope) handleHasManyPreload(field *Field, conditions []interface{})
} }
// handleBelongsToPreload used to preload belongs to associations // handleBelongsToPreload used to preload belongs to associations
func (scope *Scope) handleBelongsToPreload(field *Field, conditions []interface{}) { func (scope *Scope) handleBelongsToPreload(field *Field, conditions []interface{}, wg *sync.WaitGroup) {
defer wg.Done()
relation := field.Relationship relation := field.Relationship
// preload conditions // preload conditions
@ -236,7 +245,8 @@ func (scope *Scope) handleBelongsToPreload(field *Field, conditions []interface{
} }
// handleManyToManyPreload used to preload many to many associations // handleManyToManyPreload used to preload many to many associations
func (scope *Scope) handleManyToManyPreload(field *Field, conditions []interface{}) { func (scope *Scope) handleManyToManyPreload(field *Field, conditions []interface{}, wg *sync.WaitGroup) {
defer wg.Done()
var ( var (
relation = field.Relationship relation = field.Relationship
joinTableHandler = relation.JoinTableHandler joinTableHandler = relation.JoinTableHandler