mirror of
https://github.com/hamster1963/HomeDash-Backend.git
synced 2025-06-22 06:51:32 +08:00
update: 初始化代码提交
This commit is contained in:
170
internal/boot/boot.go
Normal file
170
internal/boot/boot.go
Normal file
@ -0,0 +1,170 @@
|
||||
package boot
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/os/gcron"
|
||||
"github.com/gogf/gf/v2/os/gctx"
|
||||
"github.com/gogf/gf/v2/os/glog"
|
||||
"home-network-watcher/utility/ha_utils"
|
||||
"home-network-watcher/utility/network_utils"
|
||||
"home-network-watcher/utility/server_utils"
|
||||
"home-network-watcher/utility/uptime_utils"
|
||||
)
|
||||
|
||||
// Boot
|
||||
//
|
||||
// @dc: 定时任务启动
|
||||
// @return: error
|
||||
func Boot() (err error) {
|
||||
_, err = gcron.AddOnce(context.TODO(), "@every 1s", func(ctx context.Context) {
|
||||
glog.Debug(context.Background(), "定时任务启动中...")
|
||||
if err := bootMethod(); err != nil {
|
||||
glog.Fatal(context.Background(), "定时任务启动失败: ", err)
|
||||
}
|
||||
glog.Debug(context.Background(), "定时任务启动成功")
|
||||
}, "开始启动定时任务")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_, err = gcron.AddOnce(context.TODO(), "@every 15s", func(ctx context.Context) {
|
||||
glog.Info(context.Background(), "定时任务测试中...")
|
||||
if err := bootCheck(); err != nil {
|
||||
glog.Fatal(context.Background(), "定时任务测试失败: ", err)
|
||||
} else {
|
||||
glog.Info(context.Background(), "定时任务测试成功")
|
||||
}
|
||||
}, "开始测试定时任务")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// bootCheck
|
||||
//
|
||||
// @Description: 定时任务测试
|
||||
// @return error
|
||||
func bootCheck() (err error) {
|
||||
return nil
|
||||
}
|
||||
|
||||
// bootMethod
|
||||
// @Description: 定时任务启动
|
||||
// @return error
|
||||
func bootMethod() (err error) {
|
||||
var ctx = gctx.New()
|
||||
|
||||
glog.Notice(ctx, "开始获取科学上网网速")
|
||||
_, err = gcron.AddSingleton(ctx, "@every 1s", func(ctx context.Context) {
|
||||
err = network_utils.ProxyNetwork.GetProxyNetwork()
|
||||
if err != nil {
|
||||
glog.Warning(ctx, "获取代理速度"+err.Error())
|
||||
}
|
||||
}, "获取代理速度")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
glog.Notice(ctx, "开始获取家庭路由器网速")
|
||||
_, err = gcron.AddSingleton(ctx, "@every 1s", func(ctx context.Context) {
|
||||
err = network_utils.NetworkUtils.GetHomeNetwork()
|
||||
if err != nil {
|
||||
glog.Warning(ctx, "获取家庭路由器速度"+err.Error())
|
||||
}
|
||||
}, "获取家庭路由器速度")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
glog.Notice(ctx, "开始获取当前代理节点信息")
|
||||
_, err = gcron.AddSingleton(ctx, "@every 5s", func(ctx context.Context) {
|
||||
err = network_utils.NodeUtils.GetNodeInfo()
|
||||
if err != nil {
|
||||
glog.Warning(ctx, "获取当前代理节点信息"+err.Error())
|
||||
}
|
||||
}, "获取当前代理节点信息")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 进行第一次机场信息缓存
|
||||
err = network_utils.ProxyProvider.GetSubscribeInfo()
|
||||
if err != nil {
|
||||
glog.Warning(ctx, "获取机场订阅信息失败"+err.Error())
|
||||
}
|
||||
|
||||
glog.Notice(ctx, "开始获取机场订阅信息")
|
||||
_, err = gcron.AddSingleton(ctx, "@every 10m", func(ctx context.Context) {
|
||||
err = network_utils.ProxyProvider.GetSubscribeInfo()
|
||||
if err != nil {
|
||||
glog.Warning(ctx, "获取机场订阅信息失败"+err.Error())
|
||||
}
|
||||
}, "开始获取机场订阅信息")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 获取服务器信息
|
||||
glog.Notice(ctx, "开始获取服务器信息")
|
||||
_, err = gcron.AddSingleton(ctx, "@every 1s", func(ctx context.Context) {
|
||||
err = server_utils.Nezha.GetAllServerInfo(ctx)
|
||||
if err != nil {
|
||||
glog.Warning(ctx, "获取服务器信息失败"+err.Error())
|
||||
}
|
||||
}, "开始获取服务器信息")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 获取 xui 信息
|
||||
glog.Notice(ctx, "开始获取 xui 信息")
|
||||
_, err = gcron.AddSingleton(ctx, "@every 5s", func(ctx context.Context) {
|
||||
err = network_utils.ProxyNetwork.GetXuiUserList()
|
||||
if err != nil {
|
||||
glog.Warning(ctx, "获取 xui 信息失败"+err.Error())
|
||||
}
|
||||
}, "开始获取 xui 信息")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 获取服务概览信息
|
||||
glog.Notice(ctx, "开始获取服务概览信息")
|
||||
_, err = gcron.AddSingleton(ctx, "@every 10s", func(ctx context.Context) {
|
||||
err = server_utils.ServerCron.CronGetDockerAndMonitor(ctx)
|
||||
if err != nil {
|
||||
glog.Warning(ctx, "获取服务概览信息失败"+err.Error())
|
||||
}
|
||||
}, "开始获取服务概览信息")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 获取 UptimeKuma 信息
|
||||
glog.Notice(ctx, "开始获取 UptimeKuma 信息")
|
||||
_, err = gcron.AddSingleton(ctx, "@every 10s", func(ctx context.Context) {
|
||||
err = uptime_utils.Kuma.GetUptimeData(ctx)
|
||||
if err != nil {
|
||||
glog.Warning(ctx, "获取 UptimeKuma 信息失败"+err.Error())
|
||||
}
|
||||
}, "开始获取 UptimeKuma 信息")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 获取智能家居信息
|
||||
glog.Notice(ctx, "开始获取智能家居信息")
|
||||
_, err = gcron.AddSingleton(ctx, "@every 1s", func(ctx context.Context) {
|
||||
err = ha_utils.HaUtils.GetEntitiesInfo(ctx)
|
||||
if err != nil {
|
||||
glog.Warning(ctx, "获取智能家居信息失败"+err.Error())
|
||||
}
|
||||
}, "开始获取智能家居信息")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
65
internal/cmd/cmd.go
Normal file
65
internal/cmd/cmd.go
Normal file
@ -0,0 +1,65 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/net/ghttp"
|
||||
"github.com/gogf/gf/v2/os/gcmd"
|
||||
"github.com/gogf/gf/v2/os/glog"
|
||||
"github.com/gogf/gf/v2/os/gres"
|
||||
"home-network-watcher/internal/boot"
|
||||
"home-network-watcher/internal/global/g_functions"
|
||||
"home-network-watcher/internal/global/g_index"
|
||||
"home-network-watcher/internal/global/g_middleware"
|
||||
"home-network-watcher/internal/router/r_hamster_router"
|
||||
binInfo "home-network-watcher/utility/bin_utils"
|
||||
"runtime"
|
||||
)
|
||||
|
||||
var (
|
||||
Main = gcmd.Command{
|
||||
Name: "main",
|
||||
Usage: "main",
|
||||
Brief: "start http server",
|
||||
Func: func(ctx context.Context, parser *gcmd.Parser) (err error) {
|
||||
gres.Dump()
|
||||
s := g.Server()
|
||||
|
||||
// 性能分析
|
||||
runtime.SetMutexProfileFraction(1) // (非必需)开启对锁调用的跟踪
|
||||
runtime.SetBlockProfileRate(1) // (非必需)开启对阻塞操作的跟踪
|
||||
s.EnablePProf()
|
||||
|
||||
// 统一日志服务
|
||||
g_functions.SetDefaultHandler()
|
||||
// 服务状态码处理
|
||||
g_middleware.SMiddlewares.ErrorsStatus(s)
|
||||
|
||||
// 全局中间件
|
||||
s.BindMiddlewareDefault(
|
||||
g_middleware.SMiddlewares.MiddlewareCORS,
|
||||
g_middleware.SMiddlewares.ResponseHandler,
|
||||
)
|
||||
|
||||
s.Group("/", func(group *ghttp.RouterGroup) {
|
||||
// 首页HTML
|
||||
group.ALL("/", func(r *ghttp.Request) {
|
||||
r.Response.Write(g_index.IndexHTML)
|
||||
})
|
||||
group.ALL("/version", func(r *ghttp.Request) {
|
||||
r.Response.Write(binInfo.VersionString())
|
||||
})
|
||||
// 接口绑定
|
||||
r_hamster_router.BindController(group)
|
||||
})
|
||||
|
||||
// 初始化
|
||||
if err := boot.Boot(); err != nil {
|
||||
glog.Fatal(ctx, "初始化任务失败: ", err)
|
||||
}
|
||||
|
||||
s.Run()
|
||||
return nil
|
||||
},
|
||||
}
|
||||
)
|
5
internal/controller/data_core/data_core.go
Normal file
5
internal/controller/data_core/data_core.go
Normal file
@ -0,0 +1,5 @@
|
||||
// =================================================================================
|
||||
// This is auto-generated by GoFrame CLI tool only once. Fill this file as you wish.
|
||||
// =================================================================================
|
||||
|
||||
package data_core
|
15
internal/controller/data_core/data_core_new.go
Normal file
15
internal/controller/data_core/data_core_new.go
Normal file
@ -0,0 +1,15 @@
|
||||
// =================================================================================
|
||||
// Code generated and maintained by GoFrame CLI tool. DO NOT EDIT.
|
||||
// =================================================================================
|
||||
|
||||
package data_core
|
||||
|
||||
import (
|
||||
"home-network-watcher/api/data_core"
|
||||
)
|
||||
|
||||
type ControllerV1 struct{}
|
||||
|
||||
func NewV1() data_core.IDataCoreV1 {
|
||||
return &ControllerV1{}
|
||||
}
|
@ -0,0 +1,40 @@
|
||||
package data_core
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/encoding/gjson"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/os/gcache"
|
||||
"home-network-watcher/manifest"
|
||||
"time"
|
||||
|
||||
"home-network-watcher/api/data_core/v1"
|
||||
)
|
||||
|
||||
func (c *ControllerV1) GetDockerMonitorSSE(ctx context.Context, _ *v1.GetDockerMonitorSSEReq) (res *v1.GetDockerMonitorSSERes, err error) {
|
||||
request := g.RequestFromCtx(ctx)
|
||||
request.Response.Header().Set("Content-Type", "text/event-stream")
|
||||
request.Response.Header().Set("Cache-Control", "no-cache")
|
||||
request.Response.Header().Set("Connection", "keep-alive")
|
||||
request.Response.Header().Set("Access-Control-Allow-Origin", "*")
|
||||
request.Response.Header().Set("X-Accel-Buffering", "no")
|
||||
|
||||
for {
|
||||
// 从缓存中获取数据
|
||||
dockerData, err := gcache.Get(ctx, manifest.DockerMonitorCacheKey)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
res = &v1.GetDockerMonitorSSERes{DockerData: dockerData}
|
||||
// 发送数据
|
||||
request.Response.Writefln("data: " + gjson.New(res).MustToJsonString() + "\n")
|
||||
request.Response.Flush()
|
||||
|
||||
// 等待10秒或者上下文取消
|
||||
select {
|
||||
case <-time.After(10 * time.Second):
|
||||
case <-ctx.Done():
|
||||
return nil, ctx.Err()
|
||||
}
|
||||
}
|
||||
}
|
@ -0,0 +1,39 @@
|
||||
package data_core
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/encoding/gjson"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/os/gcache"
|
||||
"home-network-watcher/api/data_core/v1"
|
||||
"home-network-watcher/manifest"
|
||||
"time"
|
||||
)
|
||||
|
||||
func (c *ControllerV1) GetHomeDataSSE(ctx context.Context, _ *v1.GetHomeDataSSEReq) (res *v1.GetHomeDataSSERes, err error) {
|
||||
request := g.RequestFromCtx(ctx)
|
||||
request.Response.Header().Set("Content-Type", "text/event-stream")
|
||||
request.Response.Header().Set("Cache-Control", "no-cache")
|
||||
request.Response.Header().Set("Connection", "keep-alive")
|
||||
request.Response.Header().Set("Access-Control-Allow-Origin", "*")
|
||||
request.Response.Header().Set("X-Accel-Buffering", "no")
|
||||
|
||||
for {
|
||||
// 从缓存中获取数据
|
||||
homeData, err := gcache.Get(ctx, manifest.HaEntitiesCacheKey)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
res = &v1.GetHomeDataSSERes{HomeData: homeData}
|
||||
// 发送数据
|
||||
request.Response.Writefln("data: " + gjson.New(res).MustToJsonString() + "\n")
|
||||
request.Response.Flush()
|
||||
|
||||
// 等待1秒或者上下文取消
|
||||
select {
|
||||
case <-time.After(1 * time.Second):
|
||||
case <-ctx.Done():
|
||||
return nil, ctx.Err()
|
||||
}
|
||||
}
|
||||
}
|
@ -0,0 +1,63 @@
|
||||
package data_core
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/encoding/gjson"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/os/gcache"
|
||||
"home-network-watcher/manifest"
|
||||
"time"
|
||||
|
||||
"home-network-watcher/api/data_core/v1"
|
||||
)
|
||||
|
||||
func (c *ControllerV1) GetNetworkDataSSE(ctx context.Context, _ *v1.GetNetworkDataSSEReq) (res *v1.GetNetworkDataSSERes, err error) {
|
||||
request := g.RequestFromCtx(ctx)
|
||||
request.Response.Header().Set("Content-Type", "text/event-stream")
|
||||
request.Response.Header().Set("Cache-Control", "no-cache")
|
||||
request.Response.Header().Set("Connection", "keep-alive")
|
||||
request.Response.Header().Set("Access-Control-Allow-Origin", "*")
|
||||
request.Response.Header().Set("X-Accel-Buffering", "no")
|
||||
|
||||
for {
|
||||
// 从缓存中获取数据
|
||||
nodeInfo, err := gcache.Get(ctx, manifest.ProxyNodeCacheKey)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
homeNetwork, err := gcache.Get(ctx, manifest.HomeNetworkCacheKey)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
proxyNetwork, err := gcache.Get(ctx, manifest.ProxyNetworkCacheKey)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
coffeeInfo, err := gcache.Get(ctx, manifest.ProxySubscribeCacheKey)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
serverInfo, err := gcache.Get(ctx, manifest.ServerDataCacheKey)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
res = &v1.GetNetworkDataSSERes{
|
||||
NodeInfo: nodeInfo,
|
||||
HomeNetwork: homeNetwork,
|
||||
ProxyNetwork: proxyNetwork,
|
||||
CoffeeInfo: coffeeInfo,
|
||||
ServerInfo: serverInfo,
|
||||
}
|
||||
// 发送数据
|
||||
request.Response.Writefln("data: " + gjson.New(res).MustToJsonString() + "\n")
|
||||
request.Response.Flush()
|
||||
|
||||
// 等待1秒或者上下文取消
|
||||
select {
|
||||
case <-time.After(1 * time.Second):
|
||||
case <-ctx.Done():
|
||||
return nil, ctx.Err()
|
||||
}
|
||||
}
|
||||
|
||||
}
|
24
internal/controller/data_core/data_core_v1_get_task_list.go
Normal file
24
internal/controller/data_core/data_core_v1_get_task_list.go
Normal file
@ -0,0 +1,24 @@
|
||||
package data_core
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/os/gcron"
|
||||
|
||||
"home-network-watcher/api/data_core/v1"
|
||||
)
|
||||
|
||||
// GetTaskList 获取任务列表
|
||||
func (c *ControllerV1) GetTaskList(_ context.Context, _ *v1.GetTaskListReq) (res *v1.GetTaskListRes, err error) {
|
||||
var taskList []map[string]interface{}
|
||||
for _, entry := range gcron.Entries() {
|
||||
taskList = append(taskList, map[string]interface{}{
|
||||
"name": entry.Name,
|
||||
"status": entry.Status(),
|
||||
"time": entry.Time,
|
||||
})
|
||||
}
|
||||
res = &v1.GetTaskListRes{
|
||||
TaskList: taskList,
|
||||
}
|
||||
return res, nil
|
||||
}
|
@ -0,0 +1,39 @@
|
||||
package data_core
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/encoding/gjson"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/os/gcache"
|
||||
"home-network-watcher/api/data_core/v1"
|
||||
"home-network-watcher/manifest"
|
||||
"time"
|
||||
)
|
||||
|
||||
func (c *ControllerV1) GetUptimeDataSSE(ctx context.Context, _ *v1.GetUptimeDataSSEReq) (res *v1.GetUptimeDataSSERes, err error) {
|
||||
request := g.RequestFromCtx(ctx)
|
||||
request.Response.Header().Set("Content-Type", "text/event-stream")
|
||||
request.Response.Header().Set("Cache-Control", "no-cache")
|
||||
request.Response.Header().Set("Connection", "keep-alive")
|
||||
request.Response.Header().Set("Access-Control-Allow-Origin", "*")
|
||||
request.Response.Header().Set("X-Accel-Buffering", "no")
|
||||
|
||||
for {
|
||||
// 从缓存中获取数据
|
||||
uptimeData, err := gcache.Get(ctx, manifest.UptimeCacheKey)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
res = &v1.GetUptimeDataSSERes{UptimeData: uptimeData}
|
||||
// 发送数据
|
||||
request.Response.Writefln("data: " + gjson.New(res).MustToJsonString() + "\n")
|
||||
request.Response.Flush()
|
||||
|
||||
// 等待10秒或者上下文取消
|
||||
select {
|
||||
case <-time.After(10 * time.Second):
|
||||
case <-ctx.Done():
|
||||
return nil, ctx.Err()
|
||||
}
|
||||
}
|
||||
}
|
@ -0,0 +1,35 @@
|
||||
package data_core
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/encoding/gjson"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/os/gcache"
|
||||
"home-network-watcher/manifest"
|
||||
"time"
|
||||
|
||||
"home-network-watcher/api/data_core/v1"
|
||||
)
|
||||
|
||||
func (c *ControllerV1) GetXuiDataSSE(ctx context.Context, _ *v1.GetXuiDataSSEReq) (res *v1.GetXuiDataSSERes, err error) {
|
||||
request := g.RequestFromCtx(ctx)
|
||||
request.Response.Header().Set("Content-Type", "text/event-stream")
|
||||
request.Response.Header().Set("Cache-Control", "no-cache")
|
||||
request.Response.Header().Set("Connection", "keep-alive")
|
||||
request.Response.Header().Set("Access-Control-Allow-Origin", "*")
|
||||
request.Response.Header().Set("X-Accel-Buffering", "no")
|
||||
|
||||
for {
|
||||
// 从缓存中获取数据
|
||||
xuiData, err := gcache.Get(context.Background(), manifest.XUIUserListCacheKey)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
res = &v1.GetXuiDataSSERes{XuiData: xuiData}
|
||||
// 发送数据
|
||||
request.Response.Writefln("data: " + gjson.New(res).MustToJsonString() + "\n")
|
||||
request.Response.Flush()
|
||||
time.Sleep(5 * time.Second)
|
||||
}
|
||||
|
||||
}
|
25
internal/controller/data_core/data_core_v1_recover_task.go
Normal file
25
internal/controller/data_core/data_core_v1_recover_task.go
Normal file
@ -0,0 +1,25 @@
|
||||
package data_core
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/os/gcron"
|
||||
"home-network-watcher/internal/global/g_functions"
|
||||
|
||||
"home-network-watcher/api/data_core/v1"
|
||||
)
|
||||
|
||||
// RecoverTask 恢复任务
|
||||
func (c *ControllerV1) RecoverTask(_ context.Context, req *v1.RecoverTaskReq) (res *v1.RecoverTaskRes, err error) {
|
||||
var taskMap = make(map[string]int)
|
||||
for _, entry := range gcron.Entries() {
|
||||
taskMap[entry.Name] = entry.Status()
|
||||
}
|
||||
if _, ok := taskMap[req.Name]; !ok {
|
||||
return nil, g_functions.ResErr(400, "任务不存在")
|
||||
}
|
||||
if taskMap[req.Name] == 1 || taskMap[req.Name] == 0 {
|
||||
return nil, g_functions.ResErr(400, "任务已启动")
|
||||
}
|
||||
gcron.Start(req.Name)
|
||||
return
|
||||
}
|
22
internal/controller/data_core/data_core_v1_stop_task.go
Normal file
22
internal/controller/data_core/data_core_v1_stop_task.go
Normal file
@ -0,0 +1,22 @@
|
||||
package data_core
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/os/gcron"
|
||||
"home-network-watcher/internal/global/g_functions"
|
||||
|
||||
"home-network-watcher/api/data_core/v1"
|
||||
)
|
||||
|
||||
// StopTask 停止任务
|
||||
func (c *ControllerV1) StopTask(_ context.Context, req *v1.StopTaskReq) (res *v1.StopTaskRes, err error) {
|
||||
var taskMap = make(map[string]string)
|
||||
for _, entry := range gcron.Entries() {
|
||||
taskMap[entry.Name] = entry.Name
|
||||
}
|
||||
if _, ok := taskMap[req.Name]; !ok {
|
||||
return nil, g_functions.ResErr(400, "任务不存在")
|
||||
}
|
||||
gcron.Stop(req.Name)
|
||||
return
|
||||
}
|
42
internal/global/g_functions/functions.go
Normal file
42
internal/global/g_functions/functions.go
Normal file
@ -0,0 +1,42 @@
|
||||
package g_functions
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/errors/gcode"
|
||||
"github.com/gogf/gf/v2/errors/gerror"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/os/glog"
|
||||
"github.com/gogf/gf/v2/util/gconv"
|
||||
)
|
||||
|
||||
func ResErr(code int, errMsg ...interface{}) error {
|
||||
resMsg := ""
|
||||
errDetail := ""
|
||||
switch len(errMsg) {
|
||||
case 0:
|
||||
resMsg = "Unknown error reason"
|
||||
case 1:
|
||||
resMsg = gconv.String(errMsg[0])
|
||||
default:
|
||||
for _, v := range gconv.SliceStr(errMsg[:len(errMsg)-1]) {
|
||||
resMsg += v + ","
|
||||
}
|
||||
resMsg = resMsg[:len(resMsg)-1]
|
||||
errDetail = gconv.String(errMsg[len(errMsg)-1])
|
||||
}
|
||||
return gerror.NewCode(gcode.New(code, resMsg, nil), errDetail)
|
||||
}
|
||||
|
||||
// SetDefaultHandler 替代默认的日志handler
|
||||
func SetDefaultHandler() {
|
||||
glog.SetDefaultHandler(func(ctx context.Context, in *glog.HandlerInput) {
|
||||
m := map[string]interface{}{
|
||||
"stdout": true,
|
||||
"writerColorEnable": false,
|
||||
"file": "inside-{Y-m-d}.log",
|
||||
"path": g.Config().MustGet(ctx, "server.logger.path", "log/").String(), // 此处必须重新设置,才可以实现db的log写入文件
|
||||
}
|
||||
_ = in.Logger.SetConfigWithMap(m)
|
||||
in.Next(ctx)
|
||||
})
|
||||
}
|
33
internal/global/g_index/index.go
Normal file
33
internal/global/g_index/index.go
Normal file
@ -0,0 +1,33 @@
|
||||
package g_index
|
||||
|
||||
const IndexHTML = `
|
||||
<!DOCTYPE html>
|
||||
<html lang="zh">
|
||||
<head>
|
||||
<title>仓鼠咖啡馆</title>
|
||||
<style>
|
||||
body {
|
||||
margin: 0;
|
||||
height: 100vh;
|
||||
background-image: url('https://s2.loli.net/2023/05/16/Nyaxz3QrsSKmvHE.jpg');
|
||||
background-size: cover;
|
||||
background-position: center;
|
||||
display: flex;
|
||||
justify-content: center;
|
||||
align-items: center;
|
||||
text-align: center;
|
||||
}
|
||||
.text {
|
||||
font-family: Arial, sans-serif;
|
||||
font-size: 24px;
|
||||
color: white;
|
||||
background-color: rgba(0, 0, 0, 0.7);
|
||||
padding: 10px 20px;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<div class="text">仓鼠咖啡馆</div>
|
||||
</body>
|
||||
</html>
|
||||
`
|
191
internal/global/g_middleware/middleware.go
Normal file
191
internal/global/g_middleware/middleware.go
Normal file
@ -0,0 +1,191 @@
|
||||
package g_middleware
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/errors/gerror"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/net/ghttp"
|
||||
"github.com/gogf/gf/v2/os/glog"
|
||||
"github.com/golang-jwt/jwt/v5"
|
||||
"home-network-watcher/manifest"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
type sMiddleware struct{} // 创建结构体
|
||||
var (
|
||||
once = &sync.Once{} // 创建互锁
|
||||
s *sMiddleware // 创建指针
|
||||
SMiddlewares = newMiddleware() // 对外暴露
|
||||
)
|
||||
|
||||
// defaultHandlerResponse 返回结构体
|
||||
type defaultHandlerResponse struct {
|
||||
Status int `json:"status" dc:"Error code"` // 可业务需要可以更改json字段
|
||||
Message string `json:"msg" dc:"Error message"`
|
||||
Data interface{} `json:"data" dc:"Result data for certain request according API definition"`
|
||||
}
|
||||
|
||||
// newMiddleware 单例中间件
|
||||
func newMiddleware() *sMiddleware {
|
||||
once.Do(func() {
|
||||
s = &sMiddleware{}
|
||||
})
|
||||
return s
|
||||
}
|
||||
|
||||
func (s *sMiddleware) MiddlewareCORS(r *ghttp.Request) {
|
||||
r.Response.CORSDefault()
|
||||
r.Middleware.Next()
|
||||
}
|
||||
|
||||
// resWriteJson 返回json输出
|
||||
func (s *sMiddleware) resWriteJson(r *ghttp.Request, in defaultHandlerResponse) {
|
||||
r.Response.ClearBuffer()
|
||||
r.Response.WriteJson(defaultHandlerResponse{
|
||||
Status: in.Status,
|
||||
Message: in.Message,
|
||||
Data: in.Data,
|
||||
})
|
||||
}
|
||||
|
||||
// ErrorsStatus 服务器错误码处理
|
||||
func (s *sMiddleware) ErrorsStatus(server *ghttp.Server) {
|
||||
server.BindStatusHandlerByMap(map[int]ghttp.HandlerFunc{
|
||||
500: func(r *ghttp.Request) {
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: 500,
|
||||
Message: "Error 500,Internal Server Error",
|
||||
Data: "",
|
||||
})
|
||||
},
|
||||
404: func(r *ghttp.Request) {
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: 404,
|
||||
Message: "Error 404,Not Found",
|
||||
Data: "",
|
||||
})
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
// JWTAuth 鉴权中间件
|
||||
func (s *sMiddleware) JWTAuth(r *ghttp.Request) {
|
||||
JWTString := r.GetHeader("Authorization")
|
||||
if JWTString == "" {
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: 401,
|
||||
Message: "JWT为空",
|
||||
Data: nil,
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// 验证JWT
|
||||
token, err := jwt.Parse(JWTString, func(token *jwt.Token) (interface{}, error) {
|
||||
return manifest.JWTKey, nil
|
||||
})
|
||||
if err != nil {
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: 401,
|
||||
Message: "JWT验证失败",
|
||||
Data: nil,
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
if !token.Valid {
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: 401,
|
||||
Message: "JWT验证失败",
|
||||
Data: nil,
|
||||
})
|
||||
return
|
||||
}
|
||||
// 验证是否过期
|
||||
expirationTime, err := token.Claims.GetExpirationTime()
|
||||
if err != nil {
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: 401,
|
||||
Message: "JWT验证失败",
|
||||
Data: nil,
|
||||
})
|
||||
return
|
||||
}
|
||||
if expirationTime.Before(time.Now()) {
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: 401,
|
||||
Message: "JWT已过期",
|
||||
Data: nil,
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
glog.Info(context.TODO(), "JWT验证通过")
|
||||
// 验证通过,设置用户信息
|
||||
audience, err := token.Claims.GetAudience()
|
||||
if err != nil {
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: 401,
|
||||
Message: "JWT验证失败",
|
||||
Data: nil,
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
r.SetCtxVar("user_id", audience[0])
|
||||
r.Middleware.Next()
|
||||
}
|
||||
|
||||
// ResponseHandler is the middleware handling handler response object and its error. 中间件处理处理程序响应对象及其错误
|
||||
func (s *sMiddleware) ResponseHandler(r *ghttp.Request) {
|
||||
//defer action_log.NewActionLog().LogAdd(r) // 日志钩子
|
||||
r.Middleware.Next()
|
||||
|
||||
// There's custom buffer content, it then exits current handler.
|
||||
if r.Response.BufferLength() > 0 {
|
||||
return
|
||||
}
|
||||
|
||||
// 定义参数和类型.方便调用
|
||||
var (
|
||||
code int
|
||||
msg string
|
||||
resData = r.GetHandlerResponse() // 检索并返回处理程序响应对象及其错误
|
||||
err = r.GetError() // 返回请求过程中发生的错误。如果没有错误,则返回 nil
|
||||
errCode = gerror.Code(err) // 将错误信息通过接口解析处理
|
||||
)
|
||||
code = 200 // 默认返回状态码信息码
|
||||
msg = "success" // 默认成功返回响应体文本信息
|
||||
if err != nil {
|
||||
code = 400 // 默认返回状态码信息码
|
||||
switch errCode.Code() {
|
||||
case -1:
|
||||
msg = "Nonstandard error return"
|
||||
resData = "I won't expose it to you. Go back and change it!"
|
||||
case 500:
|
||||
r.Response.Writer.Status = 500
|
||||
return
|
||||
default:
|
||||
msg = errCode.Message()
|
||||
resData = err.Error()
|
||||
r.Response.Writer.Status = 400
|
||||
if msg == "" || msg == "Unknown error reason" {
|
||||
msg = "Nonstandard error return"
|
||||
resData = "I won't expose it to you. Go back and change it!"
|
||||
}
|
||||
if resData == msg {
|
||||
resData = ""
|
||||
}
|
||||
}
|
||||
}
|
||||
// 如果是空res返回,则resData返回空字符串
|
||||
if g.IsNil(resData) {
|
||||
resData = ""
|
||||
}
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: code,
|
||||
Message: msg,
|
||||
Data: resData,
|
||||
})
|
||||
}
|
0
internal/logic/.gitkeep
Normal file
0
internal/logic/.gitkeep
Normal file
7
internal/logic/logic.go
Normal file
7
internal/logic/logic.go
Normal file
@ -0,0 +1,7 @@
|
||||
// ==========================================================================
|
||||
// Code generated by GoFrame CLI tool. DO NOT EDIT.
|
||||
// ==========================================================================
|
||||
|
||||
package logic
|
||||
|
||||
import ()
|
1
internal/packed/packed.go
Normal file
1
internal/packed/packed.go
Normal file
@ -0,0 +1 @@
|
||||
package packed
|
19
internal/router/r_hamster_router/router.go
Normal file
19
internal/router/r_hamster_router/router.go
Normal file
@ -0,0 +1,19 @@
|
||||
package r_hamster_router
|
||||
|
||||
import (
|
||||
"github.com/gogf/gf/v2/net/ghttp"
|
||||
"home-network-watcher/internal/controller/data_core"
|
||||
)
|
||||
|
||||
func BindController(group *ghttp.RouterGroup) {
|
||||
group.Group("/", func(group *ghttp.RouterGroup) {
|
||||
BindDataCore(group)
|
||||
})
|
||||
}
|
||||
|
||||
// BindDataCore 注册核心数据路由
|
||||
func BindDataCore(group *ghttp.RouterGroup) {
|
||||
group.Group("/", func(group *ghttp.RouterGroup) {
|
||||
group.Bind(data_core.NewV1())
|
||||
})
|
||||
}
|
Reference in New Issue
Block a user