Method RowIter

server/node/validate_constraint.go:93–152

RowIter implements sql.ExecSourceRel.

(ctx *sql.Context, r sql.Row)

// RowIter implements sql.ExecSourceRel.
func (v *ValidateConstraint) RowIter(ctx *sql.Context, r sql.Row) (sql.RowIter, error) {
	db, err := v.DbProvider.Database(ctx, ctx.GetCurrentDatabase())
	if err != nil {
		return nil, err
	}

	schemaName := v.schemaName
	if schemaName == "" {
		schemaName, err = core.GetCurrentSchema(ctx)
		if err != nil {
			return nil, err
		}
	}
	schemaDb, ok := db.(sql.SchemaDatabase)
	if !ok {
		return nil, errors.Errorf("database does not support schemas")
	}
	dbSchema, ok, err := schemaDb.GetSchema(ctx, schemaName)
	if err != nil {
		return nil, err
	}
	if !ok {
		return nil, errors.Errorf("schema %s does not exist", schemaName)
	}
	tblNode, _, err := dbSchema.GetTableInsensitive(ctx, v.tableName)
	if err != nil {
		return nil, err
	}
	if tblNode == nil {
		return nil, sql.ErrTableNotFound.New(v.tableName)
	}

	// validate foreign key constraint
	if fkTbl, ok := tblNode.(sql.ForeignKeyTable); ok {
		fks, err := fkTbl.GetDeclaredForeignKeys(ctx)
		if err != nil {
			return nil, err
		}
		for _, fk := range fks {
			if strings.EqualFold(fk.Name, v.constraintName) {
				return v.validateForeignKey(ctx, db, fkTbl, fk)
			}
		}
	}

	// validate check constraint
	if checkTbl, ok := tblNode.(sql.CheckTable); ok {
		checks, err := checkTbl.GetChecks(ctx)
		if err != nil {
			return nil, err
		}
		for _, check := range checks {
			if strings.EqualFold(check.Name, v.constraintName) {
				return v.validateCheckConstraint(ctx, tblNode, check)
			}
		}
	}