示例#1
0
//TODO(yingsong.wu):需要优化
func (this *Agent) Dial(Type string) {
	var (
		tempDelay time.Duration
		err       error
		session   *tcp.Session
	)
retry:
	switch Type {
	case "cfc":
		session, err = this.Connect(GlobalConfig.CFC_ADDR, nil)
	case "repeater":
		session, err = this.Connect(GlobalConfig.REPEATER_ADDR, nil)
	default:

	}
	if err != nil {
		lg.Error("%s: %s", Type, err.Error())
		if tempDelay == 0 {
			tempDelay = 5 * time.Millisecond
		} else {
			tempDelay *= 2
		}
		if max := 5 * time.Second; tempDelay > max {
			tempDelay = max
		}
		time.Sleep(tempDelay)
		goto retry
	}
	switch Type {
	case "cfc":
		this.cfc = session
		this.Addr = session.LocalAddr()
	case "repeater":
		this.repeater = session
	}
	for {
		switch Type {
		case "cfc":
			if this.cfc.IsClosed() {
				goto retry
			}
		case "repeater":
			if this.repeater.IsClosed() {
				goto retry
			}
		}
		time.Sleep(time.Second * 1)
	}
}
示例#2
0
//数据包逻辑处理
func (this *handle) Handle(sess *tcp.Session, data []byte) {
	defer func() {
		if r := recover(); r != nil {
			lg.Error("Recovered in Handle", r)
		}
	}()
	var err error
	mt := types.MessageType(data[0])
	lg.Info("receive %v %v", types.MessageTypeText[mt], string(data[1:]))
	if proxy.cfc.IsClosed() {
		return
	}
	switch mt {

	case types.MESS_POST_METRIC, types.MESS_POST_HOST_CONFIG, types.MESS_POST_HOST_ALIVE:
		proxy.cfc.Send(data)
	case types.MESS_GET_HOST_PLUGIN_LIST:
		proxy.cfc.Send(data)
		host := types.Host{}
		if err = host.Decode(data[1:]); err != nil {
			lg.Error("decode host error ", err.Error())
			return
		}
		if resp := proxy.PluginList.Get(host.ID); len(resp) > 0 {
			sess.Send(
				types.Pack(
					types.MESS_GET_HOST_PLUGIN_LIST_RESP,
					&types.GetPluginResp{host.ID, resp},
				))
		}
	case types.MESS_GET_HOST_PLUGIN_LIST_RESP:
		resp := types.GetPluginResp{}
		if err = resp.Decode(data[1:]); err != nil {
			lg.Error("decode plugin response error %s", err)
			return
		}
		proxy.PluginList.Set(resp.HostID, resp.Plugins)
	default:
		lg.Error("Unknown Option: %v", mt)
	}
}
示例#3
0
func (this *ControllerHandle) Handle(sess *tcp.Session, data []byte) {
	defer func() {
		if err := recover(); err != nil {
			lg.Error("Recovered in HandleMessage", err)
		}
	}()
	mt := types.AlarmMessageType(data[0])
	switch mt {
	case types.ALAR_MESS_INSPECTOR_HEARTBEAT:
		lg.Info("Receive %v %v", types.AlarmMessageTypeText[mt], string(data[1:]))
		heartbeat := &types.HeartBeat{}
		if err := heartbeat.Decode(data[1:]); err != nil {
			lg.Error(err.Error())
			return
		}
		controller.refreshNode(heartbeat)
	case types.ALAR_MESS_INSPECTOR_TASK_REQUEST:
		tasks_resp := getInspectorTask()
		sess.Send(types.AlarmPack(types.ALAR_MESS_INSPECTOR_TASKS, tasks_resp))
	case types.ALAR_MESS_INSPECTOR_RESULT:
		lg.Info("Receive %v %v", types.AlarmMessageTypeText[mt], string(data[1:]))
		result := &types.StrategyResult{}
		if err := result.Decode(data[1:]); err != nil {
			lg.Error(err.Error())
			return
		}
		switch result.Priority {
		case types.PRIORITY_HIGH_LEVEL:
			controller.highResultPool.PutResult(result)
		case types.PRIORITY_MIDDLE_LEVEL:
			controller.lowResultPool.PutResult(result)
		case types.PRIORITY_LOW_LEVEL:
			controller.lowResultPool.PutResult(result)
		default:
			lg.Error("Unknown priority: %v", result.Priority)
		}
	default:
		lg.Error("Unknown option: %v", mt)
	}
}
示例#4
0
func (this *ControllerHandle) LostSession(sess *tcp.Session) {
	lg.Info("%s disconnect ", sess.RemoteAddr())
}
示例#5
0
func (this *ControllerHandle) MakeSession(sess *tcp.Session) {
	lg.Info("%s new connection ", sess.RemoteAddr())
}
示例#6
0
文件: handle.go 项目: avldya/OWL-v3
func (this *InspectorHandle) LostSession(sess *tcp.Session) {
	lg.Info("%s disconnect ", sess.RemoteAddr())
}
示例#7
0
//数据包逻辑处理
//TODO: 获取所有插件md5、更新插件
func (this *handle) Handle(sess *tcp.Session, data []byte) {
	defer func() {
		if r := recover(); r != nil {
			lg.Error("Recovered in HandlerMessage", r)
		}
	}()
	var err error
	mt := types.MessageType(data[0])
	lg.Info("receive %v %v", types.MessageTypeText[mt], string(data[1:]))
	switch mt {
	//agent上传基础信息, ip hostname agentVersion
	case types.MESS_POST_HOST_CONFIG:
		host := &types.Host{}
		err = host.Decode(data[1:])
		if err != nil {
			lg.Error(err.Error())
			sess.Close()
			return
		}
		if h := mydb.GetHost(host.ID); h == nil { //主机不存在
			err := mydb.CreateHost(host.ID, host.SN, host.IP, host.Hostname, host.AgentVersion)
			if err != nil {
				lg.Error("create host error: %s", err.Error())
			} else {
				lg.Info("create host:%v", host)
			}
		} else if h.IP != host.IP ||
			h.Hostname != host.Hostname ||
			h.AgentVersion != host.AgentVersion ||
			h.SN != host.SN {
			lg.Info("update host: %v->%v", h, host)
			mydb.UpdateHost(host)
		}
		//同步metric
	case types.MESS_POST_METRIC:
		cfg := types.MetricConfig{}
		err = cfg.Decode(data[1:])
		if err != nil {
			lg.Error(err.Error())
			sess.Close()
		}
		host := mydb.GetHost(cfg.HostID)
		if host == nil {
			return
		}
		if mydb.MetricIsExists(cfg.HostID, cfg.SeriesData.GetMetric()) {
			return
		}
		if err = mydb.CreateMetric(cfg.HostID, cfg.SeriesData); err != nil {
			lg.Error("create metric error %v", err.Error())
		}
		//获取需要执行的插件列表
	case types.MESS_GET_HOST_PLUGIN_LIST:
		host := &types.Host{}
		err = host.Decode(data[1:])
		if err != nil {
			return
		}
		sess.Send(types.Pack(types.MESS_GET_HOST_PLUGIN_LIST_RESP,
			&types.GetPluginResp{HostID: host.ID, Plugins: mydb.GetPlugins(host.ID)}),
		)
	case types.MESS_POST_HOST_ALIVE:
		cfg := &types.Host{}
		err = cfg.Decode(data[1:])
		if err != nil {
			lg.Error(err.Error())
			sess.Close()
		}
		host := mydb.GetHost(cfg.ID)
		if host == nil {
			return
		}
		mydb.UpdateHost(host)
	default:
		lg.Error("Unknown Option: %v", mt)
	}
}