Function exportResultToZip

backend/api/v1/sql_service.go:1257–1283

exportResultToZip exports a single query result to the ZIP archive. It writes both the SQL statement and the formatted result data.

(
	ctx context.Context,
	zipw *zip.Writer,
	stores *store.Store,
	instance *store.InstanceMessage,
	database *store.DatabaseMessage,
	result *v1pb.QueryResult,
	request *v1pb.ExportRequest,
	statementNumber int,
)

// exportResultToZip exports a single query result to the ZIP archive.
// It writes both the SQL statement and the formatted result data.
func exportResultToZip(
	ctx context.Context,
	zipw *zip.Writer,
	stores *store.Store,
	instance *store.InstanceMessage,
	database *store.DatabaseMessage,
	result *v1pb.QueryResult,
	request *v1pb.ExportRequest,
	statementNumber int,
) error {
	baseFilename := fmt.Sprintf("%s/%s/statement-%d", database.InstanceID, database.DatabaseName, statementNumber)

	// Write statement file
	statementFilename := fmt.Sprintf("%s.sql", baseFilename)
	if err := export.WriteZipEntry(zipw, statementFilename, []byte(result.Statement), request.GetPassword()); err != nil {
		return errors.Wrap(err, "failed to write statement")
	}

	// Write result file by streaming directly to ZIP
	resultExt := strings.ToLower(request.Format.String())
	resultFilename := fmt.Sprintf("%s.result.%s", baseFilename, resultExt)
	if err := formatExportToZip(ctx, zipw, resultFilename, stores, instance, database, result, request); err != nil {
		return errors.Wrap(err, "failed to write formatted result")
	}

	return nil
}

// formatExportToZip formats query results and writes them directly to a ZIP entry.
// This function streams the formatted data to minimize memory usage.