重构shell嵌套加速
This commit is contained in:
starry
56
src/main.go
56
src/main.go
@@ -198,9 +198,28 @@ func proxyWithRedirect(c *gin.Context, u string, redirectCount int) {
|
||||
resp.Header.Del("Referrer-Policy")
|
||||
resp.Header.Del("Strict-Transport-Security")
|
||||
|
||||
// 对于需要处理的shell文件，使用chunked传输
|
||||
isShellFile := strings.HasSuffix(strings.ToLower(u), ".sh")
|
||||
if isShellFile {
|
||||
// 智能处理系统 - 自动识别需要加速的内容
|
||||
// 获取真实域名
|
||||
realHost := c.Request.Header.Get("X-Forwarded-Host")
|
||||
if realHost == "" {
|
||||
realHost = c.Request.Host
|
||||
}
|
||||
// 如果域名中没有协议前缀，添加https://
|
||||
if !strings.HasPrefix(realHost, "http://") && !strings.HasPrefix(realHost, "https://") {
|
||||
realHost = "https://" + realHost
|
||||
}
|
||||
|
||||
// 使用智能处理器自动处理所有内容
|
||||
processedBody, processedSize, err := ProcessSmart(resp.Body, resp.Header.Get("Content-Encoding") == "gzip", realHost)
|
||||
if err != nil {
|
||||
// 优雅降级 - 处理失败时直接返回原内容
|
||||
c.String(http.StatusInternalServerError, fmt.Sprintf("内容处理错误: %v", err))
|
||||
return
|
||||
}
|
||||
|
||||
// 智能设置响应头
|
||||
if processedSize > 0 {
|
||||
// 内容被处理过，使用chunked传输以获得最佳性能
|
||||
resp.Header.Del("Content-Length")
|
||||
resp.Header.Set("Transfer-Encoding", "chunked")
|
||||
}
|
||||
@@ -224,32 +243,9 @@ func proxyWithRedirect(c *gin.Context, u string, redirectCount int) {
|
||||
|
||||
c.Status(resp.StatusCode)
|
||||
|
||||
// 处理响应体
|
||||
if isShellFile {
|
||||
// 获取真实域名
|
||||
realHost := c.Request.Header.Get("X-Forwarded-Host")
|
||||
if realHost == "" {
|
||||
realHost = c.Request.Host
|
||||
}
|
||||
// 如果域名中没有协议前缀，添加https://
|
||||
if !strings.HasPrefix(realHost, "http://") && !strings.HasPrefix(realHost, "https://") {
|
||||
realHost = "https://" + realHost
|
||||
}
|
||||
// 使用ProcessGitHubURLs处理.sh文件
|
||||
processedBody, _, err := ProcessGitHubURLs(resp.Body, resp.Header.Get("Content-Encoding") == "gzip", realHost, true)
|
||||
if err != nil {
|
||||
c.String(http.StatusInternalServerError, fmt.Sprintf("处理shell文件时发生错误: %v", err))
|
||||
return
|
||||
}
|
||||
if _, err := io.Copy(c.Writer, processedBody); err != nil {
|
||||
c.String(http.StatusInternalServerError, fmt.Sprintf("写入响应时发生错误: %v", err))
|
||||
return
|
||||
}
|
||||
} else {
|
||||
// 对于非.sh文件，直接复制响应体
|
||||
if _, err := io.Copy(c.Writer, resp.Body); err != nil {
|
||||
return
|
||||
}
|
||||
// 输出处理后的内容
|
||||
if _, err := io.Copy(c.Writer, processedBody); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
@@ -261,5 +257,3 @@ func checkURL(u string) []string {
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
|
||||
549
src/proxysh.go
549
src/proxysh.go
@@ -1,192 +1,435 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"bytes"
|
||||
"compress/gzip"
|
||||
"crypto/md5"
|
||||
"fmt"
|
||||
"io"
|
||||
"regexp"
|
||||
"strings"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"time"
|
||||
"unsafe"
|
||||
)
|
||||
|
||||
var (
|
||||
// gitHubDomains 定义所有支持的GitHub相关域名
|
||||
gitHubDomains = []string{
|
||||
"github.com",
|
||||
"raw.githubusercontent.com",
|
||||
"raw.github.com",
|
||||
"gist.githubusercontent.com",
|
||||
"gist.github.com",
|
||||
"api.github.com",
|
||||
}
|
||||
|
||||
// 自动识别、零配置、高性能、优雅降级
|
||||
|
||||
const (
|
||||
// 性能优化常量
|
||||
SMART_BUFFER_SIZE = 32 * 1024 // 32KB 缓冲区
|
||||
CACHE_DURATION = 20 * time.Minute // 20分钟缓存
|
||||
MAX_CACHE_ITEMS = 500 // 最大缓存条目
|
||||
|
||||
// urlPattern 使用gitHubDomains构建正则表达式
|
||||
urlPattern = regexp.MustCompile(`https?://(?:` + strings.Join(gitHubDomains, "|") + `)[^\s'"]+`)
|
||||
|
||||
// 是否启用脚本嵌套代理的调试日志
|
||||
DebugLog = true
|
||||
// 处理限制
|
||||
MAX_CONTENT_SIZE = 50 * 1024 * 1024 // 50MB文件大小限制
|
||||
PROCESS_TIMEOUT = 5 * time.Second // 5秒处理超时
|
||||
)
|
||||
|
||||
// 打印调试日志的辅助函数
|
||||
func debugPrintf(format string, args ...interface{}) {
|
||||
if DebugLog {
|
||||
fmt.Printf(format, args...)
|
||||
}
|
||||
// SmartProcessor 智能处理器 - 核心引擎
|
||||
type SmartProcessor struct {
|
||||
// 编译优化
|
||||
githubRegex *regexp.Regexp // GitHub URL正则
|
||||
shellPatterns []string // Shell特征模式
|
||||
|
||||
// 高性能缓存
|
||||
cache sync.Map // 线程安全缓存
|
||||
bufferPool sync.Pool // 缓冲池
|
||||
|
||||
// 原子统计
|
||||
totalRequests int64 // 总请求数
|
||||
cacheHits int64 // 缓存命中
|
||||
processed int64 // 实际处理
|
||||
|
||||
// 控制开关
|
||||
enabled int32 // 启用状态
|
||||
}
|
||||
|
||||
// ProcessGitHubURLs 处理数据流中的GitHub URL，将其替换为代理URL。
|
||||
// 此处思路借鉴了 https://github.com/WJQSERVER-STUDIO/ghproxy/blob/main/proxy/nest.go
|
||||
// CacheItem 缓存项 - 内存对齐
|
||||
type CacheItem struct {
|
||||
content string // 处理后内容
|
||||
timestamp int64 // 创建时间
|
||||
hits int32 // 访问次数
|
||||
}
|
||||
|
||||
func ProcessGitHubURLs(input io.ReadCloser, isCompressed bool, host string, isShellFile bool) (io.Reader, int64, error) {
|
||||
debugPrintf("开始处理文件: isCompressed=%v, host=%s, isShellFile=%v\n", isCompressed, host, isShellFile)
|
||||
|
||||
if !isShellFile {
|
||||
debugPrintf("非shell文件，跳过处理\n")
|
||||
return input, 0, nil
|
||||
}
|
||||
// 全局单例
|
||||
var (
|
||||
smartProcessor *SmartProcessor
|
||||
smartOnce sync.Once
|
||||
)
|
||||
|
||||
// 使用更大的缓冲区以提高性能
|
||||
pipeReader, pipeWriter := io.Pipe()
|
||||
var written int64
|
||||
|
||||
go func() {
|
||||
var err error
|
||||
defer func() {
|
||||
if err != nil {
|
||||
debugPrintf("处理过程中发生错误: %v\n", err)
|
||||
_ = pipeWriter.CloseWithError(err)
|
||||
} else {
|
||||
_ = pipeWriter.Close()
|
||||
}
|
||||
}()
|
||||
|
||||
defer input.Close()
|
||||
|
||||
var reader io.Reader = input
|
||||
if isCompressed {
|
||||
debugPrintf("检测到压缩文件，进行解压处理\n")
|
||||
gzipReader, gzipErr := gzip.NewReader(input)
|
||||
if gzipErr != nil {
|
||||
err = gzipErr
|
||||
return
|
||||
}
|
||||
defer gzipReader.Close()
|
||||
reader = gzipReader
|
||||
}
|
||||
|
||||
// 使用更大的缓冲区
|
||||
bufReader := bufio.NewReaderSize(reader, 32*1024) // 32KB buffer
|
||||
var writer io.Writer = pipeWriter
|
||||
|
||||
if isCompressed {
|
||||
gzipWriter := gzip.NewWriter(writer)
|
||||
defer gzipWriter.Close()
|
||||
writer = gzipWriter
|
||||
}
|
||||
|
||||
bufWriter := bufio.NewWriterSize(writer, 32*1024) // 32KB buffer
|
||||
defer bufWriter.Flush()
|
||||
|
||||
written, err = processContent(bufReader, bufWriter, host)
|
||||
if err != nil {
|
||||
debugPrintf("处理内容时发生错误: %v\n", err)
|
||||
return
|
||||
// GetSmartProcessor 获取智能处理器实例
|
||||
func GetSmartProcessor() *SmartProcessor {
|
||||
smartOnce.Do(func() {
|
||||
smartProcessor = &SmartProcessor{
|
||||
githubRegex: regexp.MustCompile(`https?://(?:github\.com|raw\.githubusercontent\.com|raw\.github\.com|gist\.githubusercontent\.com|gist\.github\.com|api\.github\.com)[^\s'"]+`),
|
||||
shellPatterns: []string{
|
||||
"#!/bin/bash", "#!/bin/sh", "#!/usr/bin/env bash",
|
||||
"curl ", "wget ", "git clone", "docker ", "kubectl ",
|
||||
"npm install", "pip install", "go get", "export ",
|
||||
},
|
||||
enabled: 1, // 默认启用
|
||||
}
|
||||
|
||||
debugPrintf("文件处理完成，共处理 %d 字节\n", written)
|
||||
}()
|
||||
|
||||
return pipeReader, written, nil
|
||||
// 初始化缓冲池
|
||||
smartProcessor.bufferPool.New = func() interface{} {
|
||||
return make([]byte, SMART_BUFFER_SIZE)
|
||||
}
|
||||
|
||||
// 启动后台清理
|
||||
go smartProcessor.backgroundCleanup()
|
||||
})
|
||||
return smartProcessor
|
||||
}
|
||||
|
||||
// processContent 优化处理文件内容的函数
|
||||
func processContent(reader *bufio.Reader, writer *bufio.Writer, host string) (int64, error) {
|
||||
var written int64
|
||||
lineNum := 0
|
||||
// ProcessSmart 智能处理函数 - 替换所有其他处理函数
|
||||
func ProcessSmart(input io.ReadCloser, isCompressed bool, host string) (io.Reader, int64, error) {
|
||||
processor := GetSmartProcessor()
|
||||
|
||||
// 快速检查是否启用
|
||||
if atomic.LoadInt32(&processor.enabled) == 0 {
|
||||
return input, 0, nil
|
||||
}
|
||||
|
||||
atomic.AddInt64(&processor.totalRequests, 1)
|
||||
|
||||
// 快速读取内容
|
||||
content, err := processor.readContent(input, isCompressed)
|
||||
if err != nil || len(content) == 0 {
|
||||
return input, 0, nil // 优雅降级
|
||||
}
|
||||
|
||||
contentSize := int64(len(content))
|
||||
|
||||
// 大文件检查 - 自动支持大文件
|
||||
if contentSize > MAX_CONTENT_SIZE {
|
||||
return strings.NewReader(content), contentSize, nil // 直接返回，不处理
|
||||
}
|
||||
|
||||
// 智能内容分析
|
||||
if !processor.needsProcessing(content) {
|
||||
return strings.NewReader(content), contentSize, nil
|
||||
}
|
||||
|
||||
// 缓存检查
|
||||
cacheKey := processor.getCacheKey(content, host)
|
||||
if cached := processor.getFromCache(cacheKey); cached != nil {
|
||||
atomic.AddInt64(&processor.cacheHits, 1)
|
||||
return strings.NewReader(cached.content), int64(len(cached.content)), nil
|
||||
}
|
||||
|
||||
// 智能处理
|
||||
processed := processor.processContent(content, host)
|
||||
atomic.AddInt64(&processor.processed, 1)
|
||||
|
||||
// 存入缓存
|
||||
processor.saveToCache(cacheKey, processed)
|
||||
|
||||
return strings.NewReader(processed), int64(len(processed)), nil
|
||||
}
|
||||
|
||||
|
||||
// needsProcessing 智能判断是否需要处理
|
||||
func (sp *SmartProcessor) needsProcessing(content string) bool {
|
||||
contentStr := string(content)
|
||||
|
||||
// 快速检查：是否包含GitHub URL
|
||||
if !strings.Contains(contentStr, "github.com") &&
|
||||
!strings.Contains(contentStr, "githubusercontent.com") {
|
||||
return false
|
||||
}
|
||||
|
||||
// 智能检查：是否为脚本类内容
|
||||
return sp.isScriptContent(contentStr)
|
||||
}
|
||||
|
||||
// isScriptContent 智能脚本内容识别
|
||||
func (sp *SmartProcessor) isScriptContent(content string) bool {
|
||||
// 空内容检查
|
||||
if len(content) < 10 {
|
||||
return false
|
||||
}
|
||||
|
||||
// 二进制文件快速排除
|
||||
if sp.isBinaryContent(content) {
|
||||
return false
|
||||
}
|
||||
|
||||
// 检查前500字符的特征
|
||||
preview := content
|
||||
if len(content) > 500 {
|
||||
preview = content[:500]
|
||||
}
|
||||
|
||||
// Shell特征检查
|
||||
for _, pattern := range sp.shellPatterns {
|
||||
if strings.Contains(preview, pattern) {
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
// Dockerfile特征
|
||||
if strings.Contains(preview, "FROM ") || strings.Contains(preview, "RUN ") {
|
||||
return true
|
||||
}
|
||||
|
||||
// GitHub Actions特征
|
||||
if strings.Contains(preview, "uses: ") || strings.Contains(preview, "github.com/") {
|
||||
return true
|
||||
}
|
||||
|
||||
// 其他脚本特征
|
||||
scriptIndicators := []string{
|
||||
"#!/", "set -e", "function ", "echo ", "mkdir ", "chmod ",
|
||||
"apt-get", "yum ", "brew ", "sudo ", "source ", ". /",
|
||||
}
|
||||
|
||||
for _, indicator := range scriptIndicators {
|
||||
if strings.Contains(preview, indicator) {
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
// isBinaryContent 二进制内容检测
|
||||
func (sp *SmartProcessor) isBinaryContent(content string) bool {
|
||||
if len(content) < 100 {
|
||||
return false
|
||||
}
|
||||
|
||||
// 检查前100字节中的空字节
|
||||
for i := 0; i < 100 && i < len(content); i++ {
|
||||
if content[i] == 0 {
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
// readContent 高效内容读取
|
||||
func (sp *SmartProcessor) readContent(input io.ReadCloser, isCompressed bool) (string, error) {
|
||||
defer input.Close()
|
||||
|
||||
var reader io.Reader = input
|
||||
|
||||
// 压缩处理
|
||||
if isCompressed {
|
||||
gzReader, err := gzip.NewReader(input)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
defer gzReader.Close()
|
||||
reader = gzReader
|
||||
}
|
||||
|
||||
// 使用缓冲池
|
||||
buffer := sp.bufferPool.Get().([]byte)
|
||||
defer sp.bufferPool.Put(buffer)
|
||||
|
||||
var result strings.Builder
|
||||
result.Grow(SMART_BUFFER_SIZE) // 预分配
|
||||
|
||||
for {
|
||||
lineNum++
|
||||
line, err := reader.ReadString('\n')
|
||||
if err != nil && err != io.EOF {
|
||||
return written, fmt.Errorf("读取行时发生错误: %w", err)
|
||||
n, err := reader.Read(buffer)
|
||||
if n > 0 {
|
||||
result.Write(buffer[:n])
|
||||
}
|
||||
|
||||
if line != "" {
|
||||
// 在处理前先检查是否包含GitHub URL
|
||||
if strings.Contains(line, "github.com") ||
|
||||
strings.Contains(line, "raw.githubusercontent.com") {
|
||||
matches := urlPattern.FindAllString(line, -1)
|
||||
if len(matches) > 0 {
|
||||
debugPrintf("\n在第 %d 行发现 %d 个GitHub URL:\n", lineNum, len(matches))
|
||||
for _, match := range matches {
|
||||
debugPrintf("原始URL: %s\n", match)
|
||||
}
|
||||
}
|
||||
|
||||
modifiedLine := processLine(line, host, lineNum)
|
||||
n, writeErr := writer.WriteString(modifiedLine)
|
||||
if writeErr != nil {
|
||||
return written, fmt.Errorf("写入修改后的行时发生错误: %w", writeErr)
|
||||
}
|
||||
written += int64(n)
|
||||
} else {
|
||||
// 如果行中没有GitHub URL，直接写入
|
||||
n, writeErr := writer.WriteString(line)
|
||||
if writeErr != nil {
|
||||
return written, fmt.Errorf("写入原始行时发生错误: %w", writeErr)
|
||||
}
|
||||
written += int64(n)
|
||||
}
|
||||
}
|
||||
|
||||
if err == io.EOF {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
// 确保所有数据都被写入
|
||||
if err := writer.Flush(); err != nil {
|
||||
return written, fmt.Errorf("刷新缓冲区时发生错误: %w", err)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
// 大文件保护
|
||||
if result.Len() > MAX_CONTENT_SIZE {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
return written, nil
|
||||
return result.String(), nil
|
||||
}
|
||||
|
||||
// processLine 处理单行文本，替换所有匹配的GitHub URL
|
||||
func processLine(line string, host string, lineNum int) string {
|
||||
return urlPattern.ReplaceAllStringFunc(line, func(url string) string {
|
||||
newURL := modifyGitHubURL(url, host)
|
||||
if newURL != url {
|
||||
debugPrintf("第 %d 行URL替换:\n 原始: %s\n 替换后: %s\n", lineNum, url, newURL)
|
||||
// processContent 智能内容处理
|
||||
func (sp *SmartProcessor) processContent(content, host string) string {
|
||||
// 预分配结果缓冲区
|
||||
var result bytes.Buffer
|
||||
result.Grow(len(content) + len(content)/5) // 预留20%空间
|
||||
|
||||
// 高效URL替换
|
||||
matches := sp.githubRegex.FindAllStringIndex(content, -1)
|
||||
if len(matches) == 0 {
|
||||
return content // 无需处理
|
||||
}
|
||||
|
||||
lastIndex := 0
|
||||
for _, match := range matches {
|
||||
// 写入URL前的内容
|
||||
result.WriteString(content[lastIndex:match[0]])
|
||||
|
||||
// 处理URL
|
||||
originalURL := content[match[0]:match[1]]
|
||||
newURL := sp.transformURL(originalURL, host)
|
||||
result.WriteString(newURL)
|
||||
|
||||
lastIndex = match[1]
|
||||
}
|
||||
|
||||
// 写入剩余内容
|
||||
result.WriteString(content[lastIndex:])
|
||||
|
||||
return result.String()
|
||||
}
|
||||
|
||||
// transformURL 智能URL转换
|
||||
func (sp *SmartProcessor) transformURL(url, host string) string {
|
||||
// 避免重复处理
|
||||
if strings.Contains(url, host) {
|
||||
return url
|
||||
}
|
||||
|
||||
// 协议标准化
|
||||
if !strings.HasPrefix(url, "https://") {
|
||||
if strings.HasPrefix(url, "http://") {
|
||||
url = "https://" + url[7:]
|
||||
} else if !strings.HasPrefix(url, "//") {
|
||||
url = "https://" + url
|
||||
}
|
||||
return newURL
|
||||
}
|
||||
|
||||
// Host清理
|
||||
cleanHost := strings.TrimPrefix(host, "https://")
|
||||
cleanHost = strings.TrimPrefix(cleanHost, "http://")
|
||||
cleanHost = strings.TrimSuffix(cleanHost, "/")
|
||||
|
||||
// 构建新URL
|
||||
return cleanHost + "/" + url
|
||||
}
|
||||
|
||||
// getCacheKey 生成缓存键
|
||||
func (sp *SmartProcessor) getCacheKey(content, host string) string {
|
||||
// 使用MD5快速哈希
|
||||
hasher := md5.New()
|
||||
hasher.Write(*(*[]byte)(unsafe.Pointer(&content)))
|
||||
hasher.Write(*(*[]byte)(unsafe.Pointer(&host)))
|
||||
return fmt.Sprintf("%x", hasher.Sum(nil))[:16]
|
||||
}
|
||||
|
||||
// getFromCache 从缓存获取
|
||||
func (sp *SmartProcessor) getFromCache(key string) *CacheItem {
|
||||
if value, ok := sp.cache.Load(key); ok {
|
||||
item := value.(*CacheItem)
|
||||
|
||||
// TTL检查
|
||||
if time.Now().UnixNano()-item.timestamp > int64(CACHE_DURATION) {
|
||||
sp.cache.Delete(key)
|
||||
return nil
|
||||
}
|
||||
|
||||
atomic.AddInt32(&item.hits, 1)
|
||||
return item
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// saveToCache 保存到缓存
|
||||
func (sp *SmartProcessor) saveToCache(key, content string) {
|
||||
item := &CacheItem{
|
||||
content: content,
|
||||
timestamp: time.Now().UnixNano(),
|
||||
hits: 0,
|
||||
}
|
||||
sp.cache.Store(key, item)
|
||||
}
|
||||
|
||||
|
||||
func (sp *SmartProcessor) backgroundCleanup() {
|
||||
ticker := time.NewTicker(10 * time.Minute)
|
||||
defer ticker.Stop()
|
||||
|
||||
for range ticker.C {
|
||||
sp.cleanExpiredCache()
|
||||
sp.logStats()
|
||||
}
|
||||
}
|
||||
|
||||
// cleanExpiredCache 清理过期缓存
|
||||
func (sp *SmartProcessor) cleanExpiredCache() {
|
||||
now := time.Now().UnixNano()
|
||||
count := 0
|
||||
|
||||
sp.cache.Range(func(key, value interface{}) bool {
|
||||
count++
|
||||
item := value.(*CacheItem)
|
||||
|
||||
// 清理过期项
|
||||
if now-item.timestamp > int64(CACHE_DURATION) {
|
||||
sp.cache.Delete(key)
|
||||
return true
|
||||
}
|
||||
|
||||
// 限制缓存大小（LRU）
|
||||
if count > MAX_CACHE_ITEMS && item.hits < 2 {
|
||||
sp.cache.Delete(key)
|
||||
}
|
||||
|
||||
return true
|
||||
})
|
||||
}
|
||||
|
||||
// 判断代理域名前缀
|
||||
func modifyGitHubURL(url string, host string) string {
|
||||
for _, domain := range gitHubDomains {
|
||||
hasHttps := strings.HasPrefix(url, "https://"+domain)
|
||||
hasHttp := strings.HasPrefix(url, "http://"+domain)
|
||||
|
||||
if hasHttps || hasHttp || strings.HasPrefix(url, domain) {
|
||||
if !hasHttps && !hasHttp {
|
||||
url = "https://" + url
|
||||
}
|
||||
if hasHttp {
|
||||
url = "https://" + strings.TrimPrefix(url, "http://")
|
||||
}
|
||||
// 移除host开头的协议头（如果有）
|
||||
host = strings.TrimPrefix(host, "https://")
|
||||
host = strings.TrimPrefix(host, "http://")
|
||||
// 返回组合后的URL
|
||||
return host + "/" + url
|
||||
}
|
||||
// 调试日志开关
|
||||
var smartDebugLog = true
|
||||
|
||||
// logStats 统计日志
|
||||
func (sp *SmartProcessor) logStats() {
|
||||
if !smartDebugLog {
|
||||
return
|
||||
}
|
||||
return url
|
||||
|
||||
total := atomic.LoadInt64(&sp.totalRequests)
|
||||
hits := atomic.LoadInt64(&sp.cacheHits)
|
||||
processed := atomic.LoadInt64(&sp.processed)
|
||||
|
||||
hitRate := float64(0)
|
||||
if total > 0 {
|
||||
hitRate = float64(hits) / float64(total) * 100
|
||||
}
|
||||
|
||||
fmt.Printf("智能处理器统计: 总请求=%d, 缓存命中=%d(%.1f%%), 实际处理=%d\n",
|
||||
total, hits, hitRate, processed)
|
||||
}
|
||||
|
||||
// IsShellFile 检查文件是否为shell文件（基于文件名）
|
||||
func IsShellFile(filename string) bool {
|
||||
return strings.HasSuffix(filename, ".sh")
|
||||
// Enable 启用处理器
|
||||
func (sp *SmartProcessor) Enable() {
|
||||
atomic.StoreInt32(&sp.enabled, 1)
|
||||
}
|
||||
|
||||
// Disable 禁用处理器
|
||||
func (sp *SmartProcessor) Disable() {
|
||||
atomic.StoreInt32(&sp.enabled, 0)
|
||||
}
|
||||
|
||||
// IsEnabled 检查状态
|
||||
func (sp *SmartProcessor) IsEnabled() bool {
|
||||
return atomic.LoadInt32(&sp.enabled) == 1
|
||||
}
|
||||
|
||||
// ClearCache 清空缓存
|
||||
func (sp *SmartProcessor) ClearCache() {
|
||||
sp.cache.Range(func(key, value interface{}) bool {
|
||||
sp.cache.Delete(key)
|
||||
return true
|
||||
})
|
||||
}
|
||||
|
||||
// GetStats 日志
|
||||
func (sp *SmartProcessor) GetStats() (total, hits, processed int64) {
|
||||
return atomic.LoadInt64(&sp.totalRequests),
|
||||
atomic.LoadInt64(&sp.cacheHits),
|
||||
atomic.LoadInt64(&sp.processed)
|
||||
}
|
||||
Reference in New Issue
Block a user