Method insertLastIDs

dialect/sql/sqlgraph/graph.go:1952–2012

insertLastIDs invokes the batch insert query on the transaction and returns the LastInsertID of all entities.

(ctx context.Context, tx dialect.ExecQuerier, insert *sql.InsertBuilder)

// insertLastIDs invokes the batch insert query on the transaction and returns the LastInsertID of all entities.
func (c *batchCreator) insertLastIDs(ctx context.Context, tx dialect.ExecQuerier, insert *sql.InsertBuilder) error {
	query, args, err := insert.QueryErr()
	if err != nil {
		return err
	}
	// MySQL does not support the "RETURNING" clause.
	if insert.Dialect() != dialect.MySQL {
		rows := &sql.Rows{}
		if err := tx.Query(ctx, query, args, rows); err != nil {
			return err
		}
		defer rows.Close()
		for i := 0; rows.Next(); i++ {
			node := c.Nodes[i]
			switch _, ok := node.ID.Value.(field.ValueScanner); {
			case ok:
				// If the ID implements the sql.Scanner
				// interface it should be a pointer type.
				if err := rows.Scan(node.ID.Value); err != nil {
					return err
				}
			case node.ID.Type.Numeric():
				// Normalize the type to int64 to make it looks
				// like LastInsertId.
				var id int64
				if err := rows.Scan(&id); err != nil {
					return err
				}
				node.ID.Value = id
			default:
				if err := rows.Scan(&node.ID.Value); err != nil {
					return err
				}
			}
		}
		return rows.Err()
	}
	// MySQL.
	var res sql.Result
	if err := tx.Exec(ctx, query, args, &res); err != nil {
		return err
	}
	// If the ID field is not numeric (e.g. string),
	// there is no way to scan the LAST_INSERT_ID.
	if len(c.Nodes) > 0 && c.Nodes[0].ID.Type.Numeric() {
		id, err := res.LastInsertId()
		if err != nil {
			return err
		}
		affected, err := res.RowsAffected()
		if err != nil {
			return err
		}
		// Assume the ID field is AUTO_INCREMENT
		// if its type is numeric.
		for i := 0; int64(i) < affected && i < len(c.Nodes); i++ {
			c.Nodes[i].ID.Value = id + int64(i)
		}