MetricsTemplate.java

/*
 * Copyright 2013 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.springframework.xd.shell.command;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Random;

import org.springframework.shell.core.JLineShellComponent;
import org.springframework.xd.shell.command.fixtures.FieldValueCounterSink;
import org.springframework.xd.shell.command.fixtures.RichGaugeSink;
import org.springframework.xd.test.fixtures.AggregateCounterSink;
import org.springframework.xd.test.fixtures.CounterSink;
import org.springframework.xd.test.fixtures.Disposable;


/**
 * Access point for creation and remembering of metric sink fixtures.
 * 
 * @author Eric Bottard
 */
public class MetricsTemplate implements Disposable {

	private JLineShellComponent shell;

	private List<Disposable> disposables = new ArrayList<Disposable>();

	private Random random = new Random();

	public MetricsTemplate(JLineShellComponent shell) {
		this.shell = shell;
	}

	public CounterSink newCounterSink(String name) {
		return remember(new CounterSink(shell, name));
	}

	public CounterSink newCounterSink() {
		return remember(new CounterSink(shell, randomName()));
	}

	public AggregateCounterSink newAggregateCounterSink(String name) {
		return remember(new AggregateCounterSink(shell, name));
	}

	public AggregateCounterSink newAggregateCounterSink() {
		return remember(new AggregateCounterSink(shell, randomName()));
	}

	public RichGaugeSink newRichGauge(String name) {
		return remember(new RichGaugeSink(shell, name));
	}

	public RichGaugeSink newRichGauge() {
		return remember(new RichGaugeSink(shell, randomName()));
	}

	public FieldValueCounterSink newFieldValueCounterSink(String name, String fieldName) {
		return remember(new FieldValueCounterSink(shell, name, fieldName));
	}

	public FieldValueCounterSink newFieldValueCounterSink(String fieldName) {
		return remember(new FieldValueCounterSink(shell, randomName(), fieldName));
	}

	private String randomName() {
		return "random" + random.nextInt(100000);
	}

	private <T extends Disposable> T remember(T disposable) {
		disposables.add(disposable);
		return disposable;
	}

	@Override
	public void cleanup() {
		Collections.reverse(disposables);
		for (Disposable disposable : disposables) {
			disposable.cleanup();
		}
	}

}