func V1AddMonitor(c *middleware.Context, cmd m.AddMonitorCommand) {
cmd.OrgId = c.OrgId
if cmd.EndpointId == 0 {
c.JSON(400, "EndpointId not set.")
return
}
if cmd.MonitorTypeId == 0 {
c.JSON(400, "MonitorTypeId not set.")
return
}
if cmd.MonitorTypeId > 4 {
c.JSON(400, "Invlaid MonitorTypeId.")
return
}
if cmd.Frequency == 0 {
c.JSON(400, "Frequency not set.")
return
}
// get the endpoint that the check belongs too.
endpoint, err := sqlstore.GetEndpointById(c.OrgId, cmd.EndpointId)
if err != nil {
handleError(c, err)
return
}
if endpoint == nil {
c.JSON(400, "endpoint does not exist.")
return
}
for _, check := range endpoint.Checks {
if checkTypeToId(check.Type) == cmd.MonitorTypeId {
c.JSON(400, fmt.Sprintf("Endpoint already has a %s check.", check.Type))
return
}
}
route := &m.CheckRoute{}
if len(cmd.CollectorTags) > 0 {
route.Type = m.RouteByTags
route.Config = map[string]interface{}{
"tags": cmd.CollectorTags,
}
} else {
route.Type = m.RouteByIds
route.Config = map[string]interface{}{
"ids": cmd.CollectorIds,
}
}
check := m.Check{
OrgId: cmd.OrgId,
EndpointId: cmd.EndpointId,
Type: m.MonitorTypeToCheckTypeMap[cmd.MonitorTypeId-1],
Frequency: cmd.Frequency,
Enabled: cmd.Enabled,
HealthSettings: cmd.HealthSettings,
Created: time.Now(),
Updated: time.Now(),
Route: route,
Settings: m.MonitorSettingsDTO(cmd.Settings).ToV2Setting(m.MonitorTypeToCheckTypeMap[cmd.MonitorTypeId-1]),
}
err = sqlstore.ValidateCheckRoute(&check)
if err != nil {
handleError(c, err)
return
}
endpoint.Checks = append(endpoint.Checks, check)
//Update endpoint
err = sqlstore.UpdateEndpoint(endpoint)
if err != nil {
handleError(c, err)
return
}
var monitor m.MonitorDTO
for _, check := range endpoint.Checks {
if check.Type == m.MonitorTypeToCheckTypeMap[cmd.MonitorTypeId-1] {
monitor = m.MonitorDTOFromCheck(check, endpoint.Slug)
if check.Enabled {
probeList, err := sqlstore.GetProbesForCheck(&check)
if err != nil {
handleError(c, err)
return
}
monitor.Collectors = probeList
}
break
}
}
c.JSON(200, monitor)
return
}
func V1UpdateMonitor(c *middleware.Context, cmd m.UpdateMonitorCommand) {
cmd.OrgId = c.OrgId
if cmd.EndpointId == 0 {
c.JSON(400, "EndpointId not set.")
return
}
if cmd.MonitorTypeId == 0 {
c.JSON(400, "MonitorTypeId not set.")
return
}
if cmd.MonitorTypeId > 4 {
c.JSON(400, "Invlaid MonitorTypeId.")
return
}
if cmd.Frequency == 0 {
c.JSON(400, "Frequency not set.")
return
}
// get the endpoint that the check belongs too.
endpoint, err := sqlstore.GetEndpointById(c.OrgId, cmd.EndpointId)
if err != nil {
handleError(c, err)
return
}
if endpoint == nil {
c.JSON(400, "endpoint does not exist.")
return
}
route := &m.CheckRoute{}
if len(cmd.CollectorTags) > 0 {
route.Type = m.RouteByTags
route.Config = map[string]interface{}{
"tags": cmd.CollectorTags,
}
} else {
route.Type = m.RouteByIds
route.Config = map[string]interface{}{
"ids": cmd.CollectorIds,
}
}
checkPos := 0
found := false
for pos, check := range endpoint.Checks {
if check.Id == cmd.Id {
checkPos = pos
found = true
log.Debug("updating check %d of endpoint %s", check.Id, endpoint.Slug)
if check.Type != m.MonitorTypeToCheckTypeMap[cmd.MonitorTypeId-1] {
c.JSON(400, "monitor Type cant be changed.")
return
}
break
}
}
if !found {
c.JSON(400, "check does not exist in endpoint.")
return
}
endpoint.Checks[checkPos].Frequency = cmd.Frequency
endpoint.Checks[checkPos].Enabled = cmd.Enabled
endpoint.Checks[checkPos].HealthSettings = cmd.HealthSettings
endpoint.Checks[checkPos].Updated = time.Now()
endpoint.Checks[checkPos].Route = route
endpoint.Checks[checkPos].Settings = m.MonitorSettingsDTO(cmd.Settings).ToV2Setting(m.MonitorTypeToCheckTypeMap[cmd.MonitorTypeId-1])
err = sqlstore.ValidateCheckRoute(&endpoint.Checks[checkPos])
if err != nil {
handleError(c, err)
return
}
err = sqlstore.UpdateEndpoint(endpoint)
if err != nil {
handleError(c, err)
return
}
c.JSON(200, "Monitor updated")
}