add middleware

common/api.go

@@ -190,12 +190,21 @@ func (s *SchemaApi) String() string {
 @server(
 	prefix: v1/%s
 	group: %s
-	jwt: Auth
+	%s
+	%s
 )`
+	jwt := ""
+	if s.pconf.Jwt {
+		jwt = "jwt:Auth"
+	}
+	middleware := ""
+	if len(s.pconf.Middleware) > 0 {
+		middleware = "middleware:" + strings.Join(s.pconf.Middleware, ",")
+	}
 
 	for _, tab := range s.Tables {
 		buf.WriteString("//--------------------------------" + tab.Comment + "--------------------------------")
-		buf.WriteString(fmt.Sprintf(temp, s.ServiceName, tab.TableName))
+		buf.WriteString(fmt.Sprintf(temp, s.ServiceName, tab.TableName, jwt, middleware))
 		buf.WriteString("\n")
 		buf.WriteString("service " + s.ServiceName + "-api{")
 		buf.WriteString("\n")

common/types.go

@@ -19,7 +19,8 @@ type ProtoConfig struct {
 	GoPackageName string   `yaml:"goPackageName"`
 	OutFile       string   `yaml:"outFile"`
 	IgnoreTables  []string `yaml:"ignoreTables"`
-
+	Jwt           bool     `yaml:"jwt"`
+	Middleware    []string `yaml:"middleware"`    //中间件
 	OnlySearch    []string `yaml:"onlySearch"`    //查询时出现，但增改时不会出现的字段，如id,created_at,updated_at
 	IgnoreColumns []string `yaml:"ignoreColumns"` //都不会出现的字段,如deleted_at,version
 

yaml/api.yaml

@@ -10,6 +10,9 @@ tables:
   - merchant_users
 serviceName: merchant
 outFile: merchant.api
+jwt: false
+middleware:
+  - CheckRole
 ignoreColumns: 
   - deleted_at 
   - version