feat(database): add Limit for Export

oom-ai · Jan 18, 2022 · 6dfbef0 · 6dfbef0
1 parent 360d928
commit 6dfbef0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/internal/database/offline/sqlutil/export.go b/internal/database/offline/sqlutil/export.go
@@ -112,6 +112,9 @@ func DoExport(ctx context.Context, dbOpt dbutil.DBOpt, opt DoExportOpt) (*types.
 	if err != nil {
 		return nil, err
 	}
+	if opt.Limit != nil {
+		query = fmt.Sprintf("%s LIMIT %d", query, *opt.Limit)
+	}
 	args := make([]interface{}, 0, len(opt.CdcTables)*2)
 	for i := 0; i < len(opt.CdcTables)*2; i++ {
 		args = append(args, opt.UnixMilli)

diff --git a/internal/database/offline/sqlutil/export_helper.go b/internal/database/offline/sqlutil/export_helper.go
@@ -53,7 +53,6 @@ LEFT JOIN
 ) AS {{ qt $t0 }}
 ON e.{{ qt $.EntityName }} = {{ qt $t0 }}.{{ qt $.EntityName }}
 {{end}}
-
 `
 
 type unionEntityQueryParams struct {
-Original file line number
+Diff line change
@@ Expand Up / @@ -53,7 +53,6 @@ LEFT JOIN @@
     ) AS {{ qt $t0 }}
     ON e.{{ qt $.EntityName }} = {{ qt $t0 }}.{{ qt $.EntityName }}
     {{end}}
     `
     type unionEntityQueryParams struct {
@@ Expand Down @@