Function generateForeignKeyConstraints

scripts/dbgen/constraint.go:177–205

generateForeignKeyConstraints generates the ForeignKeyConstraint enum.

()

// generateForeignKeyConstraints generates the ForeignKeyConstraint enum.
func generateForeignKeyConstraints() error {
	localPath, err := localFilePath()
	if err != nil {
		return err
	}
	databasePath := filepath.Join(localPath, "..", "..", "..", "coderd", "database")
	dumpPath := filepath.Join(databasePath, "dump.sql")
	outputPath := filepath.Join(databasePath, "foreign_key_constraint.go")

	fn := func(query string) ([]constraint, error) {
		if strings.Contains(query, "FOREIGN KEY") {
			name := ""
			switch {
			case strings.Contains(query, "ALTER TABLE") && strings.Contains(query, "ADD CONSTRAINT"):
				name = strings.Split(query, " ")[6]
			default:
				return nil, xerrors.Errorf("unknown foreign key constraint format: %s", query)
			}
			return []constraint{
				{
					name:    name,
					comment: query,
				},
			}, nil
		}
		return []constraint{}, nil
	}
	return generateConstraints(dumpPath, outputPath, constraintTypeForeignKey, fn)
}

// generateCheckConstraints generates the CheckConstraint enum.
func generateCheckConstraints() error {