aboutsummaryrefslogtreecommitdiffstats
path: root/zkfacade/src/main/java/com/yahoo/vespa/curator/stats/LockStats.java
blob: 7765b0b3e92c3d9e2f6f5816d7ee716c1f310303 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.curator.stats;

import java.util.Comparator;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.PriorityQueue;
import java.util.concurrent.ConcurrentHashMap;
import java.util.logging.Logger;

/**
 * This class manages statistics related to lock attempts on {@link com.yahoo.vespa.curator.Lock}.
 *
 * @author hakon
 */
public class LockStats {
    private static final Logger logger = Logger.getLogger(LockStats.class.getName());

    // No 'volatile' is needed because field is only ever changed for testing which is single-threaded.
    private static LockStats stats = new LockStats();

    private final ConcurrentHashMap<Thread, ThreadLockStats> statsByThread = new ConcurrentHashMap<>();

    /** Modified only by Thread actually holding the lock on the path (key). */
    private final ConcurrentHashMap<String, Thread> lockPathsHeld = new ConcurrentHashMap<>();

    private final LockAttemptSamples completedLockAttemptSamples = new LockAttemptSamples(3);

    // Keep recordings in a priority queue, with the smallest element having the smallest duration.
    // Recordings can be large, so keep the number of recordings low.
    private static final int MAX_RECORDINGS = 3;
    private final Object interestingRecordingsMonitor = new Object();
    private final PriorityQueue<RecordedLockAttempts> interestingRecordings =
            new PriorityQueue<>(MAX_RECORDINGS, Comparator.comparing(RecordedLockAttempts::duration));

    private final ConcurrentHashMap<String, LockMetrics> metricsByLockPath = new ConcurrentHashMap<>();

    /** Returns global stats. */
    public static LockStats getGlobal() { return stats; }

    /** Returns stats tied to the current thread. */
    public static ThreadLockStats getForCurrentThread() { return stats.getForThread(Thread.currentThread()); }

    public static void clearForTesting() {
        stats = new LockStats();
    }

    private LockStats() {}

    public Map<String, LockMetrics> getLockMetricsByPath() { return Map.copyOf(metricsByLockPath); }
    public List<ThreadLockStats> getThreadLockStats() { return List.copyOf(statsByThread.values()); }
    public List<LockAttempt> getLockAttemptSamples() { return completedLockAttemptSamples.asList(); }

    public List<RecordedLockAttempts> getHistoricRecordings() {
        synchronized (interestingRecordingsMonitor) {
            return List.copyOf(interestingRecordings);
        }
    }

    /** Non-private for testing. */
    ThreadLockStats getForThread(Thread thread) {
        return statsByThread.computeIfAbsent(thread, ThreadLockStats::new);
    }

    /** Must be invoked only after the first and non-reentry acquisition of the lock. */
    void notifyOfThreadHoldingLock(Thread currentThread, String lockPath) {
        Thread oldThread = lockPathsHeld.put(lockPath, currentThread);
        if (oldThread != null) {
            getLockMetrics(lockPath).incrementAcquireWithoutReleaseCount();
            logger.warning("Thread " + currentThread.getName() + " reports it has the lock on " +
                           lockPath + ", but thread " + oldThread.getName() +
                           " has not reported it released the lock");
        }
    }

    /** Must be invoked only before the last and non-reentry release of the lock. */
    void notifyOfThreadReleasingLock(Thread currentThread, String lockPath) {
        Thread oldThread = lockPathsHeld.remove(lockPath);
        if (oldThread == null) {
            getLockMetrics(lockPath).incrementNakedReleaseCount();
            logger.warning("Thread " + currentThread.getName() + " is releasing the lock " + lockPath +
                           ", but nobody owns that lock");
        } else if (oldThread != currentThread) {
            getLockMetrics(lockPath).incrementForeignReleaseCount();
            logger.warning("Thread " + currentThread.getName() +
                           " is releasing the lock " + lockPath + ", but it was owned by thread " +
                           oldThread.getName());
        }
    }

    /**
     * Returns the ThreadLockStats holding the lock on the path, but the info may be outdated already
     * on return, either no-one holds the lock or another thread may hold the lock.
     */
    Optional<ThreadLockStats> getThreadLockStatsHolding(String lockPath) {
        return Optional.ofNullable(lockPathsHeld.get(lockPath)).map(statsByThread::get);
    }

    LockMetrics getLockMetrics(String lockPath) {
        return metricsByLockPath.computeIfAbsent(lockPath, __ -> new LockMetrics());
    }

    void maybeSample(LockAttempt lockAttempt) {
        completedLockAttemptSamples.maybeSample(lockAttempt);
    }

    void reportNewStoppedRecording(RecordedLockAttempts recording) {
        synchronized (interestingRecordings) {
            if (interestingRecordings.size() < MAX_RECORDINGS) {
                interestingRecordings.add(recording);
            } else if (recording.duration().compareTo(interestingRecordings.peek().duration()) > 0) {
                // peek() retrieves the smallest element according to the PriorityQueue's
                // comparator.

                interestingRecordings.poll();
                interestingRecordings.add(recording);
            }
        }

    }
}