示例#1
0
func TestPoll(t *testing.T) {
	setNowMock("2011-09-09T23:36:13Z")
	defer resetNowMock()

	intAgg := health.NewIntervalAggregation(now())
	data := &health.HealthAggregationsResponse{
		InstanceId:           "web22.12345",
		IntervalDuration:     time.Minute,
		IntervalAggregations: []*health.IntervalAggregation{intAgg},
	}
	stop := serveJson(":5050", data)
	defer func() {
		stop()
	}()

	responses := make(chan *pollResponse, 2)
	poll(health.NewStream(), ":5050", responses)
	response := <-responses

	assert.NotNil(t, response)
	assert.Equal(t, response.HostPort, ":5050")
	assert.Equal(t, response.Timestamp, now())
	assert.Nil(t, response.Err)
	assert.Equal(t, response.Code, 200)
	assert.True(t, response.Nanos > 0 && response.Nanos < int64(time.Second))
	assert.Equal(t, response.InstanceId, "web22.12345")
	// we'll just "trust" that the other stuff gets unmarshalled correctly. We didn't really put anything in there anyway in this test.
}
示例#2
0
文件: api.go 项目: grepory/awsthingy
func combineAggregations(aggregations []*health.IntervalAggregation) *health.IntervalAggregation {
	if len(aggregations) == 0 {
		return nil
	}

	overallAgg := health.NewIntervalAggregation(aggregations[0].IntervalStart)
	for _, ia := range aggregations {
		overallAgg.Merge(ia)
	}
	return overallAgg
}
示例#3
0
func (hd *HealthD) recalculateIntervals() {
	job := hd.stream.NewJob("recalculate")

	for k, _ := range hd.intervalsNeedingRecalculation {
		intAggsAtTime := []*health.IntervalAggregation{}

		for key, intAgg := range hd.hostAggregations {
			if key.Time == k {
				intAggsAtTime = append(intAggsAtTime, intAgg)
			}
		}

		overallAgg := health.NewIntervalAggregation(k)
		for _, ia := range intAggsAtTime {
			overallAgg.Merge(ia)
		}
		hd.setAggregation(overallAgg)
	}

	// Reset everything:
	hd.intervalsNeedingRecalculation = make(map[time.Time]struct{})

	job.Complete(health.Success)
}