insert.go

//nolint:errcheck
package psql

import (
	"github.com/dosco/graphjin/core/internal/qcode"
	"github.com/dosco/graphjin/core/internal/sdata"
)

func (c *compilerContext) renderInsert() {
	for _, m := range c.qc.Mutates {
		switch {
		case m.Type == qcode.MTInsert:
			c.renderInsertStmt(m, false)
		case m.Type == qcode.MTUpsert:
			c.renderInsertStmt(m, true)
		case m.Rel.Type == sdata.RelOneToOne && m.Type == qcode.MTConnect:
			c.renderOneToOneConnectStmt(m)
		}
	}
	c.w.WriteString(` `)
}

func (c *compilerContext) renderInsertStmt(m qcode.Mutate, embedded bool) {
	c.w.WriteString(`, `)
	c.renderCteName(m)
	c.w.WriteString(` AS (`)

	c.renderOneToManyModifiers(m)

	c.w.WriteString(`INSERT INTO `)
	c.quoted(m.Ti.Name)

	c.w.WriteString(` (`)
	n := c.renderInsertUpdateColumns(m, false)
	c.renderNestedRelColumns(m, false, false, n)
	c.w.WriteString(`)`)

	c.w.WriteString(` SELECT `)
	n = c.renderInsertUpdateColumns(m, true)
	c.renderNestedRelColumns(m, true, false, n)

	c.w.WriteString(` FROM _sg_input i`)
	c.renderNestedRelTables(m, false)

	if m.Array {
		c.w.WriteString(`, json_populate_recordset`)
	} else {
		c.w.WriteString(`, json_populate_record`)
	}

	c.w.WriteString(`(NULL::"`)
	c.w.WriteString(m.Ti.Name)
	joinPath(c.w, `", i.j`, m.Path)
	c.w.WriteString(`) t`)

	if !embedded {
		c.w.WriteString(` RETURNING *)`)
	}
}