Examples with ReplicatorStatsImpl org.apache.pulsar.common.policies.data.stats.ReplicatorStatsImpl used on opensource projects

Example 1 with ReplicatorStatsImpl

use of org.apache.pulsar.common.policies.data.stats.ReplicatorStatsImpl in project pulsar by apache.

the class PersistentTopicStatsTest method testPersistentTopicStatsAggregationPartialProducerIsNotSupported.

@Test
public void testPersistentTopicStatsAggregationPartialProducerIsNotSupported() {
    TopicStatsImpl topicStats1 = new TopicStatsImpl();
    topicStats1.msgRateIn = 1;
    topicStats1.msgThroughputIn = 1;
    topicStats1.msgRateOut = 1;
    topicStats1.msgThroughputOut = 1;
    topicStats1.averageMsgSize = 1;
    topicStats1.storageSize = 1;
    final PublisherStatsImpl publisherStats1 = new PublisherStatsImpl();
    publisherStats1.setSupportsPartialProducer(false);
    publisherStats1.setProducerName("name1");
    topicStats1.addPublisher(publisherStats1);
    topicStats1.subscriptions.put("test_ns", new SubscriptionStatsImpl());
    topicStats1.replication.put("test_ns", new ReplicatorStatsImpl());
    TopicStatsImpl topicStats2 = new TopicStatsImpl();
    topicStats2.msgRateIn = 1;
    topicStats2.msgThroughputIn = 2;
    topicStats2.msgRateOut = 3;
    topicStats2.msgThroughputOut = 4;
    topicStats2.averageMsgSize = 5;
    topicStats2.storageSize = 6;
    final PublisherStatsImpl publisherStats2 = new PublisherStatsImpl();
    publisherStats2.setSupportsPartialProducer(false);
    publisherStats2.setProducerName("name1");
    topicStats2.addPublisher(publisherStats2);
    topicStats2.subscriptions.put("test_ns", new SubscriptionStatsImpl());
    topicStats2.replication.put("test_ns", new ReplicatorStatsImpl());
    TopicStatsImpl target = new TopicStatsImpl();
    target.add(topicStats1);
    target.add(topicStats2);
    assertEquals(target.msgRateIn, 2.0);
    assertEquals(target.msgThroughputIn, 3.0);
    assertEquals(target.msgRateOut, 4.0);
    assertEquals(target.msgThroughputOut, 5.0);
    assertEquals(target.averageMsgSize, 3.0);
    assertEquals(target.storageSize, 7);
    assertEquals(target.getPublishers().size(), 1);
    assertEquals(target.subscriptions.size(), 1);
    assertEquals(target.replication.size(), 1);
}

Example 2 with ReplicatorStatsImpl

use of org.apache.pulsar.common.policies.data.stats.ReplicatorStatsImpl in project pulsar by apache.

the class PersistentTopicStatsTest method testPersistentTopicStatsAggregationByProducerName.

