1 Star 0 Fork 0

Cruvie Kang / kk_go_kit

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
generate_api.go 3.70 KB
一键复制 编辑 原始数据 按行查看 历史
cruvie 提交于 2024-05-03 01:32 . update
package kk_http
import (
"fmt"
"log/slog"
"os"
)
type ApiModel struct {
ApiPkgName string
GroupFuncName string
HandlerName string
HandlerFuncName string
PreUrl string //for fe api
}
func GenerateApi(apiModel ApiModel) {
slog.Info("copy content of new_api.api to target file and delete new_api.go")
content := fmt.Sprintf(`
apiGroup.POST(kk_func.GetFunctionName(%s), %s)
`, apiModel.GroupFuncName, apiModel.GroupFuncName)
content += fmt.Sprintf(
`
// %s
//
// @Description %s
// @Accept octet-stream
// @Produce octet-stream
// @Param %sParam body %s.%sParam true "%sParam"
// @Success 200 {object} %s.%sResponse
// @Router %s%s [post]
func %s(c *gin.Context) {
stage := global_stage.GetRequestStage(c)
span := stage.StartTrace(kk_func.GetCurrentFunctionName())
defer span.End()
var param %s.%sParam
if err := param.BindCheck(stage); err != nil {
kk_http.ResponseErrPB(stage, err)
return
}
err, resp := %s.%s(stage, &param)
if err != nil {
kk_http.ResponseErrPB(stage, err)
return
}
kk_http.ResponseSuccessPB(stage, nil, resp)
}`, apiModel.GroupFuncName,
apiModel.GroupFuncName,
apiModel.HandlerFuncName, apiModel.ApiPkgName, apiModel.HandlerFuncName, apiModel.HandlerFuncName,
apiModel.ApiPkgName, apiModel.HandlerFuncName,
apiModel.PreUrl, apiModel.GroupFuncName,
apiModel.GroupFuncName,
apiModel.ApiPkgName, apiModel.HandlerFuncName,
apiModel.HandlerName, apiModel.HandlerFuncName)
content += fmt.Sprintf(`
func (Handler) %s(stage *kk_stage.Stage, param *%s.%sParam) (error, *%s.%sResponse) {
span := stage.StartTrace(kk_func.GetCurrentFunctionName())
defer span.End()
ser.%s(stage, param)
return err, nil
}
func (Service) %s(stage *kk_stage.Stage, param *%s.%sParam) error {
return err
}
`, apiModel.HandlerFuncName, apiModel.ApiPkgName, apiModel.HandlerFuncName, apiModel.ApiPkgName, apiModel.HandlerFuncName,
apiModel.HandlerFuncName,
apiModel.HandlerFuncName, apiModel.ApiPkgName, apiModel.HandlerFuncName)
filename := "go_api.api"
writeFile(filename, content)
}
func GenerateDartApi(apiModel ApiModel) {
content := fmt.Sprintf(
`
static Future<void> %s(%sParam param,
{Function(%sResponse)? okFunc,
Function(%sResponse)? errorFunc}) async {
%sResponse response = %sResponse();
PBResponse res = await HttpTool.postReq("%s%s",
queryParameters: param.writeToBuffer());
try {
if (res.code == 200) {
res.data.unpackInto(response);
if (okFunc != null) {
await okFunc(response);
}
} else if (errorFunc != null) {
await errorFunc(response);
}
} catch (e) {
log.info(e);
}
}`, apiModel.GroupFuncName, apiModel.HandlerFuncName,
apiModel.HandlerFuncName,
apiModel.HandlerFuncName,
apiModel.HandlerFuncName, apiModel.HandlerFuncName,
apiModel.PreUrl, apiModel.GroupFuncName)
filename := "dart_api.dart"
writeFile(filename, content)
}
func writeFile(filename string, content string) {
// 检查文件是否存在
_, err := os.Stat(filename)
if os.IsNotExist(err) {
// 文件不存在,创建并写入内容
err = os.WriteFile(filename, []byte(content), 0644)
if err != nil {
slog.Error(err.Error())
return
}
} else if err != nil {
// Stat函数返回了其他类型的错误
slog.Error(err.Error())
return
} else {
// 文件存在,以追加模式打开文件并写入内容
file, err := os.OpenFile(filename, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
if err != nil {
slog.Error(err.Error())
return
}
defer func(file *os.File) {
err := file.Close()
if err != nil {
slog.Error(err.Error())
}
}(file)
_, err = file.Write([]byte(content))
if err != nil {
slog.Error(err.Error())
return
}
}
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/cruvie/kk_go_kit.git
git@gitee.com:cruvie/kk_go_kit.git
cruvie
kk_go_kit
kk_go_kit
8ab331d8547b

搜索帮助

344bd9b3 5694891 D2dac590 5694891