heartbeat.go

package restful

import (
	"net"

	ogin "github.com/Cepave/open-falcon-backend/common/gin"
	"github.com/Cepave/open-falcon-backend/common/gin/mvc"
	cModel "github.com/Cepave/open-falcon-backend/common/model"
	nqmModel "github.com/Cepave/open-falcon-backend/common/model/nqm"
	"github.com/Cepave/open-falcon-backend/modules/mysqlapi/rdb"
	"github.com/Cepave/open-falcon-backend/modules/mysqlapi/service"
	"github.com/gin-gonic/gin"
)

type heartbeatOfFalconAgents []*cModel.FalconAgentHeartbeat

func (agents *heartbeatOfFalconAgents) Bind(context *gin.Context) {
	ogin.BindJson(context, agents)
}

func falconAgentHeartbeat(
	agents *heartbeatOfFalconAgents,
	q *struct {
		UpdateOnly bool `mvc:"query[update_only]"`
	},
) mvc.OutputBody {
	retBody := rdb.FalconAgentHeartbeat(*agents, q.UpdateOnly)
	return mvc.JsonOutputBody(retBody)
}

func nqmAgentHeartbeat(
	req *nqmModel.HeartbeatRequest,
) mvc.OutputBody {
	r := rdb.SelectNqmAgentByConnId(req.ConnectionId)
	if r != nil {
		service.NqmQueue.Put(req)
		r = overwriteNqmAgent(r, req)
	} else {
		r = rdb.InsertNqmAgentByHeartbeat(req)
	}
	return mvc.JsonOutputBody(r)
}

// overwrittenNqmAgent overwrites the result with the values from the heartbeat
// request. The values in the database should be identical in the end.
func overwriteNqmAgent(r *nqmModel.Agent, req *nqmModel.HeartbeatRequest) *nqmModel.Agent {
	r.ConnectionId = req.ConnectionId
	r.Hostname = req.Hostname
	r.IpAddress = net.ParseIP(req.IpAddress.String())
	r.LastHeartBeat = req.Timestamp
	return r
}

func nqmAgentHeartbeatTargetList(
	p *struct {
		AgentID int32 `mvc:"param[agent_id]"`
	},
) mvc.OutputBody {
	l := service.NqmCachedTargetList.Load(p.AgentID)
	return mvc.JsonOutputBody(l)
}