|
@@ -64,7 +64,7 @@ func (s *URLStore) Set(key, url *string) error {
|
|
|
return nil
|
|
return nil
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-func (s *URLStore) Count() int {
|
|
|
|
|
|
|
+func (s *URLStore) count() int {
|
|
|
s.mu.RLock()
|
|
s.mu.RLock()
|
|
|
defer s.mu.RUnlock()
|
|
defer s.mu.RUnlock()
|
|
|
return len(s.urls)
|
|
return len(s.urls)
|
|
@@ -72,7 +72,7 @@ func (s *URLStore) Count() int {
|
|
|
|
|
|
|
|
func (s *URLStore) Put(url, key *string) error {
|
|
func (s *URLStore) Put(url, key *string) error {
|
|
|
for {
|
|
for {
|
|
|
- *key = genKey(s.Count())
|
|
|
|
|
|
|
+ *key = genKey(s.count())
|
|
|
if err := s.Set(key, url); err == nil {
|
|
if err := s.Set(key, url); err == nil {
|
|
|
break
|
|
break
|
|
|
}
|
|
}
|