@Test
public void testPersistentTopicStatsAggregationByProducerName() {
    TopicStatsImpl topicStats1 = new TopicStatsImpl();
    topicStats1.msgRateIn = 1;
    topicStats1.msgThroughputIn = 1;
    topicStats1.msgRateOut = 1;
    topicStats1.msgThroughputOut = 1;
    topicStats1.averageMsgSize = 1;
    topicStats1.storageSize = 1;
    final PublisherStatsImpl publisherStats1 = new PublisherStatsImpl();
    publisherStats1.setSupportsPartialProducer(true);
    publisherStats1.msgRateIn = 1;
    publisherStats1.setProducerName("name1");
    topicStats1.addPublisher(publisherStats1);
    topicStats1.subscriptions.put("test_ns", new SubscriptionStatsImpl());
    topicStats1.replication.put("test_ns", new ReplicatorStatsImpl());
    TopicStatsImpl topicStats2 = new TopicStatsImpl();
    topicStats2.msgRateIn = 1;
    topicStats2.msgThroughputIn = 2;
    topicStats2.msgRateOut = 3;
    topicStats2.msgThroughputOut = 4;
    topicStats2.averageMsgSize = 5;
    topicStats2.storageSize = 6;
    final PublisherStatsImpl publisherStats2 = new PublisherStatsImpl();
    publisherStats2.setSupportsPartialProducer(true);
    publisherStats2.msgRateIn = 1;
    publisherStats2.setProducerName("name1");
    topicStats2.addPublisher(publisherStats2);
    topicStats2.subscriptions.put("test_ns", new SubscriptionStatsImpl());
    topicStats2.replication.put("test_ns", new ReplicatorStatsImpl());
    TopicStatsImpl topicStats3 = new TopicStatsImpl();
    topicStats3.msgRateIn = 0;
    topicStats3.msgThroughputIn = 0;
    topicStats3.msgRateOut = 0;
    topicStats3.msgThroughputOut = 0;
    topicStats3.averageMsgSize = 0;
    topicStats3.storageSize = 0;
    final PublisherStatsImpl publisherStats3 = new PublisherStatsImpl();
    publisherStats3.setSupportsPartialProducer(true);
    publisherStats3.msgRateIn = 1;
    publisherStats3.setProducerName("name2");
    topicStats3.addPublisher(publisherStats3);
    topicStats3.subscriptions.put("test_ns", new SubscriptionStatsImpl());
    topicStats3.replication.put("test_ns", new ReplicatorStatsImpl());
    TopicStatsImpl target = new TopicStatsImpl();
    target.add(topicStats1);
    target.add(topicStats2);
    target.add(topicStats3);
    assertEquals(target.msgRateIn, 2.0);
    assertEquals(target.msgThroughputIn, 3.0);
    assertEquals(target.msgRateOut, 4.0);
    assertEquals(target.msgThroughputOut, 5.0);
    assertEquals(target.averageMsgSize, 2.0);
    assertEquals(target.storageSize, 7);
    assertEquals(target.getPublishers().size(), 2);
    final Map<String, Double> expectedPublishersMap = Maps.newHashMap();
    expectedPublishersMap.put("name1", 2.0);
    expectedPublishersMap.put("name2", 1.0);
    assertEquals(target.getPublishers().stream().collect(Collectors.toMap(PublisherStats::getProducerName, e -> ((PublisherStatsImpl) e).msgRateIn)), expectedPublishersMap);
    assertEquals(target.subscriptions.size(), 1);
    assertEquals(target.replication.size(), 1);
}

Example 3 with ReplicatorStatsImpl

use of org.apache.pulsar.common.policies.data.stats.ReplicatorStatsImpl in project pulsar by yahoo.

the class PersistentTopic method updateRates.

