fix

example.proto

@@ -68,3 +68,10 @@ message DownloadResp {}
 message UpdateUserInfoReq {}
 
 message UpdateUserInfoResp {}
+
+
+// @route_group: true
+// @route_api: /
+// @gen_to: ./
+service UserInfo {}
+

gobuf.go

@@ -2,6 +2,7 @@ package gobuf
 
 import (
 	"bytes"
+	"fmt"
 	"github.com/emicklei/proto"
 	"gitter.top/sync/proto-contrib/pkg/protofmt"
 	"io"
@@ -104,6 +105,26 @@ func (parser *Parser) AddRPC(serviceName, rpcName string) error {
 	return parser.writeSync()
 }
 
+func (parser *Parser) AddService(serviceName string) error {
+	if parser.ExistService(serviceName) {
+		return fmt.Errorf("service name exist")
+	}
+
+	parser.proto.Elements = append(parser.proto.Elements, &proto.Service{
+		Comment: &proto.Comment{
+			Lines: []string{
+				" @route_group: true",
+				" @base_url: /",
+				" @gen_to: ./",
+			},
+		},
+		Name:   serviceName,
+		Parent: parser.proto,
+	})
+
+	return parser.writeSync()
+}
+
 func (parser *Parser) writeSync() error {
 	var buf = new(bytes.Buffer)
 

gobuf_test.go

@@ -20,3 +20,10 @@ func TestParser_AddRPC(t *testing.T) {
 	err = parser.AddRPC("File", "UpdateUserInfo")
 	assert.Nil(t, err)
 }
+
+func TestParser_AddService(t *testing.T) {
+	parser, err := NewParser("example.proto")
+	assert.Nil(t, err)
+	err = parser.AddService("UserInfo")
+	assert.Nil(t, err)
+}