FilesExpand file tree

 master

/

LoggerRateLimiter.java

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

55 lines (49 loc) · 1.66 KB

 master

/

LoggerRateLimiter.java

Copy path

Top

File metadata and controls

• Code
• Blame

55 lines (49 loc) · 1.66 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

import java.util.Comparator;

import java.util.HashMap;

import java.util.PriorityQueue;

import java.util.Queue;

/**

* 如果要考虑到空间消耗，则要另外定义一个List<String>，每次新来一条日志，都要遍历

* list中10s外的poll了，同时从map中remove

*

*/

public class LoggerRateLimiter {

private HashMap<String, Integer> mMap;

private Queue<String> mQueue;

/** Initialize your data structure here. */

public LoggerRateLimiter() {

mMap = new HashMap<>();

mQueue = new PriorityQueue<>(new Comparator<String>() {

@Override

public int compare(String o1, String o2) {

int time1 = mMap.getOrDefault(o1, -100);

int time2 = mMap.getOrDefault(o2, -100);

return time1 - time2;

}

});

}

private void clearExpired(int timestamp) {

while (!mQueue.isEmpty()) {

String msg = mQueue.peek();

if (timestamp - mMap.get(msg) >= 10) {

mQueue.poll();

mMap.remove(msg);

} else {

break;

}

}

}

/** Returns true if the message should be printed in the given timestamp, otherwise returns false.

If this method returns false, the message will not be printed.

The timestamp is in seconds granularity. */

public boolean shouldPrintMessage(int timestamp, String message) {

int time = mMap.getOrDefault(message, -100);

boolean ret = false;

if (timestamp - time >= 10) {

mMap.put(message, timestamp);

ret = true;

}

clearExpired(timestamp);

return ret;

}

}