diff --git a/pkg/platform.go b/pkg/platform.go index 999e075..a0bfdea 100644 --- a/pkg/platform.go +++ b/pkg/platform.go @@ -42,24 +42,28 @@ func (platform *Platform) FetchCollections( fmt.Printf("Collections: %v\n", len(platform.Collections)) - err = BulkCreateNode(platform._conn, platform.Collections) + err = BulkCreateNode(platform._conn, values) if err != nil { return err } - for _, item := range platform.Collections { - err := platform.AddRelationship( - &Relationship{ - _class: "PLATFORM_HAS_COLLECTION", - From: platform.Id, - To: item.Id, - }) + relationships := make([]*Relationship, 0, len(values)) + + for _, item := range values { + relation := &Relationship{ + _class: "PLATFORM_HAS_COLLECTION", + From: platform.Id, + To: item.Id, + } + + err := platform.AddRelationship(relation) if err != nil { return err } + relationships = append(relationships, relation) } - err = BulkCreateRelationships(platform._conn, platform._relationships) + err = BulkCreateRelationships(platform._conn, relationships) if err != nil { return err }