Support read only source database

Read only database does not expose detailed schema so we have to build it from ground up.

.gitignore

@@ -1,3 +1,5 @@
 .pgsql/data
 bin
 dist/
+*.sql
+.nix-profile*

.goreleaser.yaml

@@ -1,17 +1,22 @@
 before:
   hooks:
     - go mod tidy
+universal_binaries:
+  - replace: true
+
 builds:
   - env:
       - CGO_ENABLED=0
     goos:
       - linux
-      - windows
       - darwin
     main: ./cli
-
+    goarch:
+      - amd64
+      - arm64
 archives:
   - format: tar.gz
+    strip_parent_binary_folder: true
     # this name template makes the OS and Arch compatible with the results of uname.
     name_template: >-
       {{ .ProjectName }}_

.vscode/settings.json

@@ -0,0 +1,11 @@
+{
+    "spellright.language": [
+        "en"
+    ],
+    "spellright.documentTypes": [
+        "markdown",
+        "latex",
+        "plaintext",
+        "go"
+    ]
+}

Makefile

@@ -14,7 +14,7 @@ is-postgres-running:
 
 .PHONY: pgweb
 pgweb:is-postgres-running
-	@pgweb --url "postgres://test_source@localhost:5432/test_source?sslmode=disable"
+	@pgweb --url "postgres://test_target@localhost:5432/test_target?sslmode=disable"
 
 build:
 	rm -rf dist

README.md

@@ -3,45 +3,65 @@
 [![lint](https://github.com/teamniteo/pg-subsetter/actions/workflows/lint.yml/badge.svg)](https://github.com/teamniteo/pg-subsetter/actions/workflows/lint.yml) [![build](https://github.com/teamniteo/pg-subsetter/actions/workflows/go.yml/badge.svg)](https://github.com/teamniteo/pg-subsetter/actions/workflows/go.yml) [![vuln](https://github.com/teamniteo/pg-subsetter/actions/workflows/vuln.yml/badge.svg)](https://github.com/teamniteo/pg-subsetter/actions/workflows/vuln.yml)
 
 
-`pg-subsetter` is a tool designed to synchronize a fraction of a PostgreSQL database to another PostgreSQL database on the fly, it does not copy the SCHEMA. This means that your target database has to have schema populated in some other way.
+`pg-subsetter` is a tool designed to synchronize a fraction of a PostgreSQL database to another PostgreSQL database on the fly, it does not copy the SCHEMA. 
+
 
 ### Database Fraction Synchronization
 `pg-subsetter` allows you to select and sync a specific subset of your database. Whether it's a fraction of a table or a particular dataset, you can have it replicated in another database without synchronizing the entire DB.
 
 ### Integrity Preservation with Foreign Keys
-Foreign keys play a vital role in maintaining the relationships between tables. `pg-subsetter` ensures that all foreign keys are handled correctly during the synchronization process, maintaining the integrity and relationships of the data.
+Foreign keys play a vital role in maintaining the relationships between tables. `pg-subsetter` ensures that all foreign keys(one-to-one, one-to many, many-to-many) are handled correctly during the synchronization process, maintaining the integrity and relationships of the data. 
 
 ### Efficient COPY Method
 Utilizing the native PostgreSQL COPY command, `pg-subsetter` performs data transfer with high efficiency. This method significantly speeds up the synchronization process, minimizing downtime and resource consumption.
 
 ### Stateless Operation
 `pg-subsetter` is built to be stateless, meaning it does not maintain any internal state between runs. This ensures that each synchronization process is independent, enhancing reliability and making it easier to manage and scale.
 
+### Sync required rows
+`pg-subsetter` can be instructed to copy certain rows in specific tables, the command can be used multiple times to sync more data.
 
 ## Usage
 
 ```
 Usage of subsetter:
   -dst string
-        Destination database DSN
+    	Destination database DSN
+  -exclude value
+    	Query to ignore tables 'users: id = 1', can be used multiple times
   -f float
-        Fraction of rows to copy (default 0.05)
-  -force value
-        Query to copy required tables (users: id = 1)
+    	Fraction of rows to copy (default 0.05)
+  -include value
+    	Query to copy required tables 'users: id = 1', can be used multiple times
   -src string
-        Source database DSN
+    	Source database DSN
+  -v	Release information
+  -verbose
+    	Show more information during sync (default true)
 ```
 
 
 ### Example
 
-Copy a fraction of the database and force certain rows to be also copied over.
+
+Prepare schema in target database:
+
+```bash
+pg_dump --schema-only -n public -f schemadump.sql "postgres://test_source@localhost:5432/test_source?sslmode=disable"
+psql -f schemadump.sql "postgres://test_target@localhost:5432/test_target?sslmode=disable"
+```
+
+Copy a fraction of the database and force certain rows to be also copied over:
 
 ```
 pg-subsetter \
       -src "postgres://test_source@localhost:5432/test_source?sslmode=disable" \
       -dst "postgres://test_target@localhost:5432/test_target?sslmode=disable" \
-      -f 0.05
+      -f 0.5
+      -include "user: id=1"
+      -include "group: id=1"
+      -exclude "domains: domain_name ilike '%.si'"
+
 ```
 
 # Installing

cli/extra.go

@@ -0,0 +1,24 @@
+package main
+
+import (
+	"fmt"
+	"strings"
+
+	"niteo.co/subsetter/subsetter"
+)
+
+type arrayExtra []subsetter.Rule
+
+func (i *arrayExtra) String() string {
+	return fmt.Sprintf("%v", *i)
+}
+
+func (i *arrayExtra) Set(value string) error {
+	q := strings.Split(strings.TrimSpace(value), ":")
+
+	*i = append(*i, subsetter.Rule{
+		Table: strings.TrimSpace(q[0]),
+		Where: strings.TrimSpace(q[1]),
+	})
+	return nil
+}

cli/extra_test.go

@@ -0,0 +1,29 @@
+package main
+
+import (
+	"fmt"
+	"testing"
+)
+
+func Test_arrayExtra_Set(t *testing.T) {
+
+	tests := []struct {
+		name    string
+		value   string
+		rules   arrayExtra
+		wantErr bool
+	}{
+		{"With tables", "simple: id < 10", arrayExtra{{Table: "simple", Where: "id < 10"}}, false},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			r := arrayExtra{}
+			if err := r.Set(tt.value); (err != nil) != tt.wantErr {
+				t.Errorf("arrayExtra.Set() error = %v, wantErr %v", err, tt.wantErr)
+			}
+			if fmt.Sprintf("%v", r) != fmt.Sprintf("%v", tt.rules) {
+				t.Errorf("arrayExtra.Set() = %v, want %v", r, tt.rules)
+			}
+		})
+	}
+}

