mirror of
https://github.com/crawlab-team/crawlab.git
synced 2026-01-22 17:31:03 +01:00
38 lines
754 B
Go
38 lines
754 B
Go
package msg_handler
|
|
|
|
import (
|
|
"crawlab/constants"
|
|
"crawlab/entity"
|
|
"github.com/apex/log"
|
|
)
|
|
|
|
type Handler interface {
|
|
Handle() error
|
|
}
|
|
|
|
func GetMsgHandler(msg entity.NodeMessage) Handler {
|
|
log.Debugf("received msg , type is : %s", msg.Type)
|
|
//if msg.Type == constants.MsgTypeGetLog || msg.Type == constants.MsgTypeRemoveLog {
|
|
// // 日志相关
|
|
// return &Log{
|
|
// msg: msg,
|
|
// }
|
|
//} else if msg.Type == constants.MsgTypeCancelTask {
|
|
// // 任务相关
|
|
// return &Task{
|
|
// msg: msg,
|
|
// }
|
|
if msg.Type == constants.MsgTypeGetSystemInfo {
|
|
// 系统信息相关
|
|
return &SystemInfo{
|
|
msg: msg,
|
|
}
|
|
} else if msg.Type == constants.MsgTypeRemoveSpider {
|
|
// 爬虫相关
|
|
return &Spider{
|
|
SpiderId: msg.SpiderId,
|
|
}
|
|
}
|
|
return nil
|
|
}
|