Example #1
0
func GetMonitors(c *middleware.Context, query m.GetMonitorsQuery) Response {
	query.OrgId = c.OrgId
	query.IsGrafanaAdmin = c.IsGrafanaAdmin

	if err := bus.Dispatch(&query); err != nil {
		return ApiError(500, "Failed to query monitors", err)
	}
	return Json(200, query.Result)
}
Example #2
0
func GetMonitors(query *m.GetMonitorsQuery) error {
	sess := x.Table("monitor")
	rawParams := make([]interface{}, 0)
	rawSql := `
SELECT
    GROUP_CONCAT(DISTINCT(monitor_collector.collector_id)) as collector_ids,
    GROUP_CONCAT(DISTINCT(monitor_collector_tag.tag)) as collector_tags,
    GROUP_CONCAT(DISTINCT(collector_tag.collector_id)) as tag_collectors,
    endpoint.slug as endpoint_slug,
    monitor_type.name as monitor_type_name,
    monitor.*
FROM monitor
    INNER JOIN endpoint on endpoint.id = monitor.endpoint_id
    LEFT JOIN monitor_type ON monitor.monitor_type_id = monitor_type.id
    LEFT JOIN monitor_collector ON monitor.id = monitor_collector.monitor_id
    LEFT JOIN monitor_collector_tag ON monitor.id = monitor_collector_tag.monitor_id
    LEFT JOIN collector_tag on collector_tag.tag = monitor_collector_tag.tag AND collector_tag.org_id = monitor.org_id
`
	whereSql := make([]string, 0)
	if !query.IsGrafanaAdmin {
		whereSql = append(whereSql, "monitor.org_id=?")
		rawParams = append(rawParams, query.OrgId)
	}

	if query.Enabled != "" {
		enabled, err := strconv.ParseBool(query.Enabled)
		if err != nil {
			return err
		}
		whereSql = append(whereSql, "monitor.enabled=?")
		rawParams = append(rawParams, enabled)
	}

	if len(query.EndpointId) > 0 {
		p := make([]string, len(query.EndpointId))
		for i, e := range query.EndpointId {
			p[i] = "?"
			rawParams = append(rawParams, e)
		}
		whereSql = append(whereSql, fmt.Sprintf("monitor.endpoint_id IN (%s)", strings.Join(p, ",")))
	}

	if len(query.CollectorId) > 0 {
		rawSql += "LEFT JOIN monitor_collector AS mc ON mc.monitor_id = monitor.id\n"
		rawSql += `LEFT JOIN
        (SELECT
            collector.id AS collector_id,
            collector_tag.tag as tag,
	    collector_tag.org_id as org_id
        FROM collector
        LEFT JOIN collector_tag ON collector.id = collector_tag.collector_id) as ct
		ON ct.tag = monitor_collector_tag.tag
		`
		p := make([]string, len(query.CollectorId))
		for i, c := range query.CollectorId {
			p[i] = "?"
			rawParams = append(rawParams, c)
		}

		p2 := make([]string, len(query.CollectorId))
		for i, e := range query.CollectorId {
			p2[i] = "?"
			rawParams = append(rawParams, e)
		}
		whereSql = append(whereSql, fmt.Sprintf("((ct.org_id=monitor.org_id OR ct.org_id is NULL) AND (mc.collector_id IN (%s) OR ct.collector_id IN (%s)))", strings.Join(p, ","), strings.Join(p2, ",")))
	}

	if query.Modulo > 0 {
		whereSql = append(whereSql, "(monitor.id % ?) = ?")
		rawParams = append(rawParams, query.Modulo, query.ModuloOffset)
	}

	if len(whereSql) > 0 {
		rawSql += "WHERE " + strings.Join(whereSql, " AND ")
	}
	rawSql += " GROUP BY monitor.id"

	result := make([]*MonitorWithCollectorDTO, 0)
	err := sess.Sql(rawSql, rawParams...).Find(&result)
	if err != nil {
		return err
	}

	monitors := make([]*m.MonitorDTO, 0)
	//iterate through all of the results and build out our checks model.
	for _, row := range result {
		monitorCollectorIds := make([]int64, 0)
		monitorCollectorsMap := make(map[int64]bool)
		if row.CollectorIds != "" {
			for _, l := range strings.Split(row.CollectorIds, ",") {
				i, err := strconv.ParseInt(l, 10, 64)
				if err != nil {
					return err
				}
				monitorCollectorIds = append(monitorCollectorIds, i)
				monitorCollectorsMap[i] = true
			}
		}

		monitorCollectorTags := make([]string, 0)
		if row.CollectorTags != "" {
			monitorCollectorTags = strings.Split(row.CollectorTags, ",")
			if row.TagCollectors != "" {
				for _, l := range strings.Split(row.TagCollectors, ",") {
					i, err := strconv.ParseInt(l, 10, 64)
					if err != nil {
						return err
					}
					monitorCollectorsMap[i] = true
				}
			}
		}

		mergedCollectors := make([]int64, len(monitorCollectorsMap))
		count := 0
		for k := range monitorCollectorsMap {
			mergedCollectors[count] = k
			count += 1
		}

		/*
			h := m.MonitorHealthSettingDTO{}
			h.NumCollectors = row.HealthSettings["numCollectors"]
			h.Steps = row.HealthSettings["steps"]
		*/

		monitors = append(monitors, &m.MonitorDTO{
			Id:              row.Id,
			EndpointId:      row.EndpointId,
			OrgId:           row.OrgId,
			EndpointSlug:    row.EndpointSlug,
			MonitorTypeId:   row.MonitorTypeId,
			MonitorTypeName: row.MonitorTypeName,
			CollectorIds:    monitorCollectorIds,
			CollectorTags:   monitorCollectorTags,
			Collectors:      mergedCollectors,
			State:           scrutinizeState(time.Now(), row.State, row.StateCheck, row.Frequency),
			StateChange:     row.StateChange,
			StateCheck:      row.StateCheck,
			Settings:        row.Settings,
			HealthSettings:  row.HealthSettings,
			Frequency:       row.Frequency,
			Enabled:         row.Enabled,
			Offset:          row.Offset,
			Updated:         row.Updated,
		})
	}
	query.Result = monitors

	return nil

}