1 files changed, 3 insertions, 8 deletions

diff --git a/cache.go b/cache.go
index 73b4d58..2eeb416 100644
--- a/cache.go
+++ b/cache.go
@@ -80,18 +80,13 @@ func (cache *Cache) Save() (err error) {
 	return nil
 }
 
-func (cache *Cache) Set(url string, respCode int, size int64) {
+func (cache *Cache) Set(link Link) {
 	cache.mtx.Lock()
 	defer cache.mtx.Unlock()
 
-	var scannedLink = cache.ScannedLinks[url]
+	var scannedLink = cache.ScannedLinks[link.Url]
 	if scannedLink != nil {
 		return
 	}
-	scannedLink = &Link{
-		Url:          url,
-		Size:         size,
-		ResponseCode: respCode,
-	}
-	cache.ScannedLinks[url] = scannedLink
+	cache.ScannedLinks[link.Url] = &link
 }