.
This commit is contained in:
138
internal/database/db.go
Normal file
138
internal/database/db.go
Normal file
@@ -0,0 +1,138 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"go-desk/internal/model"
|
||||
"time"
|
||||
|
||||
mysqldriver "github.com/go-sql-driver/mysql"
|
||||
"gorm.io/driver/mysql"
|
||||
"gorm.io/gorm"
|
||||
"gorm.io/gorm/logger"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrNotConnected = errors.New("数据库未连接")
|
||||
)
|
||||
|
||||
// DB 数据库连接封装
|
||||
type DB struct {
|
||||
db *gorm.DB
|
||||
}
|
||||
|
||||
var globalDB *DB
|
||||
|
||||
// Init 初始化数据库连接
|
||||
func Init() (*DB, error) {
|
||||
if globalDB != nil {
|
||||
return globalDB, nil
|
||||
}
|
||||
|
||||
// 数据库配置 - 测试服 lab_dev
|
||||
// 测试机外网IP: 39.99.243.191
|
||||
// 使用 mysqldriver.Config 结构体构建 DSN,自动处理密码中的特殊字符
|
||||
config := mysqldriver.Config{
|
||||
User: "root",
|
||||
Passwd: "123456",
|
||||
Net: "tcp",
|
||||
Addr: "127.0.0.1:3306",
|
||||
DBName: "lab_dev",
|
||||
Params: map[string]string{"charset": "utf8mb4", "parseTime": "True", "loc": "Local"},
|
||||
AllowNativePasswords: true,
|
||||
}
|
||||
dsn := config.FormatDSN()
|
||||
|
||||
// GORM 配置
|
||||
gormConfig := &gorm.Config{
|
||||
Logger: logger.Default.LogMode(logger.Info),
|
||||
}
|
||||
|
||||
db, err := gorm.Open(mysql.Open(dsn), gormConfig)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("打开数据库连接失败: %v", err)
|
||||
}
|
||||
|
||||
// 获取底层 sql.DB 设置连接池参数
|
||||
sqlDB, err := db.DB()
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("获取数据库实例失败: %v", err)
|
||||
}
|
||||
|
||||
// 测试连接
|
||||
if err := sqlDB.Ping(); err != nil {
|
||||
return nil, fmt.Errorf("数据库连接测试失败: %v", err)
|
||||
}
|
||||
|
||||
// 设置连接池参数
|
||||
sqlDB.SetMaxOpenConns(25)
|
||||
sqlDB.SetMaxIdleConns(5)
|
||||
sqlDB.SetConnMaxLifetime(time.Duration(300) * time.Second)
|
||||
|
||||
globalDB = &DB{db: db}
|
||||
return globalDB, nil
|
||||
}
|
||||
|
||||
// QueryUsers 查询用户列表
|
||||
func (d *DB) QueryUsers(keyword string, status int, role int, organid int, page int, pageSize int, sortField string, sortOrder string) (map[string]interface{}, error) {
|
||||
if d.db == nil {
|
||||
return nil, ErrNotConnected
|
||||
}
|
||||
|
||||
query := d.db.Model(&model.MemberInfo{})
|
||||
|
||||
// 关键字搜索(姓名、账号、电话)
|
||||
if keyword != "" {
|
||||
query = query.Where("membername LIKE ? OR account LIKE ? OR contactphone LIKE ?",
|
||||
"%"+keyword+"%", "%"+keyword+"%", "%"+keyword+"%")
|
||||
}
|
||||
|
||||
// 状态筛选
|
||||
if status > 0 {
|
||||
query = query.Where("status = ?", status)
|
||||
} else {
|
||||
// 默认过滤删除状态
|
||||
query = query.Where("status != ?", 3)
|
||||
}
|
||||
|
||||
// 角色筛选(需要关联查询,暂时简化)
|
||||
if role > 0 {
|
||||
// TODO: 关联 sys_member_role 表查询
|
||||
}
|
||||
|
||||
// 机构筛选
|
||||
if organid > 0 {
|
||||
query = query.Where("organid = ?", organid)
|
||||
}
|
||||
|
||||
// 排序
|
||||
if sortField != "" {
|
||||
if sortOrder == "descend" || sortOrder == "desc" {
|
||||
query = query.Order(sortField + " DESC")
|
||||
} else {
|
||||
query = query.Order(sortField + " ASC")
|
||||
}
|
||||
} else {
|
||||
// 默认按创建时间倒序
|
||||
query = query.Order("createtime DESC")
|
||||
}
|
||||
|
||||
// 总数
|
||||
var total int64
|
||||
query.Count(&total)
|
||||
|
||||
// 分页
|
||||
offset := (page - 1) * pageSize
|
||||
var users []model.MemberInfo
|
||||
if err := query.Offset(offset).Limit(pageSize).Find(&users).Error; err != nil {
|
||||
return nil, fmt.Errorf("查询用户失败: %v", err)
|
||||
}
|
||||
|
||||
// 返回结果
|
||||
result := map[string]interface{}{
|
||||
"rows": users,
|
||||
"total": total,
|
||||
}
|
||||
|
||||
return result, nil
|
||||
}
|
||||
190
internal/filesystem/fs.go
Normal file
190
internal/filesystem/fs.go
Normal file
@@ -0,0 +1,190 @@
|
||||
package filesystem
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// ReadFile 读取文件内容
|
||||
func ReadFile(path string) (string, error) {
|
||||
if !isSafePath(path) {
|
||||
return "", fmt.Errorf("路径不安全")
|
||||
}
|
||||
|
||||
data, err := os.ReadFile(path)
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("读取文件失败: %v", err)
|
||||
}
|
||||
|
||||
return string(data), nil
|
||||
}
|
||||
|
||||
// WriteFile 写入文件
|
||||
func WriteFile(path, content string) error {
|
||||
if !isSafePath(path) {
|
||||
return fmt.Errorf("路径不安全")
|
||||
}
|
||||
|
||||
dir := filepath.Dir(path)
|
||||
if err := os.MkdirAll(dir, 0755); err != nil {
|
||||
return fmt.Errorf("创建目录失败: %v", err)
|
||||
}
|
||||
|
||||
if err := os.WriteFile(path, []byte(content), 0644); err != nil {
|
||||
return fmt.Errorf("写入文件失败: %v", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// ListDir 列出目录内容
|
||||
func ListDir(path string) ([]map[string]interface{}, error) {
|
||||
if !isSafePath(path) {
|
||||
return nil, fmt.Errorf("路径不安全")
|
||||
}
|
||||
|
||||
entries, err := os.ReadDir(path)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("读取目录失败: %v", err)
|
||||
}
|
||||
|
||||
var result []map[string]interface{}
|
||||
for _, entry := range entries {
|
||||
info, err := entry.Info()
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
|
||||
fullPath := filepath.Join(path, entry.Name())
|
||||
result = append(result, map[string]interface{}{
|
||||
"name": entry.Name(),
|
||||
"path": fullPath,
|
||||
"is_dir": entry.IsDir(),
|
||||
"size": info.Size(),
|
||||
"mod_time": info.ModTime().Format("2006-01-02 15:04:05"),
|
||||
})
|
||||
}
|
||||
|
||||
return result, nil
|
||||
}
|
||||
|
||||
// CreateDir 创建目录
|
||||
func CreateDir(path string) error {
|
||||
if !isSafePath(path) {
|
||||
return fmt.Errorf("路径不安全")
|
||||
}
|
||||
|
||||
if err := os.MkdirAll(path, 0755); err != nil {
|
||||
return fmt.Errorf("创建目录失败: %v", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// DeletePath 删除文件或目录
|
||||
func DeletePath(path string) error {
|
||||
if !isSafePath(path) {
|
||||
return fmt.Errorf("路径不安全")
|
||||
}
|
||||
|
||||
info, err := os.Stat(path)
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
return fmt.Errorf("文件或目录不存在")
|
||||
}
|
||||
return fmt.Errorf("获取文件信息失败: %v", err)
|
||||
}
|
||||
|
||||
if info.IsDir() {
|
||||
if err := os.RemoveAll(path); err != nil {
|
||||
return fmt.Errorf("删除目录失败: %v", err)
|
||||
}
|
||||
} else {
|
||||
if err := os.Remove(path); err != nil {
|
||||
return fmt.Errorf("删除文件失败: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetFileInfo 获取文件信息
|
||||
func GetFileInfo(path string) (map[string]interface{}, error) {
|
||||
if !isSafePath(path) {
|
||||
return nil, fmt.Errorf("路径不安全")
|
||||
}
|
||||
|
||||
info, err := os.Stat(path)
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
return nil, fmt.Errorf("文件或目录不存在")
|
||||
}
|
||||
return nil, fmt.Errorf("获取文件信息失败: %v", err)
|
||||
}
|
||||
|
||||
formatBytes := func(bytes int64) string {
|
||||
const unit = 1024
|
||||
if bytes < unit {
|
||||
return fmt.Sprintf("%d B", bytes)
|
||||
}
|
||||
div, exp := int64(unit), 0
|
||||
for n := bytes / unit; n >= unit; n /= unit {
|
||||
div *= unit
|
||||
exp++
|
||||
}
|
||||
return fmt.Sprintf("%.2f %cB", float64(bytes)/float64(div), "KMGTPE"[exp])
|
||||
}
|
||||
|
||||
return map[string]interface{}{
|
||||
"name": info.Name(),
|
||||
"path": path,
|
||||
"size": info.Size(),
|
||||
"size_str": formatBytes(info.Size()),
|
||||
"is_dir": info.IsDir(),
|
||||
"mod_time": info.ModTime().Format("2006-01-02 15:04:05"),
|
||||
"mode": info.Mode().String(),
|
||||
}, nil
|
||||
}
|
||||
|
||||
// OpenPath 打开文件或目录(使用系统默认程序)
|
||||
func OpenPath(path string) error {
|
||||
if !isSafePath(path) {
|
||||
return fmt.Errorf("路径不安全")
|
||||
}
|
||||
|
||||
// 注意:这里需要导入 os/exec,但为了安全,暂时不实现执行命令
|
||||
// 可以考虑使用 Wails 的 runtime 包提供的功能
|
||||
return fmt.Errorf("打开功能暂未实现,请手动打开: %s", path)
|
||||
}
|
||||
|
||||
// isSafePath 检查路径是否安全(防止路径遍历攻击)
|
||||
func isSafePath(path string) bool {
|
||||
// 清理路径
|
||||
cleanPath := filepath.Clean(path)
|
||||
|
||||
// 检查是否包含路径遍历
|
||||
if strings.Contains(cleanPath, "..") {
|
||||
return false
|
||||
}
|
||||
|
||||
// Windows 下检查是否尝试访问系统关键目录
|
||||
if runtime.GOOS == "windows" {
|
||||
lowerPath := strings.ToLower(cleanPath)
|
||||
// 禁止访问系统关键目录(可根据需要调整)
|
||||
forbidden := []string{
|
||||
"c:\\windows",
|
||||
"c:\\program files",
|
||||
"c:\\programdata",
|
||||
}
|
||||
for _, fb := range forbidden {
|
||||
if strings.HasPrefix(lowerPath, fb) {
|
||||
return false
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
26
internal/model/member_info.go
Normal file
26
internal/model/member_info.go
Normal file
@@ -0,0 +1,26 @@
|
||||
package model
|
||||
|
||||
// MemberInfo 用户信息表
|
||||
type MemberInfo struct {
|
||||
Memberid int `gorm:"primaryKey;column:memberid;type:int;comment:用户ID" json:"memberid"`
|
||||
Membername string `gorm:"column:membername;type:varchar(100);comment:姓名" json:"membername"`
|
||||
Account string `gorm:"column:account;type:varchar(100);comment:账号" json:"account"`
|
||||
Password string `gorm:"column:password;type:varchar(100);comment:密码" json:"-"`
|
||||
Contactphone string `gorm:"column:contactphone;type:varchar(50);comment:联系电话" json:"contactphone"`
|
||||
Organid int `gorm:"column:organid;type:int;comment:所属机构ID" json:"organid"`
|
||||
Createtime string `gorm:"column:createtime;type:varchar(50);comment:创建时间" json:"createtime"`
|
||||
Updatetime string `gorm:"column:updatetime;type:varchar(50);comment:修改时间" json:"updatetime"`
|
||||
Role int16 `gorm:"column:role;type:smallint;comment:角色类别" json:"role"`
|
||||
Status int16 `gorm:"column:status;type:smallint;comment:状态 1正常 2停用 3删除" json:"status"`
|
||||
Calluserid string `gorm:"column:calluserid;type:varchar(100);comment:坐席用户ID" json:"calluserid"`
|
||||
Remainingexport int `gorm:"column:remainingexport;type:int;comment:本月剩余导出次数" json:"remainingexport"`
|
||||
|
||||
// 虚拟字段(关联查询)
|
||||
Organname string `gorm:"-" json:"organname"` // 机构名称
|
||||
Rolename string `gorm:"-" json:"rolename"` // 角色名称
|
||||
}
|
||||
|
||||
// TableName 指定表名
|
||||
func (MemberInfo) TableName() string {
|
||||
return "member_info"
|
||||
}
|
||||
145
internal/system/system.go
Normal file
145
internal/system/system.go
Normal file
@@ -0,0 +1,145 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"runtime"
|
||||
|
||||
"github.com/shirou/gopsutil/v3/cpu"
|
||||
"github.com/shirou/gopsutil/v3/disk"
|
||||
"github.com/shirou/gopsutil/v3/host"
|
||||
"github.com/shirou/gopsutil/v3/mem"
|
||||
)
|
||||
|
||||
// GetSystemInfo 获取系统基本信息
|
||||
func GetSystemInfo() (map[string]interface{}, error) {
|
||||
hostInfo, err := host.Info()
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("获取主机信息失败: %v", err)
|
||||
}
|
||||
|
||||
return map[string]interface{}{
|
||||
"os": runtime.GOOS,
|
||||
"arch": runtime.GOARCH,
|
||||
"hostname": hostInfo.Hostname,
|
||||
"platform": hostInfo.Platform,
|
||||
"platform_ver": hostInfo.PlatformVersion,
|
||||
"kernel_ver": hostInfo.KernelVersion,
|
||||
"uptime": hostInfo.Uptime,
|
||||
"boot_time": hostInfo.BootTime,
|
||||
"cpu_count": runtime.NumCPU(),
|
||||
}, nil
|
||||
}
|
||||
|
||||
// GetCPUInfo 获取 CPU 信息
|
||||
func GetCPUInfo() (map[string]interface{}, error) {
|
||||
cpuCount, err := cpu.Counts(true)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("获取 CPU 核心数失败: %v", err)
|
||||
}
|
||||
|
||||
cpuPercent, err := cpu.Percent(0, false)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("获取 CPU 使用率失败: %v", err)
|
||||
}
|
||||
|
||||
cpuInfo, err := cpu.Info()
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("获取 CPU 详细信息失败: %v", err)
|
||||
}
|
||||
|
||||
cpuModel := "Unknown"
|
||||
if len(cpuInfo) > 0 {
|
||||
cpuModel = cpuInfo[0].ModelName
|
||||
}
|
||||
|
||||
usage := 0.0
|
||||
if len(cpuPercent) > 0 {
|
||||
usage = cpuPercent[0]
|
||||
}
|
||||
|
||||
return map[string]interface{}{
|
||||
"cores": cpuCount,
|
||||
"model": cpuModel,
|
||||
"usage": fmt.Sprintf("%.2f%%", usage),
|
||||
"usage_raw": usage,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// GetMemoryInfo 获取内存信息
|
||||
func GetMemoryInfo() (map[string]interface{}, error) {
|
||||
memInfo, err := mem.VirtualMemory()
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("获取内存信息失败: %v", err)
|
||||
}
|
||||
|
||||
formatBytes := func(bytes uint64) string {
|
||||
const unit = 1024
|
||||
if bytes < unit {
|
||||
return fmt.Sprintf("%d B", bytes)
|
||||
}
|
||||
div, exp := int64(unit), 0
|
||||
for n := bytes / unit; n >= unit; n /= unit {
|
||||
div *= unit
|
||||
exp++
|
||||
}
|
||||
return fmt.Sprintf("%.2f %cB", float64(bytes)/float64(div), "KMGTPE"[exp])
|
||||
}
|
||||
|
||||
return map[string]interface{}{
|
||||
"total": memInfo.Total,
|
||||
"used": memInfo.Used,
|
||||
"free": memInfo.Free,
|
||||
"available": memInfo.Available,
|
||||
"usage": fmt.Sprintf("%.2f%%", memInfo.UsedPercent),
|
||||
"usage_raw": memInfo.UsedPercent,
|
||||
"total_str": formatBytes(memInfo.Total),
|
||||
"used_str": formatBytes(memInfo.Used),
|
||||
"free_str": formatBytes(memInfo.Free),
|
||||
"available_str": formatBytes(memInfo.Available),
|
||||
}, nil
|
||||
}
|
||||
|
||||
// GetDiskInfo 获取磁盘信息
|
||||
func GetDiskInfo() ([]map[string]interface{}, error) {
|
||||
partitions, err := disk.Partitions(false)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("获取磁盘分区失败: %v", err)
|
||||
}
|
||||
|
||||
var result []map[string]interface{}
|
||||
formatBytes := func(bytes uint64) string {
|
||||
const unit = 1024
|
||||
if bytes < unit {
|
||||
return fmt.Sprintf("%d B", bytes)
|
||||
}
|
||||
div, exp := int64(unit), 0
|
||||
for n := bytes / unit; n >= unit; n /= unit {
|
||||
div *= unit
|
||||
exp++
|
||||
}
|
||||
return fmt.Sprintf("%.2f %cB", float64(bytes)/float64(div), "KMGTPE"[exp])
|
||||
}
|
||||
|
||||
for _, partition := range partitions {
|
||||
usage, err := disk.Usage(partition.Mountpoint)
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
|
||||
result = append(result, map[string]interface{}{
|
||||
"device": partition.Device,
|
||||
"mountpoint": partition.Mountpoint,
|
||||
"fstype": partition.Fstype,
|
||||
"total": usage.Total,
|
||||
"used": usage.Used,
|
||||
"free": usage.Free,
|
||||
"usage": fmt.Sprintf("%.2f%%", usage.UsedPercent),
|
||||
"usage_raw": usage.UsedPercent,
|
||||
"total_str": formatBytes(usage.Total),
|
||||
"used_str": formatBytes(usage.Used),
|
||||
"free_str": formatBytes(usage.Free),
|
||||
})
|
||||
}
|
||||
|
||||
return result, nil
|
||||
}
|
||||
Reference in New Issue
Block a user