From 783b0fbb68b8b3de490f79ec98c492bc9d5d39b8 Mon Sep 17 00:00:00 2001
From: EZ4Jam1n <1286884552@qq.com>
Date: Sun, 11 Aug 2024 10:16:55 +0800
Subject: [PATCH] fix: lint error

---
 thrift-gen-rpc-swagger/generator/openapi_gen.go | 3 +--
 thrift-gen-rpc-swagger/generator/server_gen.go  | 9 ++++-----
 2 files changed, 5 insertions(+), 7 deletions(-)

diff --git a/thrift-gen-rpc-swagger/generator/openapi_gen.go b/thrift-gen-rpc-swagger/generator/openapi_gen.go
index 94ecf91..78767ad 100644
--- a/thrift-gen-rpc-swagger/generator/openapi_gen.go
+++ b/thrift-gen-rpc-swagger/generator/openapi_gen.go
@@ -34,7 +34,6 @@
 package generator
 
 import (
-	"fmt"
 	"path/filepath"
 	"regexp"
 	"sort"
@@ -214,7 +213,7 @@ func (g *OpenAPIGenerator) BuildDocument(arguments *args.Arguments) []*plugin.Ge
 
 	bytes, err := d.YAMLValue("Generated with thrift-gen-rpc-swagger\n" + infoURL)
 	if err != nil {
-		fmt.Printf("Error converting to yaml: %s\n", err)
+		logs.Errorf("Error converting to yaml: %s", err)
 		return nil
 	}
 	filePath := filepath.Clean(arguments.OutputDir)
diff --git a/thrift-gen-rpc-swagger/generator/server_gen.go b/thrift-gen-rpc-swagger/generator/server_gen.go
index 7b807e7..11f33a9 100644
--- a/thrift-gen-rpc-swagger/generator/server_gen.go
+++ b/thrift-gen-rpc-swagger/generator/server_gen.go
@@ -18,11 +18,10 @@ package generator
 
 import (
 	"bytes"
-	"fmt"
-	"os"
 	"path/filepath"
 	"text/template"
 
+	"github.com/cloudwego/hertz/cmd/hz/util/logs"
 	"github.com/cloudwego/thriftgo/parser"
 	"github.com/cloudwego/thriftgo/plugin"
 	"github.com/hertz-contrib/swagger-generate/thrift-gen-rpc-swagger/args"
@@ -42,7 +41,7 @@ func NewServerGenerator(ast *parser.Thrift, args *args.Arguments) *ServerGenerat
 
 	idlPath := ast.Filename
 	if idlPath == "" {
-		fmt.Fprintf(os.Stderr, "IDL file path is empty")
+		logs.Errorf("IDL file path is empty")
 	}
 
 	hertzAddr := args.HertzAddr
@@ -71,13 +70,13 @@ func NewServerGenerator(ast *parser.Thrift, args *args.Arguments) *ServerGenerat
 func (g *ServerGenerator) Generate() []*plugin.Generated {
 	tmpl, err := template.New("server").Delims("{{", "}}").Parse(serverTemplate)
 	if err != nil {
-		fmt.Sprintf("failed to parse template: %v", err)
+		logs.Errorf("failed to parse template: %v", err)
 	}
 
 	var buf bytes.Buffer
 	err = tmpl.Execute(&buf, g)
 	if err != nil {
-		fmt.Sprintf("failed to execute template: %v", err)
+		logs.Errorf("failed to execute template: %v", err)
 	}
 
 	filePath := filepath.Clean(g.OutputDir)