feat: fixed service and func order

sorcererxw · Sep 29, 2021 · 64c56be · 64c56be
1 parent 84d9d82
commit 64c56be
Show file tree

Hide file tree

Showing 4 changed files with 84 additions and 66 deletions.
diff --git a/example/another.pb.go b/example/another.pb.go
diff --git a/example/petstore.pb.go b/example/petstore.pb.go
diff --git a/example/petstore_grpc_mock.pb.go b/example/petstore_grpc_mock.pb.go
diff --git a/main.go b/main.go
@@ -4,6 +4,7 @@ import (
 	_ "embed"
 	"fmt"
 	pathpkg "path"
+	"sort"
 	"text/template"
 
 	"google.golang.org/protobuf/compiler/protogen"
@@ -62,6 +63,23 @@ func parseFile(path string, file *protogen.File) (data Data) {
 	for k, v := range imports {
 		data.Imports = append(data.Imports, fmt.Sprintf(`%s "%s"`, v, k))
 	}
+
+	return orderData(data)
+}
+
+func orderData(data Data) Data {
+	services := data.Services
+	sort.Slice(services, func(i, j int) bool {
+		return services[i].Service < services[j].Service
+	})
+	for k := range services {
+		funcs := services[k].Funcs
+		sort.Slice(funcs, func(i, j int) bool {
+			return funcs[i].Func < funcs[j].Func
+		})
+		services[k].Funcs = funcs
+	}
+	data.Services = services
 	return data
 }