proper support for inner join query creation

pkg/config/config.go

@@ -13,6 +13,7 @@ type (
 	// Table represents a klepto table definition
 	Table struct {
 		Name          string
+		PrimaryKey    string
 		IgnoreData    bool
 		Filter        Filter
 		Anonymise     map[string]string
@@ -28,9 +29,10 @@ type (
 
 	// Relationship represents a relationship definition
 	Relationship struct {
+		Table           string
+		ForeignKey      string
 		ReferencedTable string
 		ReferencedKey   string
-		ForeignKey      string
 	}
 )
 

pkg/dumper/generic/sql.go

@@ -94,6 +94,7 @@ func (p *sqlDumper) readAndDumpTables(done chan<- struct{}, configTables config.
 			}
 
 			opts = reader.ReadTableOpt{
+				PrimaryKey:    tableConfig.PrimaryKey,
 				Match:         tableConfig.Filter.Match,
 				Limit:         tableConfig.Filter.Limit,
 				Relationships: p.relationshipConfigToOptions(tableConfig.Relationships),
@@ -144,6 +145,7 @@ func (p *sqlDumper) relationshipConfigToOptions(relationshipsConfig []*config.Re
 
 	for _, r := range relationshipsConfig {
 		opts = append(opts, &reader.RelationshipOpt{
+			Table:           r.Table,
 			ReferencedTable: r.ReferencedTable,
 			ReferencedKey:   r.ReferencedKey,
 			ForeignKey:      r.ForeignKey,

pkg/reader/generic/sql.go

@@ -117,20 +117,25 @@ func (s *SqlReader) buildQuery(tableName string, opts reader.ReadTableOpt) (sq.S
 	var query sq.SelectBuilder
 
 	query = sq.Select(opts.Columns...).From(s.QuoteIdentifier(tableName))
-
 	for _, r := range opts.Relationships {
+		if r.Table == "" {
+			r.Table = tableName
+		}
 		query = query.Join(fmt.Sprintf(
 			"%s ON %s.%s = %s.%s",
 			r.ReferencedTable,
-			tableName,
-			r.ForeignKey,
 			r.ReferencedTable,
 			r.ReferencedKey,
+			r.Table,
+			r.ForeignKey,
 		))
 	}
 
 	if len(opts.Relationships) > 0 {
-		query = query.GroupBy(fmt.Sprintf("%s.id", tableName))
+		if opts.PrimaryKey == "" {
+			opts.PrimaryKey = "id"
+		}
+		query = query.GroupBy(fmt.Sprintf("%s.%s", tableName, opts.PrimaryKey))
 	}
 
 	if opts.Match != "" {

pkg/reader/reader.go

@@ -31,6 +31,8 @@ type (
 
 	// ReadTableOpt represents the read table options
 	ReadTableOpt struct {
+		// PrimaryKey is the primary key
+		PrimaryKey string
 		// Columns contains the (quoted) column of the table
 		Columns []string
 		// Match is a condition field to dump only certain amount data
@@ -43,9 +45,10 @@ type (
 
 	// RelationshipOpt represents the relationships options
 	RelationshipOpt struct {
+		Table           string
+		ForeignKey      string
 		ReferencedTable string
 		ReferencedKey   string
-		ForeignKey      string
 	}
 
 	// ConnOpts are the options to create a connection