@Override
public void updateRates(NamespaceStats nsStats, NamespaceBundleStats bundleStats, StatsOutputStream topicStatsStream, ClusterReplicationMetrics replStats, String namespace, boolean hydratePublishers) {
    this.publishRateLimitedTimes = 0;
    TopicStatsHelper topicStatsHelper = threadLocalTopicStats.get();
    topicStatsHelper.reset();
    replicators.forEach((region, replicator) -> replicator.updateRates());
    nsStats.producerCount += producers.size();
    bundleStats.producerCount += producers.size();
    topicStatsStream.startObject(topic);
    // start publisher stats
    topicStatsStream.startList("publishers");
    producers.values().forEach(producer -> {
        producer.updateRates();
        PublisherStatsImpl publisherStats = producer.getStats();
        topicStatsHelper.aggMsgRateIn += publisherStats.msgRateIn;
        topicStatsHelper.aggMsgThroughputIn += publisherStats.msgThroughputIn;
        if (producer.isRemote()) {
            topicStatsHelper.remotePublishersStats.put(producer.getRemoteCluster(), publisherStats);
        }
        // Populate consumer specific stats here
        if (hydratePublishers) {
            StreamingStats.writePublisherStats(topicStatsStream, publisherStats);
        }
    });
    topicStatsStream.endList();
    // if publish-rate increases (eg: 0 to 1K) then pick max publish-rate and if publish-rate decreases then keep
    // average rate.
    lastUpdatedAvgPublishRateInMsg = topicStatsHelper.aggMsgRateIn > lastUpdatedAvgPublishRateInMsg ? topicStatsHelper.aggMsgRateIn : (topicStatsHelper.aggMsgRateIn + lastUpdatedAvgPublishRateInMsg) / 2;
    lastUpdatedAvgPublishRateInByte = topicStatsHelper.aggMsgThroughputIn > lastUpdatedAvgPublishRateInByte ? topicStatsHelper.aggMsgThroughputIn : (topicStatsHelper.aggMsgThroughputIn + lastUpdatedAvgPublishRateInByte) / 2;
    // Start replicator stats
    topicStatsStream.startObject("replication");
    nsStats.replicatorCount += topicStatsHelper.remotePublishersStats.size();
    replicators.forEach((cluster, replicator) -> {
        // Update replicator cursor state
        try {
            ((PersistentReplicator) replicator).updateCursorState();
        } catch (Exception e) {
            log.warn("[{}] Failed to update cursor state ", topic, e);
        }
        // Update replicator stats
        ReplicatorStatsImpl rStat = replicator.getStats();
        // Add incoming msg rates
        PublisherStatsImpl pubStats = topicStatsHelper.remotePublishersStats.get(replicator.getRemoteCluster());
        rStat.msgRateIn = pubStats != null ? pubStats.msgRateIn : 0;
        rStat.msgThroughputIn = pubStats != null ? pubStats.msgThroughputIn : 0;
        rStat.inboundConnection = pubStats != null ? pubStats.getAddress() : null;
        rStat.inboundConnectedSince = pubStats != null ? pubStats.getConnectedSince() : null;
        topicStatsHelper.aggMsgRateOut += rStat.msgRateOut;
        topicStatsHelper.aggMsgThroughputOut += rStat.msgThroughputOut;
        // Populate replicator specific stats here
        topicStatsStream.startObject(cluster);
        topicStatsStream.writePair("connected", rStat.connected);
        topicStatsStream.writePair("msgRateExpired", rStat.msgRateExpired);
        topicStatsStream.writePair("msgRateIn", rStat.msgRateIn);
        topicStatsStream.writePair("msgRateOut", rStat.msgRateOut);
        topicStatsStream.writePair("msgThroughputIn", rStat.msgThroughputIn);
        topicStatsStream.writePair("msgThroughputOut", rStat.msgThroughputOut);
        topicStatsStream.writePair("replicationBacklog", rStat.replicationBacklog);
        topicStatsStream.writePair("replicationDelayInSeconds", rStat.replicationDelayInSeconds);
        topicStatsStream.writePair("inboundConnection", rStat.inboundConnection);
        topicStatsStream.writePair("inboundConnectedSince", rStat.inboundConnectedSince);
        topicStatsStream.writePair("outboundConnection", rStat.outboundConnection);
        topicStatsStream.writePair("outboundConnectedSince", rStat.outboundConnectedSince);
        topicStatsStream.endObject();
        nsStats.msgReplBacklog += rStat.replicationBacklog;
        if (replStats.isMetricsEnabled()) {
            String namespaceClusterKey = replStats.getKeyName(namespace, cluster);
            ReplicationMetrics replicationMetrics = replStats.get(namespaceClusterKey);
            boolean update = false;
            if (replicationMetrics == null) {
                replicationMetrics = ReplicationMetrics.get();
                update = true;
            }
            replicationMetrics.connected += rStat.connected ? 1 : 0;
            replicationMetrics.msgRateOut += rStat.msgRateOut;
            replicationMetrics.msgThroughputOut += rStat.msgThroughputOut;
            replicationMetrics.msgReplBacklog += rStat.replicationBacklog;
            if (update) {
                replStats.put(namespaceClusterKey, replicationMetrics);
            }
            // replication delay for a namespace is the max repl-delay among all the topics under this namespace
            if (rStat.replicationDelayInSeconds > replicationMetrics.maxMsgReplDelayInSeconds) {
                replicationMetrics.maxMsgReplDelayInSeconds = rStat.replicationDelayInSeconds;
            }
        }
    });
    // Close replication
    topicStatsStream.endObject();
    // Start subscription stats
    topicStatsStream.startObject("subscriptions");
    nsStats.subsCount += subscriptions.size();
    subscriptions.forEach((subscriptionName, subscription) -> {
        double subMsgRateOut = 0;
        double subMsgThroughputOut = 0;
        double subMsgRateRedeliver = 0;
        // Start subscription name & consumers
        try {
            topicStatsStream.startObject(subscriptionName);
            topicStatsStream.startList("consumers");
            for (Consumer consumer : subscription.getConsumers()) {
                ++nsStats.consumerCount;
                ++bundleStats.consumerCount;
                consumer.updateRates();
                ConsumerStatsImpl consumerStats = consumer.getStats();
                subMsgRateOut += consumerStats.msgRateOut;
                subMsgThroughputOut += consumerStats.msgThroughputOut;
                subMsgRateRedeliver += consumerStats.msgRateRedeliver;
                StreamingStats.writeConsumerStats(topicStatsStream, subscription.getType(), consumerStats);
            }
            // Close Consumer stats
            topicStatsStream.endList();
            // Populate subscription specific stats here
            topicStatsStream.writePair("msgBacklog", subscription.getNumberOfEntriesInBacklog(true));
            topicStatsStream.writePair("msgRateExpired", subscription.getExpiredMessageRate());
            topicStatsStream.writePair("msgRateOut", subMsgRateOut);
            topicStatsStream.writePair("msgThroughputOut", subMsgThroughputOut);
            topicStatsStream.writePair("msgRateRedeliver", subMsgRateRedeliver);
            topicStatsStream.writePair("numberOfEntriesSinceFirstNotAckedMessage", subscription.getNumberOfEntriesSinceFirstNotAckedMessage());
            topicStatsStream.writePair("totalNonContiguousDeletedMessagesRange", subscription.getTotalNonContiguousDeletedMessagesRange());
            topicStatsStream.writePair("type", subscription.getTypeString());
            if (Subscription.isIndividualAckMode(subscription.getType())) {
                if (subscription.getDispatcher() instanceof PersistentDispatcherMultipleConsumers) {
                    PersistentDispatcherMultipleConsumers dispatcher = (PersistentDispatcherMultipleConsumers) subscription.getDispatcher();
                    topicStatsStream.writePair("blockedSubscriptionOnUnackedMsgs", dispatcher.isBlockedDispatcherOnUnackedMsgs());
                    topicStatsStream.writePair("unackedMessages", dispatcher.getTotalUnackedMessages());
                }
            }
            // Close consumers
            topicStatsStream.endObject();
            topicStatsHelper.aggMsgRateOut += subMsgRateOut;
            topicStatsHelper.aggMsgThroughputOut += subMsgThroughputOut;
            nsStats.msgBacklog += subscription.getNumberOfEntriesInBacklog(false);
            // check stuck subscription
            if (brokerService.getPulsar().getConfig().isUnblockStuckSubscriptionEnabled()) {
                subscription.checkAndUnblockIfStuck();
            }
        } catch (Exception e) {
            log.error("Got exception when creating consumer stats for subscription {}: {}", subscriptionName, e.getMessage(), e);
        }
    });
    // Close subscription
    topicStatsStream.endObject();
    // Remaining dest stats.
    topicStatsHelper.averageMsgSize = topicStatsHelper.aggMsgRateIn == 0.0 ? 0.0 : (topicStatsHelper.aggMsgThroughputIn / topicStatsHelper.aggMsgRateIn);
    topicStatsStream.writePair("producerCount", producers.size());
    topicStatsStream.writePair("averageMsgSize", topicStatsHelper.averageMsgSize);
    topicStatsStream.writePair("msgRateIn", topicStatsHelper.aggMsgRateIn);
    topicStatsStream.writePair("msgRateOut", topicStatsHelper.aggMsgRateOut);
    topicStatsStream.writePair("msgInCount", getMsgInCounter());
    topicStatsStream.writePair("bytesInCount", getBytesInCounter());
    topicStatsStream.writePair("msgOutCount", getMsgOutCounter());
    topicStatsStream.writePair("bytesOutCount", getBytesOutCounter());
    topicStatsStream.writePair("msgThroughputIn", topicStatsHelper.aggMsgThroughputIn);
    topicStatsStream.writePair("msgThroughputOut", topicStatsHelper.aggMsgThroughputOut);
    topicStatsStream.writePair("storageSize", ledger.getTotalSize());
    topicStatsStream.writePair("backlogSize", ledger.getEstimatedBacklogSize());
    topicStatsStream.writePair("pendingAddEntriesCount", ((ManagedLedgerImpl) ledger).getPendingAddEntriesCount());
    topicStatsStream.writePair("filteredEntriesCount", getFilteredEntriesCount());
    nsStats.msgRateIn += topicStatsHelper.aggMsgRateIn;
    nsStats.msgRateOut += topicStatsHelper.aggMsgRateOut;
    nsStats.msgThroughputIn += topicStatsHelper.aggMsgThroughputIn;
    nsStats.msgThroughputOut += topicStatsHelper.aggMsgThroughputOut;
    nsStats.storageSize += ledger.getEstimatedBacklogSize();
    bundleStats.msgRateIn += topicStatsHelper.aggMsgRateIn;
    bundleStats.msgRateOut += topicStatsHelper.aggMsgRateOut;
    bundleStats.msgThroughputIn += topicStatsHelper.aggMsgThroughputIn;
    bundleStats.msgThroughputOut += topicStatsHelper.aggMsgThroughputOut;
    bundleStats.cacheSize += ((ManagedLedgerImpl) ledger).getCacheSize();
    // Close topic object
    topicStatsStream.endObject();
    // add publish-latency metrics
    this.addEntryLatencyStatsUsec.refresh();
    NamespaceStats.add(this.addEntryLatencyStatsUsec.getBuckets(), nsStats.addLatencyBucket);
    this.addEntryLatencyStatsUsec.reset();
}

