重构msg的包

backend/services/msg_handler/handler.go

@@ -0,0 +1,48 @@
+package msg_handler
+
+import (
+	"crawlab/constants"
+	"crawlab/model"
+)
+
+type Handler interface {
+	Handle() error
+}
+
+func GetMsgHandler(msg NodeMessage) Handler {
+	if msg.Type == constants.MsgTypeGetLog {
+		return &Log{
+			msg: msg,
+		}
+	} else if msg.Type == constants.MsgTypeCancelTask {
+		return &Task{
+			msg: msg,
+		}
+	} else if msg.Type == constants.MsgTypeGetSystemInfo {
+		return &SystemInfo{
+			msg: msg,
+		}
+	}
+	return nil
+}
+
+type NodeMessage struct {
+	// 通信类别
+	Type string `json:"type"`
+
+	// 任务相关
+	TaskId string `json:"task_id"` // 任务ID
+
+	// 节点相关
+	NodeId string `json:"node_id"` // 节点ID
+
+	// 日志相关
+	LogPath string `json:"log_path"` // 日志路径
+	Log     string `json:"log"`      // 日志
+
+	// 系统信息
+	SysInfo model.SystemInfo `json:"sys_info"`
+
+	// 错误相关
+	Error string `json:"error"`
+}