27 lines
565 B
Go
27 lines
565 B
Go
package build
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"git.secnex.io/secnex/pgson/schema"
|
|
"git.secnex.io/secnex/pgson/utils"
|
|
)
|
|
|
|
func TruncateSQL(s *schema.Table, cascade bool, restartIdentity bool) (string, error) {
|
|
if s == nil {
|
|
return "", fmt.Errorf("nil table provided")
|
|
}
|
|
if s.Name == "" || !utils.IsValidIdentifier(s.Name) {
|
|
return "", fmt.Errorf("invalid table name: %q", s.Name)
|
|
}
|
|
|
|
query := fmt.Sprintf("TRUNCATE TABLE %s", utils.SQLQuoteIdent(s.Name))
|
|
if restartIdentity {
|
|
query += " RESTART IDENTITY"
|
|
}
|
|
if cascade {
|
|
query += " CASCADE"
|
|
}
|
|
return query, nil
|
|
}
|