1

utils/basic/basic.go

@@ -13,9 +13,10 @@ var (
 	CodeServiceErr = &StatusResponse{510, "server logic error"} // server logic 错误
 	CodeUnAuth     = &StatusResponse{401, "unauthorized"}       // 未授权
 
-	CodeUpdateErr        = &StatusResponse{5000, "update database error"}                                             // update database logic 错误
-	CodeDupGuestErr      = &StatusResponse{5001, "the user is already a guest user and does not need to apply again"} // 用户已经是guest用户不需要重复申请 错误
-	CodeRequestParamsErr = &StatusResponse{5002, "invalid request param"}                                             // api参数校验 错误
+	CodeUpdateErr           = &StatusResponse{5000, "update database error"}                                             // update database logic 错误
+	CodeDupGuestErr         = &StatusResponse{5001, "the user is already a guest user and does not need to apply again"} // 用户已经是guest用户不需要重复申请 错误
+	CodeRequestParamsErr    = &StatusResponse{5002, "invalid request param"}                                             // api参数校验 错误
+	CodeDbRecordNotFoundErr = &StatusResponse{5002, "invalid request param"}
 )
 
 type Response struct {

utils/pdf/html_to_pdf.go

@@ -1,15 +1,16 @@
 package pdf
 
 import (
+	"encoding/base64"
 	"github.com/SebastiaanKlippert/go-wkhtmltopdf"
 	"strings"
 )
 
 // html转 Pdf dataType = 1 为网页url dataType = 2为网页内容 outFile为空则不保存
-func HtmlToPdf(content string, dataType int, outFile ...string) ([]byte, error) {
+func HtmlToPdfBase64(content string, dataType int, outFile ...string) (string, error) {
 	pdfg, err := wkhtmltopdf.NewPDFGenerator()
 	if err != nil {
-		return nil, err
+		return "", err
 	}
 	switch dataType {
 	case 1: //网页地址
@@ -23,13 +24,13 @@ func HtmlToPdf(content string, dataType int, outFile ...string) ([]byte, error)
 	pdfg.PageSize.Set(wkhtmltopdf.PageSizeA4)
 	// Create PDF document in memory
 	if err = pdfg.Create(); err != nil {
-		return nil, err
+		return "", err
 	}
 	// Write PDF to file
 	if len(outFile) != 0 {
 		if err = pdfg.WriteFile(outFile[0]); err != nil {
-			return nil, err
+			return "", err
 		}
 	}
-	return pdfg.Bytes(), nil
+	return base64.StdEncoding.EncodeToString(pdfg.Bytes()), nil
 }