Example 4 with ReplicatorStatsImpl

use of org.apache.pulsar.common.policies.data.stats.ReplicatorStatsImpl in project pulsar by yahoo.

the class PersistentTopic method asyncGetStats.

@Override
public CompletableFuture<TopicStatsImpl> asyncGetStats(boolean getPreciseBacklog, boolean subscriptionBacklogSize, boolean getEarliestTimeInBacklog) {
    CompletableFuture<TopicStatsImpl> statsFuture = new CompletableFuture<>();
    TopicStatsImpl stats = new TopicStatsImpl();
    ObjectObjectHashMap<String, PublisherStatsImpl> remotePublishersStats = new ObjectObjectHashMap<>();
    producers.values().forEach(producer -> {
        PublisherStatsImpl publisherStats = producer.getStats();
        stats.msgRateIn += publisherStats.msgRateIn;
        stats.msgThroughputIn += publisherStats.msgThroughputIn;
        if (producer.isRemote()) {
            remotePublishersStats.put(producer.getRemoteCluster(), publisherStats);
        } else {
            stats.addPublisher(publisherStats);
        }
    });
    stats.averageMsgSize = stats.msgRateIn == 0.0 ? 0.0 : (stats.msgThroughputIn / stats.msgRateIn);
    stats.msgInCounter = getMsgInCounter();
    stats.bytesInCounter = getBytesInCounter();
    stats.msgChunkPublished = this.msgChunkPublished;
    stats.waitingPublishers = getWaitingProducersCount();
    stats.bytesOutCounter = bytesOutFromRemovedSubscriptions.longValue();
    stats.msgOutCounter = msgOutFromRemovedSubscriptions.longValue();
    stats.publishRateLimitedTimes = publishRateLimitedTimes;
    subscriptions.forEach((name, subscription) -> {
        SubscriptionStatsImpl subStats = subscription.getStats(getPreciseBacklog, subscriptionBacklogSize, getEarliestTimeInBacklog);
        stats.msgRateOut += subStats.msgRateOut;
        stats.msgThroughputOut += subStats.msgThroughputOut;
        stats.bytesOutCounter += subStats.bytesOutCounter;
        stats.msgOutCounter += subStats.msgOutCounter;
        stats.subscriptions.put(name, subStats);
        stats.nonContiguousDeletedMessagesRanges += subStats.nonContiguousDeletedMessagesRanges;
        stats.nonContiguousDeletedMessagesRangesSerializedSize += subStats.nonContiguousDeletedMessagesRangesSerializedSize;
    });
    replicators.forEach((cluster, replicator) -> {
        ReplicatorStatsImpl replicatorStats = replicator.getStats();
        // Add incoming msg rates
        PublisherStatsImpl pubStats = remotePublishersStats.get(replicator.getRemoteCluster());
        if (pubStats != null) {
            replicatorStats.msgRateIn = pubStats.msgRateIn;
            replicatorStats.msgThroughputIn = pubStats.msgThroughputIn;
            replicatorStats.inboundConnection = pubStats.getAddress();
            replicatorStats.inboundConnectedSince = pubStats.getConnectedSince();
        }
        stats.msgRateOut += replicatorStats.msgRateOut;
        stats.msgThroughputOut += replicatorStats.msgThroughputOut;
        stats.replication.put(replicator.getRemoteCluster(), replicatorStats);
    });
    stats.storageSize = ledger.getTotalSize();
    stats.backlogSize = ledger.getEstimatedBacklogSize();
    stats.deduplicationStatus = messageDeduplication.getStatus().toString();
    stats.topicEpoch = topicEpoch.orElse(null);
    stats.offloadedStorageSize = ledger.getOffloadedSize();
    stats.lastOffloadLedgerId = ledger.getLastOffloadedLedgerId();
    stats.lastOffloadSuccessTimeStamp = ledger.getLastOffloadedSuccessTimestamp();
    stats.lastOffloadFailureTimeStamp = ledger.getLastOffloadedFailureTimestamp();
    Optional<CompactorMXBean> mxBean = getCompactorMXBean();
    stats.compaction.reset();
    mxBean.flatMap(bean -> bean.getCompactionRecordForTopic(topic)).map(compactionRecord -> {
        stats.compaction.lastCompactionRemovedEventCount = compactionRecord.getLastCompactionRemovedEventCount();
        stats.compaction.lastCompactionSucceedTimestamp = compactionRecord.getLastCompactionSucceedTimestamp();
        stats.compaction.lastCompactionFailedTimestamp = compactionRecord.getLastCompactionFailedTimestamp();
        stats.compaction.lastCompactionDurationTimeInMills = compactionRecord.getLastCompactionDurationTimeInMills();
        return compactionRecord;
    });
    if (getEarliestTimeInBacklog && stats.backlogSize != 0) {
        ledger.getEarliestMessagePublishTimeInBacklog().whenComplete((earliestTime, e) -> {
            if (e != null) {
                log.error("[{}] Failed to get earliest message publish time in backlog", topic, e);
                statsFuture.completeExceptionally(e);
            } else {
                stats.earliestMsgPublishTimeInBacklogs = earliestTime;
                statsFuture.complete(stats);
            }
        });
    } else {
        statsFuture.complete(stats);
    }
    return statsFuture;
}

Example 5 with ReplicatorStatsImpl

use of org.apache.pulsar.common.policies.data.stats.ReplicatorStatsImpl in project pulsar by yahoo.

the class ReplicatorStatsTest method testReplicatorStatsNull.

@Test
public void testReplicatorStatsNull() {
    ReplicatorStatsImpl rs = new ReplicatorStatsImpl();
    try {
        rs.add(null);
        Assert.fail("Must fail.");
    } catch (NullPointerException ne) {
    // ok
    }
}