init: 1.0.0

2025-04-25 12:08:29 +09:00
commit 8addcc584b
1031 changed files with 76472 additions and 0 deletions
@@ -0,0 +1,131 @@
+package common
+
+import (
+	"context"
+	"encoding/json"
+	"io"
+	"net"
+	"net/http"
+	"slices"
+	"strings"
+	"time"
+
+	"github.com/perfect-panel/ppanel-server/internal/config"
+	"github.com/perfect-panel/ppanel-server/internal/model/server"
+	"github.com/perfect-panel/ppanel-server/internal/model/user"
+	"github.com/perfect-panel/ppanel-server/internal/svc"
+	"github.com/perfect-panel/ppanel-server/internal/types"
+	"github.com/perfect-panel/ppanel-server/pkg/logger"
+	"github.com/perfect-panel/ppanel-server/pkg/xerr"
+	"github.com/pkg/errors"
+)
+
+type GetStatLogic struct {
+	logger.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+// Get Tos
+func NewGetStatLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetStatLogic {
+	return &GetStatLogic{
+		Logger: logger.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *GetStatLogic) GetStat() (resp *types.GetStatResponse, err error) {
+	respJson, err := l.svcCtx.Redis.Get(l.ctx, config.CommonStatCacheKey).Result()
+	if err == nil {
+		err = json.Unmarshal([]byte(respJson), resp)
+		if err == nil {
+			return
+		}
+	}
+	var u int64
+	err = l.svcCtx.DB.Model(&user.User{}).Where("enable = 1").Count(&u).Error
+	if err != nil {
+		l.Logger.Error("[GetStatLogic] get user count failed: ", logger.Field("error", err.Error()))
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "get user count failed: %v", err.Error())
+	}
+	if u > 100 {
+		u -= u % 100
+	} else if u > 10 {
+		u -= u % 10
+	} else {
+		u = 1
+	}
+	var n int64
+	err = l.svcCtx.DB.Model(&server.Server{}).Where("enable = 1").Count(&n).Error
+	if err != nil {
+		l.Logger.Error("[GetStatLogic] get server count failed: ", logger.Field("error", err.Error()))
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "get server count failed: %v", err.Error())
+	}
+	var nodeaddr []string
+	err = l.svcCtx.DB.Model(&server.Server{}).Where("enable = 1").Pluck("server_addr", &nodeaddr).Error
+	if err != nil {
+		l.Logger.Error("[GetStatLogic] get server_addr failed: ", logger.Field("error", err.Error()))
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "get server_addr failed: %v", err.Error())
+	}
+	type apireq struct {
+		query  string
+		fields string
+	}
+	type apiret struct {
+		CountryCode string `json:"countryCode"`
+	}
+	//map as dict
+	type void struct{}
+	var v void
+	country := make(map[string]void)
+	for c := range slices.Chunk(nodeaddr, 100) {
+		var batchreq []apireq
+		for _, addr := range c {
+			isAddr := net.ParseIP(addr)
+			if isAddr == nil {
+				ip, err := net.LookupIP(addr)
+				if err == nil && len(ip) > 0 {
+					batchreq = append(batchreq, apireq{query: ip[0].String(), fields: "countryCode"})
+				}
+			} else {
+				batchreq = append(batchreq, apireq{query: addr, fields: "countryCode"})
+			}
+		}
+		req, _ := json.Marshal(batchreq)
+		ret, err := http.Post("http://ip-api.com/batch", "application/json", strings.NewReader(string(req)))
+		if err == nil {
+			retBytes, err := io.ReadAll(ret.Body)
+			if err == nil {
+				var retStruct []apiret
+				err := json.Unmarshal(retBytes, &retStruct)
+				if err == nil {
+					for _, dat := range retStruct {
+						if dat.CountryCode != "" {
+							country[dat.CountryCode] = v
+						}
+					}
+				}
+			}
+		}
+	}
+	protocolDict := make(map[string]void)
+	var protocol []string
+	l.svcCtx.DB.Model(&server.Server{}).Where("enable = true").Pluck("protocol", &protocol)
+	for _, p := range protocol {
+		protocolDict[p] = v
+	}
+	protocol = nil
+	for p := range protocolDict {
+		protocol = append(protocol, p)
+	}
+	resp = &types.GetStatResponse{
+		User:     u,
+		Node:     n,
+		Country:  int64(len(country)),
+		Protocol: protocol,
+	}
+	val, _ := json.Marshal(*resp)
+	_ = l.svcCtx.Redis.Set(l.ctx, config.CommonStatCacheKey, string(val), time.Duration(3600)*time.Second).Err()
+	return resp, nil
+}