Function getReturningColumns

src/postgres/parse-pg-query.ts:155–172

(statement: QueryNodePath<QueryParser.Query>)

}

function getReturningColumns(statement: QueryNodePath<QueryParser.Query>): ColumnReference[] {
  const body = (statement.node as any)[statement.type]

  const relationRefs = getTableReferences(statement, false)
  const { returningList = [], targetList = [] } = body

  const resTargets: QueryParser.ResTarget[] = [...returningList, ...targetList].filter(
    node => getNodeType(node) === "ResTarget"
  )

  return resTargets
    .map(resTarget => {
      const propKey = returningList.indexOf(resTarget) > -1 ? "returningList" : "targetList"
      const resTargetPath = createQueryNodeSubpath(statement, resTarget, propKey)
      return resolveResTarget(resTargetPath, relationRefs)
    })
    .filter((ref): ref is ColumnReference => ref !== null)
}

function instantiateQuery(path: QueryNodePath<QueryParser.Query>, context: QueryContext): Query {
  const referencedColumns = getReferencedColumns(path, context.expressionSpreadTypes)