-
Notifications
You must be signed in to change notification settings - Fork 25
/
Copy path359. Logger Rate Limiter.java
executable file
·113 lines (89 loc) · 3.2 KB
/
359. Logger Rate Limiter.java
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
E
tags: Hash Table, Design
time: O(1)
space: O(n)
#### HashMap<Message, Timestamp>
- map: avoid duplicate message, records timestamp for validation
- time: O(1)
- space: O(n)
#### Queue + Set
- 1) keep a trimmed queue and set (all tasks to be within 10 secs);
- 2) use set to O(1) check if incoming message exists.
- time: O(x), trimQueue()
- space: O(n)
```
/*
Design a logger system that receive stream of messages along with its timestamps, each message should be printed if and only if it is not printed in the last 10 seconds.
Given a message and a timestamp (in seconds granularity), return true if the message should be printed in the given timestamp, otherwise returns false.
It is possible that several messages arrive roughly at the same time.
Example:
Logger logger = new Logger();
// logging string "foo" at timestamp 1
logger.shouldPrintMessage(1, "foo"); returns true;
// logging string "bar" at timestamp 2
logger.shouldPrintMessage(2,"bar"); returns true;
// logging string "foo" at timestamp 3
logger.shouldPrintMessage(3,"foo"); returns false;
// logging string "bar" at timestamp 8
logger.shouldPrintMessage(8,"bar"); returns false;
// logging string "foo" at timestamp 10
logger.shouldPrintMessage(10,"foo"); returns false;
// logging string "foo" at timestamp 11
logger.shouldPrintMessage(11,"foo"); returns true;
*/
class Logger {
Map<String, Integer> map;
/** Initialize your data structure here. */
public Logger() {
map = new HashMap();
}
/** 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) {
if (!map.containsKey(message)) {
map.put(message, timestamp);
return true;
} else if (timestamp - map.get(message) >= 10) {
map.put(message, timestamp);
return true;
}
return false;
}
}
#### Queue + Set
class Logger {
Queue<Pair<String, Integer>> queue;
Set<String> set;
/** Initialize your data structure here. */
public Logger() {
set = new HashSet();
queue = new LinkedList();
}
/** 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) {
trimQueue(timestamp);
if (!set.contains(message)) {
set.add(message);
queue.offer(new Pair<>(message, timestamp));
return true;
}
return false;
}
private void trimQueue(int timestamp) {
while (!queue.isEmpty()) {
Pair<String, Integer> task = queue.peek();
if (timestamp - task.getValue() < 10) break;
queue.poll();
set.remove(task.getKey());
}
}
}
/**
* Your Logger object will be instantiated and called as such:
* Logger obj = new Logger();
* boolean param_1 = obj.shouldPrintMessage(timestamp,message);
*/
```