Compare commits
6 commits
Author | SHA1 | Date | |
---|---|---|---|
|
aa1ead78e1 | ||
|
9367bab359 | ||
2b240d873b | |||
2e4b71e688 | |||
|
45ccbe091c | ||
|
01e4588bf7 |
2 changed files with 24 additions and 6 deletions
11
README.md
Normal file
11
README.md
Normal file
|
@ -0,0 +1,11 @@
|
||||||
|
# Usage
|
||||||
|
|
||||||
|
pgsplit takes a PostgreSQL dump with multiple databases through STDIN and extracts them to the current directory, named with the database name and an .sql extension.
|
||||||
|
|
||||||
|
`pgsplit -list` prints the database names contained in the SQL dump.
|
||||||
|
|
||||||
|
`pgsplit` extracts all databases.
|
||||||
|
|
||||||
|
`pgsplit -include-all -except foo -except bar` extracts all databases except for foo and bar.
|
||||||
|
|
||||||
|
`pgsplit -exclude-all -except foo -except bar` extracts only databases foo and bar.
|
19
main.go
19
main.go
|
@ -9,7 +9,7 @@ import (
|
||||||
"slices"
|
"slices"
|
||||||
)
|
)
|
||||||
|
|
||||||
const VERSION = "v3"
|
const VERSION = "v5"
|
||||||
const MAXLINE = 1048576
|
const MAXLINE = 1048576
|
||||||
|
|
||||||
type Db struct {
|
type Db struct {
|
||||||
|
@ -18,10 +18,12 @@ type Db struct {
|
||||||
file *os.File
|
file *os.File
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewDb(name string) (db Db, err error) {
|
func NewDb(name string) (db Db) {
|
||||||
db.Name = name
|
db.Name = name
|
||||||
fmt.Printf("Database %s\n", db.Name)
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (db *Db) Open() (err error) {
|
||||||
db.file, err = os.OpenFile(
|
db.file, err = os.OpenFile(
|
||||||
db.Name+".sql",
|
db.Name+".sql",
|
||||||
os.O_CREATE|os.O_WRONLY|os.O_TRUNC,
|
os.O_CREATE|os.O_WRONLY|os.O_TRUNC,
|
||||||
|
@ -64,12 +66,17 @@ func main() {
|
||||||
prevDb.Close()
|
prevDb.Close()
|
||||||
|
|
||||||
dbName := dbMatch[1]
|
dbName := dbMatch[1]
|
||||||
|
db = NewDb(dbName)
|
||||||
if (flagIncludeAll && slices.Contains(flagExcept, dbName)) || (flagExcludeAll && !slices.Contains(flagExcept, dbName)) {
|
if (flagIncludeAll && slices.Contains(flagExcept, dbName)) || (flagExcludeAll && !slices.Contains(flagExcept, dbName)) {
|
||||||
fmt.Printf("Skipping %s\n", dbName)
|
fmt.Printf("Skipping %s\n", dbName)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
if db, err = NewDb(dbName); err != nil {
|
fmt.Printf("Extracting %s\n", db.Name)
|
||||||
|
if flagList {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if err = db.Open(); err != nil {
|
||||||
fmt.Printf("%s: %s\n", dbName, err)
|
fmt.Printf("%s: %s\n", dbName, err)
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
|
@ -79,7 +86,7 @@ func main() {
|
||||||
db.Write(line)
|
db.Write(line)
|
||||||
}
|
}
|
||||||
|
|
||||||
if !flagList && dbDone.MatchString(line) {
|
if dbDone.MatchString(line) {
|
||||||
db.Completed = true
|
db.Completed = true
|
||||||
db.Close()
|
db.Close()
|
||||||
prevDb = db
|
prevDb = db
|
||||||
|
|
Loading…
Add table
Reference in a new issue