2
0
mirror of https://github.com/acepanel/panel.git synced 2026-02-06 18:47:31 +08:00

feat: 测试插件

This commit is contained in:
耗子
2024-09-18 02:25:24 +08:00
parent 3193a15131
commit 615744e0cf
11 changed files with 646 additions and 272 deletions

View File

@@ -1,8 +1,6 @@
package openresty
import (
"net/http"
"github.com/go-chi/chi/v5"
"github.com/TheTNB/panel/pkg/pluginloader"
@@ -11,12 +9,23 @@ import (
func init() {
pluginloader.Register(&types.Plugin{
Slug: "openresty",
Name: "OpenResty",
Order: -100,
Slug: "openresty",
Name: "OpenResty",
Description: "OpenResty® 是一款基于 NGINX 和 LuaJIT 的 Web 平台",
Version: "1.25.3.1",
Requires: []string{},
Excludes: []string{},
Install: "bash /www/panel/scripts/openresty/install.sh",
Uninstall: "bash /www/panel/scripts/openresty/uninstall.sh",
Update: "bash /www/panel/scripts/openresty/install.sh",
Route: func(r chi.Router) {
r.Get("/", func(w http.ResponseWriter, r *http.Request) {
_, _ = w.Write([]byte("Hello, World!"))
})
service := NewService()
r.Get("/load", service.Load)
r.Get("/config", service.GetConfig)
r.Post("/config", service.SaveConfig)
r.Get("/errorLog", service.ErrorLog)
r.Post("/clearErrorLog", service.ClearErrorLog)
},
})
}

View File

@@ -1 +1,188 @@
package openresty
import (
"fmt"
"net/http"
"regexp"
"time"
"github.com/go-resty/resty/v2"
"github.com/spf13/cast"
"github.com/TheTNB/panel/internal/service"
"github.com/TheTNB/panel/pkg/io"
"github.com/TheTNB/panel/pkg/shell"
"github.com/TheTNB/panel/pkg/str"
"github.com/TheTNB/panel/pkg/systemctl"
"github.com/TheTNB/panel/pkg/types"
)
type Service struct {
// Dependent services
}
func NewService() *Service {
return &Service{}
}
// GetConfig
//
// @Summary 获取配置
// @Tags 插件-OpenResty
// @Produce json
// @Security BearerToken
// @Success 200 {object} h.SuccessResponse
// @Router /plugins/openresty/config [get]
func (s *Service) GetConfig(w http.ResponseWriter, r *http.Request) {
config, err := io.Read("/www/server/openresty/conf/nginx.conf")
if err != nil {
service.Error(w, http.StatusInternalServerError, "获取配置失败")
return
}
service.Success(w, config)
}
// SaveConfig
//
// @Summary 保存配置
// @Tags 插件-OpenResty
// @Produce json
// @Security BearerToken
// @Param config body string true "配置"
// @Success 200 {object} h.SuccessResponse
// @Router /plugins/openresty/config [post]
func (s *Service) SaveConfig(w http.ResponseWriter, r *http.Request) {
config := r.FormValue("config")
if len(config) == 0 {
service.Error(w, http.StatusInternalServerError, "配置不能为空")
}
if err := io.Write("/www/server/openresty/conf/nginx.conf", config, 0644); err != nil {
service.Error(w, http.StatusInternalServerError, "保存配置失败")
}
if err := systemctl.Reload("openresty"); err != nil {
_, err = shell.Execf("openresty -t")
service.Error(w, http.StatusInternalServerError, fmt.Sprintf("重载服务失败: %v", err))
}
service.Success(w, nil)
}
// ErrorLog
//
// @Summary 获取错误日志
// @Tags 插件-OpenResty
// @Produce json
// @Security BearerToken
// @Success 200 {object} h.SuccessResponse
// @Router /plugins/openresty/errorLog [get]
func (s *Service) ErrorLog(w http.ResponseWriter, r *http.Request) {
if !io.Exists("/www/wwwlogs/nginx_error.log") {
service.Success(w, "")
}
out, err := shell.Execf("tail -n 100 /www/wwwlogs/openresty_error.log")
if err != nil {
service.Error(w, http.StatusInternalServerError, out)
}
service.Success(w, out)
}
// ClearErrorLog
//
// @Summary 清空错误日志
// @Tags 插件-OpenResty
// @Produce json
// @Security BearerToken
// @Success 200 {object} h.SuccessResponse
// @Router /plugins/openresty/clearErrorLog [post]
func (s *Service) ClearErrorLog(w http.ResponseWriter, r *http.Request) {
if out, err := shell.Execf("echo '' > /www/wwwlogs/openresty_error.log"); err != nil {
service.Error(w, http.StatusInternalServerError, out)
}
service.Success(w, nil)
}
// Load
//
// @Summary 获取负载状态
// @Tags 插件-OpenResty
// @Produce json
// @Security BearerToken
// @Success 200 {object} h.SuccessResponse
// @Router /plugins/openresty/load [get]
func (s *Service) Load(w http.ResponseWriter, r *http.Request) {
client := resty.New().SetTimeout(10 * time.Second)
resp, err := client.R().Get("http://127.0.0.1/nginx_status")
if err != nil || !resp.IsSuccess() {
service.Success(w, []types.NV{})
}
raw := resp.String()
var data []types.NV
workers, err := shell.Execf("ps aux | grep nginx | grep 'worker process' | wc -l")
if err != nil {
service.Error(w, http.StatusInternalServerError, "获取负载失败")
}
data = append(data, types.NV{
Name: "工作进程",
Value: workers,
})
out, err := shell.Execf("ps aux | grep nginx | grep 'worker process' | awk '{memsum+=$6};END {print memsum}'")
if err != nil {
service.Error(w, http.StatusInternalServerError, "获取负载失败")
}
mem := str.FormatBytes(cast.ToFloat64(out))
data = append(data, types.NV{
Name: "内存占用",
Value: mem,
})
match := regexp.MustCompile(`Active connections:\s+(\d+)`).FindStringSubmatch(raw)
if len(match) == 2 {
data = append(data, types.NV{
Name: "活跃连接数",
Value: match[1],
})
}
match = regexp.MustCompile(`server accepts handled requests\s+(\d+)\s+(\d+)\s+(\d+)`).FindStringSubmatch(raw)
if len(match) == 4 {
data = append(data, types.NV{
Name: "总连接次数",
Value: match[1],
})
data = append(data, types.NV{
Name: "总握手次数",
Value: match[2],
})
data = append(data, types.NV{
Name: "总请求次数",
Value: match[3],
})
}
match = regexp.MustCompile(`Reading:\s+(\d+)\s+Writing:\s+(\d+)\s+Waiting:\s+(\d+)`).FindStringSubmatch(raw)
if len(match) == 4 {
data = append(data, types.NV{
Name: "请求数",
Value: match[1],
})
data = append(data, types.NV{
Name: "响应数",
Value: match[2],
})
data = append(data, types.NV{
Name: "驻留进程",
Value: match[3],
})
}
service.Success(w, data)
}