Fix proto formatter.

kujtimiihoxha · Jan 6, 2018 · a2437e7 · a2437e7
1 parent 219c308
commit a2437e7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/generator/add_transport.go b/generator/add_transport.go
@@ -22,6 +22,7 @@ import (
 	"github.com/kujtimiihoxha/kit/parser"
 	"github.com/kujtimiihoxha/kit/utils"
 	"github.com/spf13/viper"
+	"github.com/emicklei/proto-contrib/pkg/protofmt"
 )
 
 // GenerateTransport implement Gen, is used to generate a service transport
@@ -708,7 +709,7 @@ func (g *generateGRPCTransportProto) Generate() (err error) {
 	}
 	g.generateRequestResponse()
 	buf := new(bytes.Buffer)
-	formatter := proto.NewFormatter(buf, " ")
+	formatter := protofmt.NewFormatter(buf, " ")
 	formatter.Format(g.protoSrc)
 	err = g.fs.WriteFile(g.pbFilePath, buf.String(), true)
 	if err != nil {

diff --git a/glide.yaml b/glide.yaml
@@ -22,3 +22,4 @@ import:
   - imports
 - package: github.com/emicklei/proto
 - package: github.com/go-yaml/yaml
+- package: github.com/emicklei/proto-contrib