@ -1,7 +1,10 @@
package parser
import (
"bufio"
"bytes"
"fmt"
"io"
"strings"
"github.com/tal-tech/go-zero/tools/goctl/api/spec"
@ -54,32 +57,49 @@ type serviceEntityParser struct {
}
func ( p * serviceEntityParser ) parseLine ( line string , api * spec . ApiSpec , annos [ ] spec . Annotation ) error {
fields := strings . Fields ( line )
if len ( fields ) < 2 {
return fmt . Errorf ( "wrong line %q" , line )
line = strings . TrimSpace ( line )
var buffer = new ( bytes . Buffer )
buffer . WriteString ( line )
reader := bufio . NewReader ( buffer )
var builder strings . Builder
var fields [ ] string
for {
ch , _ , err := reader . ReadRune ( )
if err != nil {
if err == io . EOF {
break
}
return err
}
switch {
case isSpace ( ch ) , ch == leftParenthesis , ch == rightParenthesis , ch == semicolon :
if builder . Len ( ) == 0 {
continue
}
token := builder . String ( )
builder . Reset ( )
fields = append ( fields , token )
default :
builder . WriteRune ( ch )
}
}
if len ( fields ) < 3 {
return fmt . Errorf ( "wrong line %q, %q" , line , routeSyntax )
}
method := fields [ 0 ]
pathAndRequest := fields [ 1 ]
pos := strings . Index ( pathAndRequest , "(" )
if pos < 0 {
return fmt . Errorf ( "wrong line %q" , line )
}
path := strings . TrimSpace ( pathAndRequest [ : pos ] )
pathAndRequest = pathAndRequest [ pos + 1 : ]
pos = strings . Index ( pathAndRequest , ")" )
if pos < 0 {
return fmt . Errorf ( "wrong line %q" , line )
}
req := pathAndRequest [ : pos ]
path := fields [ 1 ]
req := fields [ 2 ]
var returns string
if len ( fields ) > 2 {
returns = fields [ 2 ]
if len ( fields ) > 4 {
returns = fields [ 4 ]
if fields [ 3 ] != returnsTag {
return fmt . Errorf ( "wrong line %q, %q" , line , routeSyntax )
}
}
returns = strings . ReplaceAll ( returns , "returns" , "" )
returns = strings . ReplaceAll ( returns , "(" , "" )
returns = strings . ReplaceAll ( returns , ")" , "" )
returns = strings . TrimSpace ( returns )
p . acceptRoute ( spec . Route {
Annotations : annos ,