cli/main.go

@@ -10,19 +10,33 @@ import (
 	"niteo.co/subsetter/subsetter"
 )
 
+var (
+	version = "dev"
+	commit  = "none"
+	date    = "unknown"
+)
+
 var src = flag.String("src", "", "Source database DSN")
 var dst = flag.String("dst", "", "Destination database DSN")
 var fraction = flag.Float64("f", 0.05, "Fraction of rows to copy")
 var verbose = flag.Bool("verbose", true, "Show more information during sync")
-var forceSync arrayForce
+var ver = flag.Bool("v", false, "Release information")
+var extraInclude arrayExtra
+var extraExclude arrayExtra
 
 func main() {
 	log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr})
 	zerolog.ErrorStackMarshaler = pkgerrors.MarshalStack
 
-	flag.Var(&forceSync, "force", "Query to copy required tables (users: id = 1)")
+	flag.Var(&extraInclude, "include", "Query to copy required tables 'users: id = 1', can be used multiple times")
+	flag.Var(&extraExclude, "exclude", "Query to ignore tables 'users: id = 1', can be used multiple times")
 	flag.Parse()
 
+	if *ver {
+		log.Info().Str("version", version).Str("commit", commit).Str("date", date).Msg("Version")
+		os.Exit(0)
+	}
+
 	if *src == "" || *dst == "" {
 		log.Fatal().Msg("Source and destination DSNs are required")
 	}
@@ -31,20 +45,23 @@ func main() {
 		log.Fatal().Msg("Fraction must be between 0 and 1")
 	}
 
-	if len(forceSync) > 0 {
-		log.Info().Str("forced", forceSync.String()).Msg("Forcing sync for tables")
+	if len(extraInclude) > 0 {
+		log.Info().Str("include", extraInclude.String()).Msg("Forcibly including")
+	}
+	if len(extraExclude) > 0 {
+		log.Info().Str("exclude", extraExclude.String()).Msg("Forcibly ignoring")
 	}
 
-	s, err := subsetter.NewSync(*src, *dst, *fraction, forceSync, *verbose)
+	s, err := subsetter.NewSync(*src, *dst, *fraction, extraInclude, extraExclude, *verbose)
 	if err != nil {
-		log.Fatal().Stack().Err(err).Msg("Failed to configure sync")
+		log.Fatal().Err(err).Msg("Failed to configure sync")
 	}
 
 	defer s.Close()
 
 	err = s.Sync()
 	if err != nil {
-		log.Fatal().Stack().Err(err).Msg("Failed to sync")
+		log.Fatal().Err(err).Msg("Failed to sync")
 	}
 
 }

flake.nix

@@ -3,7 +3,7 @@
     allowed-users = [ "@wheel" "@staff" ]; # allow compiling on every device/machine
   };
   inputs = {
-    nixpkgs.url = "github:NixOS/nixpkgs/release-23.05";
+    nixpkgs.url = "github:NixOS/nixpkgs/nixpkgs-unstable";
     flake-parts.url = "github:hercules-ci/flake-parts";
   };
   outputs = inputs@{ self, nixpkgs, flake-parts, ... }:
@@ -35,7 +35,7 @@
               go
               goreleaser
               golangci-lint
-              postgresql
+              postgresql_15
               process-compose
               nixpkgs-fmt
               pgweb

go.mod

@@ -2,18 +2,21 @@ module niteo.co/subsetter
 
 go 1.20
 
-require github.com/rs/zerolog v1.30.0
+require (
+	github.com/pkg/errors v0.9.1
+	github.com/rs/zerolog v1.30.0
+)
 
 require (
 	github.com/jackc/pgpassfile v1.0.0 // indirect
 	github.com/jackc/pgservicefile v0.0.0-20221227161230-091c0ba34f0a // indirect
 	github.com/jackc/puddle/v2 v2.2.1 // indirect
-	github.com/pkg/errors v0.9.1 // indirect
 	golang.org/x/exp v0.0.0-20220303212507-bbda1eaf7a17 // indirect
 	golang.org/x/sync v0.1.0 // indirect
 )
 
 require (
+	github.com/davecgh/go-spew v1.1.1
 	github.com/jackc/pgx/v5 v5.4.3
 	github.com/mattn/go-colorable v0.1.12 // indirect
 	github.com/mattn/go-isatty v0.0.14 // indirect

go.sum

@@ -1,6 +1,7 @@
 github.com/coreos/go-systemd/v22 v22.5.0/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=
 github.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM=
 github.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg=