331 lines
8.4 KiB
Go
331 lines
8.4 KiB
Go
package main
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"log"
|
||
"os"
|
||
"path/filepath"
|
||
"time"
|
||
|
||
"github.com/darkit/plugins"
|
||
)
|
||
|
||
// DefaultLoggerPlugin 默认日志插件
|
||
// 提供文件日志和控制台日志功能，使用默认插件类型
|
||
type DefaultLoggerPlugin struct {
|
||
*plugins.BasePlugin // 嵌入基本插件结构
|
||
logFile *os.File // 日志文件
|
||
logger *log.Logger // 日志记录器
|
||
config map[string]interface{} // 配置
|
||
}
|
||
|
||
// Plugin 导出的插件变量
|
||
// 注意：变量名必须是Plugin，大小写敏感
|
||
// 这个插件使用默认的通用插件类型（PluginTypeGeneral）
|
||
var Plugin plugins.IPlugin = &DefaultLoggerPlugin{
|
||
BasePlugin: plugins.NewBasePluginWithDefaultType(
|
||
"DefaultLoggerPlugin",
|
||
"1.0.0",
|
||
"使用默认通用类型的日志记录插件",
|
||
"开发者",
|
||
), // 将自动使用 PluginTypeGeneral 类型
|
||
}
|
||
|
||
// Init 初始化插件
|
||
func (p *DefaultLoggerPlugin) Init(ctx context.Context, config map[string]interface{}) error {
|
||
p.config = config
|
||
|
||
// 获取日志文件路径
|
||
logPath, ok := config["log_path"].(string)
|
||
if !ok {
|
||
// 使用默认路径
|
||
logPath = "logs/default"
|
||
}
|
||
|
||
// 确保日志目录存在
|
||
if err := os.MkdirAll(logPath, 0o755); err != nil {
|
||
return fmt.Errorf("创建日志目录失败: %v", err)
|
||
}
|
||
|
||
// 创建日志文件
|
||
logFilePath := filepath.Join(logPath, fmt.Sprintf("default_app_%s.log", time.Now().Format("2006-01-02")))
|
||
logFile, err := os.OpenFile(logFilePath, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0o644)
|
||
if err != nil {
|
||
return fmt.Errorf("打开日志文件失败: %v", err)
|
||
}
|
||
|
||
p.logFile = logFile
|
||
p.logger = log.New(logFile, "[DefaultLoggerPlugin] ", log.LstdFlags)
|
||
|
||
p.logger.Println("默认日志插件初始化完成")
|
||
fmt.Printf("默认日志插件初始化完成，日志文件: %s，插件类型: %s\n", logFilePath, p.Type())
|
||
|
||
return nil
|
||
}
|
||
|
||
// Start 启动插件
|
||
func (p *DefaultLoggerPlugin) Start(ctx context.Context) error {
|
||
if p.logger == nil {
|
||
return fmt.Errorf("插件未初始化")
|
||
}
|
||
|
||
p.logger.Println("默认日志插件已启动")
|
||
fmt.Println("默认日志插件已启动")
|
||
return nil
|
||
}
|
||
|
||
// Stop 停止插件
|
||
func (p *DefaultLoggerPlugin) Stop(ctx context.Context) error {
|
||
if p.logger != nil {
|
||
p.logger.Println("默认日志插件正在停止")
|
||
}
|
||
|
||
if p.logFile != nil {
|
||
if err := p.logFile.Close(); err != nil {
|
||
return fmt.Errorf("关闭日志文件失败: %v", err)
|
||
}
|
||
}
|
||
|
||
fmt.Println("默认日志插件已停止")
|
||
return nil
|
||
}
|
||
|
||
// Log 记录日志
|
||
func (p *DefaultLoggerPlugin) Log(level, message string) {
|
||
if p.logger == nil {
|
||
fmt.Printf("[DEFAULT][%s] %s\n", level, message)
|
||
return
|
||
}
|
||
|
||
logMsg := fmt.Sprintf("[%s] %s", level, message)
|
||
p.logger.Println(logMsg)
|
||
|
||
// 如果配置了同时输出到控制台
|
||
if consoleOutput, ok := p.config["console_output"].(bool); ok && consoleOutput {
|
||
fmt.Println(logMsg)
|
||
}
|
||
}
|
||
|
||
// Info 记录信息日志
|
||
func (p *DefaultLoggerPlugin) Info(message string) {
|
||
p.Log("INFO", message)
|
||
}
|
||
|
||
// Warn 记录警告日志
|
||
func (p *DefaultLoggerPlugin) Warn(message string) {
|
||
p.Log("WARN", message)
|
||
}
|
||
|
||
// Error 记录错误日志
|
||
func (p *DefaultLoggerPlugin) Error(message string) {
|
||
p.Log("ERROR", message)
|
||
}
|
||
|
||
// GetOperationInfo 获取操作的参数信息
|
||
func (p *DefaultLoggerPlugin) GetOperationInfo(operation string) (*plugins.OperationInfo, error) {
|
||
switch operation {
|
||
case "log":
|
||
return &plugins.OperationInfo{
|
||
Name: "log",
|
||
Description: "记录日志",
|
||
Params: []plugins.OperationParamInfo{
|
||
{
|
||
Name: "level",
|
||
Type: "string",
|
||
Required: true,
|
||
Default: "INFO",
|
||
Description: "日志级别，可选值：INFO, WARN, ERROR",
|
||
},
|
||
{
|
||
Name: "message",
|
||
Type: "string",
|
||
Required: true,
|
||
Default: "",
|
||
Description: "日志消息",
|
||
},
|
||
},
|
||
Extra: map[string]interface{}{
|
||
"category": "日志操作",
|
||
},
|
||
}, nil
|
||
case "info":
|
||
return &plugins.OperationInfo{
|
||
Name: "info",
|
||
Description: "记录信息日志",
|
||
Params: []plugins.OperationParamInfo{
|
||
{
|
||
Name: "message",
|
||
Type: "string",
|
||
Required: true,
|
||
Default: "",
|
||
Description: "日志消息",
|
||
},
|
||
},
|
||
Extra: map[string]interface{}{
|
||
"category": "日志操作",
|
||
},
|
||
}, nil
|
||
case "warn":
|
||
return &plugins.OperationInfo{
|
||
Name: "warn",
|
||
Description: "记录警告日志",
|
||
Params: []plugins.OperationParamInfo{
|
||
{
|
||
Name: "message",
|
||
Type: "string",
|
||
Required: true,
|
||
Default: "",
|
||
Description: "日志消息",
|
||
},
|
||
},
|
||
Extra: map[string]interface{}{
|
||
"category": "日志操作",
|
||
},
|
||
}, nil
|
||
case "error":
|
||
return &plugins.OperationInfo{
|
||
Name: "error",
|
||
Description: "记录错误日志",
|
||
Params: []plugins.OperationParamInfo{
|
||
{
|
||
Name: "message",
|
||
Type: "string",
|
||
Required: true,
|
||
Default: "",
|
||
Description: "日志消息",
|
||
},
|
||
},
|
||
Extra: map[string]interface{}{
|
||
"category": "日志操作",
|
||
},
|
||
}, nil
|
||
case "console":
|
||
return &plugins.OperationInfo{
|
||
Name: "console",
|
||
Description: "向控制台打印日志",
|
||
Params: []plugins.OperationParamInfo{
|
||
{
|
||
Name: "level",
|
||
Type: "string",
|
||
Required: false,
|
||
Default: "INFO",
|
||
Description: "日志级别，可选值：INFO, WARN, ERROR",
|
||
},
|
||
{
|
||
Name: "message",
|
||
Type: "string",
|
||
Required: true,
|
||
Default: "",
|
||
Description: "日志消息",
|
||
},
|
||
},
|
||
Extra: map[string]interface{}{
|
||
"category": "控制台操作",
|
||
},
|
||
}, nil
|
||
default:
|
||
return nil, fmt.Errorf("插件 %s 不支持 %s 操作", p.Name(), operation)
|
||
}
|
||
}
|
||
|
||
// GetAllOperations 获取所有操作及其参数信息
|
||
func (p *DefaultLoggerPlugin) GetAllOperations() []*plugins.OperationInfo {
|
||
operations := []*plugins.OperationInfo{}
|
||
|
||
// 添加所有支持的操作
|
||
ops := []string{"log", "info", "warn", "error", "console"}
|
||
for _, op := range ops {
|
||
info, _ := p.GetOperationInfo(op)
|
||
operations = append(operations, info)
|
||
}
|
||
|
||
return operations
|
||
}
|
||
|
||
// Execute 实现执行插件功能的方法
|
||
func (p *DefaultLoggerPlugin) Execute(ctx context.Context, action string, params map[string]interface{}) (interface{}, error) {
|
||
switch action {
|
||
case "log":
|
||
// 获取参数
|
||
level, ok := params["level"].(string)
|
||
if !ok {
|
||
return nil, fmt.Errorf("缺少日志级别参数或类型错误")
|
||
}
|
||
message, ok := params["message"].(string)
|
||
if !ok {
|
||
return nil, fmt.Errorf("缺少日志消息参数或类型错误")
|
||
}
|
||
|
||
// 记录日志
|
||
p.Log(level, message)
|
||
return map[string]interface{}{"success": true, "message": "日志已记录"}, nil
|
||
|
||
case "info":
|
||
// 获取参数
|
||
message, ok := params["message"].(string)
|
||
if !ok {
|
||
return nil, fmt.Errorf("缺少日志消息参数或类型错误")
|
||
}
|
||
|
||
// 记录信息日志
|
||
p.Info(message)
|
||
return map[string]interface{}{"success": true, "message": "信息日志已记录"}, nil
|
||
|
||
case "warn":
|
||
// 获取参数
|
||
message, ok := params["message"].(string)
|
||
if !ok {
|
||
return nil, fmt.Errorf("缺少日志消息参数或类型错误")
|
||
}
|
||
|
||
// 记录警告日志
|
||
p.Warn(message)
|
||
return map[string]interface{}{"success": true, "message": "警告日志已记录"}, nil
|
||
|
||
case "error":
|
||
// 获取参数
|
||
message, ok := params["message"].(string)
|
||
if !ok {
|
||
return nil, fmt.Errorf("缺少日志消息参数或类型错误")
|
||
}
|
||
|
||
// 记录错误日志
|
||
p.Error(message)
|
||
return map[string]interface{}{"success": true, "message": "错误日志已记录"}, nil
|
||
|
||
case "console":
|
||
// 获取参数
|
||
level := "INFO" // 默认级别
|
||
if levelParam, ok := params["level"].(string); ok && levelParam != "" {
|
||
level = levelParam
|
||
}
|
||
|
||
message, ok := params["message"].(string)
|
||
if !ok {
|
||
return nil, fmt.Errorf("缺少日志消息参数或类型错误")
|
||
}
|
||
|
||
// 强制输出到控制台，无论配置如何
|
||
formattedMsg := fmt.Sprintf("[控制台][%s] %s", level, message)
|
||
fmt.Println(formattedMsg)
|
||
|
||
// 同时也记录到日志文件
|
||
p.Log(level, fmt.Sprintf("[Web控制台] %s", message))
|
||
|
||
return map[string]interface{}{
|
||
"success": true,
|
||
"message": "日志已输出到控制台",
|
||
"console_output": formattedMsg,
|
||
}, nil
|
||
|
||
default:
|
||
return nil, fmt.Errorf("不支持的操作: %s", action)
|
||
}
|
||
}
|
||
|
||
// main 函数是必须的，但不会被调用
|
||
func main() {
|
||
// 不会被执行，仅用于编译插件
|
||
}
|