Import changes.

  - 508945e57836dd9413411b8ed56c68430319773d

GitOrigin-RevId: 508945e57836dd9413411b8ed56c68430319773d

tools/nbictl/README.md

@@ -171,6 +171,10 @@ Insert the model NMTS Relationship contained within the file provided on the com
 
 Delete the model NMTS Relationship contained within the file provided on the command line ('-' reads from stdin).
 
+### upsert-fragment
+
+Upsert the model NMTS Fragment contained within the file provided on the command line ('-' reads from stdin).
+
 ### list-elements
 
 List all model elements (NMTS Entities and Relationships).

tools/nbictl/model.go

@@ -177,6 +177,46 @@ func ModelDeleteRelationship(appCtx *cli.Context) error {
 	return err
 }
 
+// TODO: turn these into one atomic RPC call in the modelfe.
+func ModelUpsertFragment(appCtx *cli.Context) error {
+	nmtsFragment := &nmtspb.Fragment{}
+	if err := readProtoFromCommandLineFilenameArgument(appCtx, nmtsFragment); err != nil {
+		return err
+	}
+
+	conn, err := openAPIConnection(appCtx, modelAPISubDomain)
+	if err != nil {
+		return err
+	}
+	defer conn.Close()
+	modelClient := modelpb.NewModelClient(conn)
+
+	for _, nmtsEntity := range nmtsFragment.GetEntity() {
+		_, err = modelClient.UpsertEntity(
+			appCtx.Context,
+			&modelpb.UpsertEntityRequest{
+				Entity: nmtsEntity,
+			})
+		if err != nil {
+			return err
+		}
+	}
+
+	for _, nmtsRelationship := range nmtsFragment.GetRelationship() {
+		_, err = modelClient.InsertRelationship(
+			appCtx.Context,
+			&modelpb.InsertRelationshipRequest{
+				Relationship: nmtsRelationship,
+			})
+		if err != nil {
+			return err
+		}
+	}
+
+	fmt.Fprintln(appCtx.App.ErrWriter, "# OK")
+	return nil
+}
+
 func ModelGetEntity(appCtx *cli.Context) error {
 	if appCtx.Args().Len() != 1 {
 		return fmt.Errorf("need one and only one Entity ID argument")

tools/nbictl/nbictl.go

@@ -442,6 +442,12 @@ func App() *cli.App {
 						Category: "model relationships",
 						Action:   ModelDeleteRelationship,
 					},
+					{
+						Name:     "upsert-fragment",
+						Usage:    "Upsert the model NMTS Fragment contained within the file provided on the command line ('-' reads from stdin).",
+						Category: "model relationships",
+						Action:   ModelUpsertFragment,
+					},
 					{
 						Name:     "list-elements",
 						Usage:    "List all model elements (NMTS Entities and Relationships).",