entc/gen: update mutation-id on post mutation

examples/o2m2types/ent/mutation.go

@@ -116,8 +116,8 @@ func (m PetMutation) Tx() (*Tx, error) {
 	return tx, nil
 }
 
-// ID returns the ID value in the mutation. Note that the ID
-// is only available if it was provided to the builder.
+// ID returns the ID value in the mutation. Note that the ID is only available
+// if it was provided to the builder or after it was returned from the database.
 func (m *PetMutation) ID() (id int, exists bool) {
 	if m.id == nil {
 		return
@@ -475,8 +475,8 @@ func (m UserMutation) Tx() (*Tx, error) {
 	return tx, nil
 }
 
-// ID returns the ID value in the mutation. Note that the ID
-// is only available if it was provided to the builder.
+// ID returns the ID value in the mutation. Note that the ID is only available
+// if it was provided to the builder or after it was returned from the database.
 func (m *UserMutation) ID() (id int, exists bool) {
 	if m.id == nil {
 		return

examples/o2m2types/ent/pet_create.go

@@ -76,6 +76,7 @@ func (pc *PetCreate) Save(ctx context.Context) (*Pet, error) {
 			}
 			pc.mutation = mutation
 			node, err = pc.sqlSave(ctx)
+			mutation.id = &node.ID
 			mutation.done = true
 			return node, err
 		})
@@ -196,6 +197,7 @@ func (pcb *PetCreateBulk) Save(ctx context.Context) ([]*Pet, error) {
 						}
 					}
 				}
+				mutation.id = &nodes[i].ID
 				mutation.done = true
 				if err != nil {
 					return nil, err

examples/o2m2types/ent/user_create.go

@@ -78,6 +78,7 @@ func (uc *UserCreate) Save(ctx context.Context) (*User, error) {
 			}
 			uc.mutation = mutation
 			node, err = uc.sqlSave(ctx)
+			mutation.id = &node.ID
 			mutation.done = true
 			return node, err
 		})
@@ -208,6 +209,7 @@ func (ucb *UserCreateBulk) Save(ctx context.Context) ([]*User, error) {
 						}
 					}
 				}
+				mutation.id = &nodes[i].ID
 				mutation.done = true
 				if err != nil {
 					return nil, err