fix: unordered columns of export

crawlab-team · Jul 2, 2023 · f42ee05 · f42ee05
1 parent afb0e8d
commit f42ee05
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/export/csv_service.go b/export/csv_service.go
@@ -19,6 +19,7 @@ import (
 	mongo2 "go.mongodb.org/mongo-driver/mongo"
 	"os"
 	"path"
+	"sort"
 	"strconv"
 	"time"
 )
@@ -261,9 +262,23 @@ func (svc *CsvService) getColumns(query bson.M, export interfaces.Export) (colum
 	// columns
 	columns = make([]string, 0, len(columnsSet))
 	for k := range columnsSet {
+		// skip task key
+		if k == constants.TaskKey {
+			continue
+		}
+
+		// skip _id
+		if k == "_id" {
+			continue
+		}
+
+		// append to columns
 		columns = append(columns, k)
 	}
 
+	// order columns
+	sort.Strings(columns)
+
 	return columns, nil
 }