123 lines
2.7 KiB
Go
123 lines
2.7 KiB
Go
package main
|
||
|
||
import (
|
||
"context"
|
||
"go-desk/internal/database"
|
||
"go-desk/internal/filesystem"
|
||
"go-desk/internal/system"
|
||
"os"
|
||
)
|
||
|
||
// App 应用结构体
|
||
type App struct {
|
||
ctx context.Context
|
||
db *database.DB
|
||
}
|
||
|
||
// NewApp 创建新的应用实例
|
||
func NewApp() *App {
|
||
return &App{}
|
||
}
|
||
|
||
// startup 应用启动时调用
|
||
func (a *App) startup(ctx context.Context) {
|
||
a.ctx = ctx
|
||
|
||
// 初始化数据库连接
|
||
db, err := database.Init()
|
||
if err != nil {
|
||
println("数据库连接失败:", err.Error())
|
||
return
|
||
}
|
||
a.db = db
|
||
}
|
||
|
||
// QueryUsers 查询用户列表
|
||
func (a *App) QueryUsers(keyword string, status int, role int, organid int, page int, pageSize int, sortField string, sortOrder string) (map[string]interface{}, error) {
|
||
if a.db == nil {
|
||
return map[string]interface{}{
|
||
"rows": []interface{}{},
|
||
"total": 0,
|
||
}, nil
|
||
}
|
||
|
||
return a.db.QueryUsers(keyword, status, role, organid, page, pageSize, sortField, sortOrder)
|
||
}
|
||
|
||
// Greet 测试方法
|
||
func (a *App) Greet(name string) string {
|
||
return "Hello " + name + ", It's show time!"
|
||
}
|
||
|
||
// GetSystemInfo 获取系统信息
|
||
func (a *App) GetSystemInfo() (map[string]interface{}, error) {
|
||
return system.GetSystemInfo()
|
||
}
|
||
|
||
// GetCPUInfo 获取 CPU 信息
|
||
func (a *App) GetCPUInfo() (map[string]interface{}, error) {
|
||
return system.GetCPUInfo()
|
||
}
|
||
|
||
// GetMemoryInfo 获取内存信息
|
||
func (a *App) GetMemoryInfo() (map[string]interface{}, error) {
|
||
return system.GetMemoryInfo()
|
||
}
|
||
|
||
// GetDiskInfo 获取磁盘信息
|
||
func (a *App) GetDiskInfo() ([]map[string]interface{}, error) {
|
||
return system.GetDiskInfo()
|
||
}
|
||
|
||
// ReadFile 读取文件
|
||
func (a *App) ReadFile(path string) (string, error) {
|
||
return filesystem.ReadFile(path)
|
||
}
|
||
|
||
// WriteFile 写入文件
|
||
func (a *App) WriteFile(path, content string) error {
|
||
return filesystem.WriteFile(path, content)
|
||
}
|
||
|
||
// ListDir 列出目录
|
||
func (a *App) ListDir(path string) ([]map[string]interface{}, error) {
|
||
return filesystem.ListDir(path)
|
||
}
|
||
|
||
// CreateDir 创建目录
|
||
func (a *App) CreateDir(path string) error {
|
||
return filesystem.CreateDir(path)
|
||
}
|
||
|
||
// DeletePath 删除文件或目录
|
||
func (a *App) DeletePath(path string) error {
|
||
return filesystem.DeletePath(path)
|
||
}
|
||
|
||
// GetFileInfo 获取文件信息
|
||
func (a *App) GetFileInfo(path string) (map[string]interface{}, error) {
|
||
return filesystem.GetFileInfo(path)
|
||
}
|
||
|
||
// GetEnvVars 获取环境变量
|
||
func (a *App) GetEnvVars() (map[string]string, error) {
|
||
envVars := make(map[string]string)
|
||
for _, env := range os.Environ() {
|
||
parts := splitEnv(env)
|
||
if len(parts) == 2 {
|
||
envVars[parts[0]] = parts[1]
|
||
}
|
||
}
|
||
return envVars, nil
|
||
}
|
||
|
||
// splitEnv 分割环境变量字符串（key=value）
|
||
func splitEnv(env string) []string {
|
||
for i := 0; i < len(env); i++ {
|
||
if env[i] == '=' {
|
||
return []string{env[:i], env[i+1:]}
|
||
}
|
||
}
|
||
return []string{env}
|
||
}
|