apps/monibuca
1
0
Fork 0
mirror of https://github.com/langhuihui/monibuca.git synced 2025-10-27 05:10:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: complete task system and console

Browse Source
This commit is contained in:
langhuihui
2024-08-23 14:24:04 +08:00
parent 84b7240f7b
commit e883bb94cd
18 changed files with 674 additions and 239 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
plugin/monitor/api.go
View File

@@ -2,6 +2,8 @@ package plugin_monitor
import (
"context"
"errors"
"google.golang.org/protobuf/types/known/emptypb"
"google.golang.org/protobuf/types/known/timestamppb"
"m7s.live/m7s/v5/plugin/monitor/pb"
monitor "m7s.live/m7s/v5/plugin/monitor/pkg"
@@ -9,22 +11,51 @@ import (
)
func (cfg *MonitorPlugin) SearchTask(ctx context.Context, req *pb.SearchTaskRequest) (res *pb.SearchTaskResponse, err error) {
if cfg.DB == nil {
return nil, errors.New("database is not initialized")
}
res = &pb.SearchTaskResponse{}
var tasks []*monitor.Task
tx := cfg.DB.Find(&tasks)
if err = tx.Error; err == nil {
res.Data = slices.Collect(func(yield func(*pb.Task) bool) {
for _, t := range tasks {
yield(&pb.Task{
Id: t.TaskID,
StartTime: timestamppb.New(t.StartTime),
EndTime: timestamppb.New(t.EndTime),
Owner: t.OwnerType,
Type: uint32(t.TaskType),
Description: t.Description,
Reason: t.Reason,
SessionId: t.SessionID,
ParentId: t.ParentID,
if t.SessionID == req.SessionId {
yield(&pb.Task{
Id: t.TaskID,
StartTime: timestamppb.New(t.StartTime),
EndTime: timestamppb.New(t.EndTime),
Owner: t.OwnerType,
Type: uint32(t.TaskType),
Description: t.Description,
Reason: t.Reason,
SessionId: t.SessionID,
ParentId: t.ParentID,
})
}
}
})
}
return
}
func (cfg *MonitorPlugin) SessionList(context.Context, *emptypb.Empty) (res *pb.SessionListResponse, err error) {
if cfg.DB == nil {
return nil, errors.New("database is not initialized")
}
res = &pb.SessionListResponse{}
var sessions []*monitor.Session
tx := cfg.DB.Find(&sessions)
err = tx.Error
if err == nil {
res.Data = slices.Collect(func(yield func(*pb.Session) bool) {
for _, s := range sessions {
yield(&pb.Session{
Id: s.ID,
Pid: uint32(s.PID),
Args: s.Args,
StartTime: timestamppb.New(s.StartTime),
EndTime: timestamppb.New(s.EndTime),
})
}
})