Function generateUniqueConstraints

scripts/dbgen/constraint.go:144–174

generateUniqueConstraints generates the UniqueConstraint enum.

()

// generateUniqueConstraints generates the UniqueConstraint enum.
func generateUniqueConstraints() 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, "unique_constraint.go")

	fn := func(query string) ([]constraint, error) {
		if strings.Contains(query, "UNIQUE") || strings.Contains(query, "PRIMARY KEY") {
			name := ""
			switch {
			case strings.Contains(query, "ALTER TABLE") && strings.Contains(query, "ADD CONSTRAINT"):
				name = strings.Split(query, " ")[6]
			case strings.Contains(query, "CREATE UNIQUE INDEX"):
				name = strings.Split(query, " ")[3]
			default:
				return nil, xerrors.Errorf("unknown unique constraint format: %s", query)
			}
			return []constraint{
				{
					name:    name,
					comment: query,
				},
			}, nil
		}
		return nil, nil
	}
	return generateConstraints(dumpPath, outputPath, constraintTypeUnique, fn)
}

// generateForeignKeyConstraints generates the ForeignKeyConstraint enum.
func generateForeignKeyConstraints() error {