增加ip2region库管理

2025-11-17 06:00:25 +08:00 · 2020-11-04 15:50:53 +08:00
parent c3b54f25d5
commit a166d7e5f6
13 changed files with 408 additions and 0 deletions
--- a/internal/web/actions/default/servers/components/ip-library/uploadPopup.go
+++ b/internal/web/actions/default/servers/components/ip-library/uploadPopup.go
@@ -0,0 +1,106 @@
+package iplibrary
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+	"github.com/iwind/TeaGo/actions"
+	"io"
+)
+
+type UploadPopupAction struct {
+	actionutils.ParentAction
+}
+
+func (this *UploadPopupAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *UploadPopupAction) RunGet(params struct{}) {
+	this.Data["types"] = serverconfigs.IPLibraryTypes
+
+	this.Show()
+}
+
+func (this *UploadPopupAction) RunPost(params struct {
+	Type string
+	File *actions.File
+
+	Must *actions.Must
+}) {
+	libraryType := serverconfigs.FindIPLibraryWithType(params.Type)
+	if libraryType == nil {
+		this.Fail("错误的IP类型")
+	}
+
+	if params.File == nil {
+		this.Fail("请选择要上传的文件")
+	}
+
+	if params.File.Size == 0 {
+		this.Fail("文件内容不能为空")
+	}
+
+	if params.File.Ext != libraryType.GetString("ext") {
+		this.Fail("IP库文件扩展名错误，应该为：" + libraryType.GetString("ext"))
+	}
+
+	reader, err := params.File.OriginFile.Open()
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	defer func() {
+		_ = reader.Close()
+	}()
+
+	// 创建文件
+	fileResp, err := this.RPC().FileRPC().CreateFile(this.AdminContext(), &pb.CreateFileRequest{
+		Filename: params.File.Filename,
+		Size:     params.File.Size,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	fileId := fileResp.FileId
+
+	// 上传内容
+	buf := make([]byte, 512*1024)
+	for {
+		n, err := reader.Read(buf)
+		if n > 0 {
+			_, err = this.RPC().FileChunkRPC().CreateFileChunk(this.AdminContext(), &pb.CreateFileChunkRequest{
+				FileId: fileId,
+				Data:   buf[:n],
+			})
+			if err != nil {
+				this.Fail("上传失败：" + err.Error())
+			}
+		}
+		if err != nil {
+			if err == io.EOF {
+				break
+			}
+			this.Fail("上传失败：" + err.Error())
+		}
+	}
+
+	// 置为已完成
+	_, err = this.RPC().FileRPC().UpdateFileFinished(this.AdminContext(), &pb.UpdateFileFinishedRequest{FileId: fileId})
+	if err != nil {
+		this.ErrorPage(err)
+	}
+
+	// 保存
+	_, err = this.RPC().IPLibraryRPC().CreateIPLibrary(this.AdminContext(), &pb.CreateIPLibraryRequest{
+		Type:   params.Type,
+		FileId: fileId,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Success()
+}