Function generateCheckConstraints

scripts/dbgen/constraint.go:208–241

generateCheckConstraints generates the CheckConstraint enum.

()

// generateCheckConstraints generates the CheckConstraint enum.
func generateCheckConstraints() 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, "check_constraint.go")

	var (
		tableRegex = regexp.MustCompile(`CREATE TABLE\s+([^\s]+)`)
		checkRegex = regexp.MustCompile(`CONSTRAINT\s+([^\s]+)\s+CHECK`)
	)
	fn := func(query string) ([]constraint, error) {
		constraints := []constraint{}

		tableMatches := tableRegex.FindStringSubmatch(query)
		if len(tableMatches) > 0 {
			table := tableMatches[1]

			// Find every CONSTRAINT xxx CHECK occurrence.
			matches := checkRegex.FindAllStringSubmatch(query, -1)
			for _, match := range matches {
				constraints = append(constraints, constraint{
					name:    match[1],
					comment: table,
				})
			}
		}
		return constraints, nil
	}

	return generateConstraints(dumpPath, outputPath, constraintTypeCheck, fn)
}