org.springframework.data.gemfire.listener.ContinuousQueryListenerContainer Java Examples

@Override
protected void doHealthCheck(Health.Builder builder) {

	if (getContinuousQueryListenerContainer().isPresent()) {

		Optional<QueryService> queryService = getContinuousQueryListenerContainer()
			.map(ContinuousQueryListenerContainer::getQueryService);

		List<CqQuery> continuousQueries = queryService
			.map(QueryService::getCqs)
			.map(Arrays::asList)
			.orElseGet(Collections::emptyList);

		builder.withDetail("geode.continuous-query.count", continuousQueries.size());

		queryService
			.map(QueryService::getCqStatistics)
			.ifPresent(cqServiceStatistics ->

				builder.withDetail("geode.continuous-query.number-of-active", cqServiceStatistics.numCqsActive())
					.withDetail("geode.continuous-query.number-of-closed", cqServiceStatistics.numCqsClosed())
					.withDetail("geode.continuous-query.number-of-created", cqServiceStatistics.numCqsCreated())
					.withDetail("geode.continuous-query.number-of-stopped", cqServiceStatistics.numCqsStopped())
					.withDetail("geode.continuous-query.number-on-client", cqServiceStatistics.numCqsOnClient())
			);

		continuousQueries.stream()
			.filter(Objects::nonNull)
			.forEach(continuousQuery -> {

				String continuousQueryName = continuousQuery.getName();

				builder.withDetail(continuousQueryKey(continuousQueryName,"oql-query-string"), continuousQuery.getQueryString())
					.withDetail(continuousQueryKey(continuousQueryName, "closed"), toYesNoString(continuousQuery.isClosed()))
					.withDetail(continuousQueryKey(continuousQueryName, "closing"), toYesNoString(continuousQuery.getState()))
					.withDetail(continuousQueryKey(continuousQueryName, "durable"), toYesNoString(continuousQuery.isDurable()))
					.withDetail(continuousQueryKey(continuousQueryName, "running"), toYesNoString(continuousQuery.isRunning()))
					.withDetail(continuousQueryKey(continuousQueryName, "stopped"), toYesNoString(continuousQuery.isStopped()));

				Query query = continuousQuery.getQuery();

				if (query != null) {

					QueryStatistics queryStatistics = query.getStatistics();

					if (queryStatistics != null) {
						builder.withDetail(continuousQueryQueryKey(continuousQueryName, "number-of-executions"), queryStatistics.getNumExecutions())
							.withDetail(continuousQueryQueryKey(continuousQueryName, "total-execution-time"), queryStatistics.getTotalExecutionTime());
					}
				}

				CqStatistics continuousQueryStatistics = continuousQuery.getStatistics();

				if (continuousQueryStatistics != null) {

					builder.withDetail(continuousQueryStatisticsKey(continuousQueryName, "number-of-deletes"), continuousQueryStatistics.numDeletes())
						.withDetail(continuousQueryStatisticsKey(continuousQueryName, "number-of-events"), continuousQueryStatistics.numEvents())
						.withDetail(continuousQueryStatisticsKey(continuousQueryName, "number-of-inserts"), continuousQueryStatistics.numInserts())
						.withDetail(continuousQueryStatisticsKey(continuousQueryName, "number-of-updates"), continuousQueryStatistics.numUpdates());
				}
			});

		builder.up();

		return;
	}

	builder.unknown();
}
 

@Bean("GeodeContinuousQueryHealthIndicator")
GeodeContinuousQueriesHealthIndicator continuousQueriesHealthIndicator(
		@Autowired(required = false) ContinuousQueryListenerContainer continuousQueryListenerContainer) {

	return new GeodeContinuousQueriesHealthIndicator(continuousQueryListenerContainer);
}
 

@Before
public void setup() {

	ContinuousQueryListenerContainer container = new ContinuousQueryListenerContainer();

	container.setQueryService(this.mockQueryService);

	this.continuousQueriesHealthIndicator = new GeodeContinuousQueriesHealthIndicator(container);
}
 

/**
 * Returns an {@link Optional} reference to the configured {@link ContinuousQueryListenerContainer}.
 *
 * @return an {@link Optional} reference to the configured {@link ContinuousQueryListenerContainer}.
 * @see org.springframework.data.gemfire.listener.ContinuousQueryListenerContainer
 * @see java.util.Optional
 */
protected Optional<ContinuousQueryListenerContainer> getContinuousQueryListenerContainer() {
	return Optional.ofNullable(this.continuousQueryListenerContainer);
}
 

/**
 * Constructs an instance of the {@link GeodeContinuousQueriesHealthIndicator} initialized with a reference to
 * the {@link ContinuousQueryListenerContainer}.
 *
 * @param continuousQueryListenerContainer reference to the SDG {@link ContinuousQueryListenerContainer}.
 * @see org.springframework.data.gemfire.listener.ContinuousQueryListenerContainer
 */
public GeodeContinuousQueriesHealthIndicator(ContinuousQueryListenerContainer continuousQueryListenerContainer) {

	super("Continuous Queries health check enabled");

	this.continuousQueryListenerContainer = continuousQueryListenerContainer;
}