| 505 | } |
| 506 | |
| 507 | func (nvq *NodeVersionQuery) loadNode(ctx context.Context, query *NodeQuery, nodes []*NodeVersion, init func(*NodeVersion), assign func(*NodeVersion, *Node)) error { |
| 508 | ids := make([]string, 0, len(nodes)) |
| 509 | nodeids := make(map[string][]*NodeVersion) |
| 510 | for i := range nodes { |
| 511 | fk := nodes[i].NodeID |
| 512 | if _, ok := nodeids[fk]; !ok { |
| 513 | ids = append(ids, fk) |
| 514 | } |
| 515 | nodeids[fk] = append(nodeids[fk], nodes[i]) |
| 516 | } |
| 517 | if len(ids) == 0 { |
| 518 | return nil |
| 519 | } |
| 520 | query.Where(node.IDIn(ids...)) |
| 521 | neighbors, err := query.All(ctx) |
| 522 | if err != nil { |
| 523 | return err |
| 524 | } |
| 525 | for _, n := range neighbors { |
| 526 | nodes, ok := nodeids[n.ID] |
| 527 | if !ok { |
| 528 | return fmt.Errorf(`unexpected foreign-key "node_id" returned %v`, n.ID) |
| 529 | } |
| 530 | for i := range nodes { |
| 531 | assign(nodes[i], n) |
| 532 | } |
| 533 | } |
| 534 | return nil |
| 535 | } |
| 536 | func (nvq *NodeVersionQuery) loadStorageFile(ctx context.Context, query *StorageFileQuery, nodes []*NodeVersion, init func(*NodeVersion), assign func(*NodeVersion, *StorageFile)) error { |
| 537 | ids := make([]uuid.UUID, 0, len(nodes)) |
| 538 | nodeids := make(map[uuid.UUID][]*NodeVersion) |