X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FDefaultElementLoader.kt;h=3f33efe52711ba5bb3c857c3e566bd46cc821cf4;hb=7a70c5564da777276295d6ec29ab76446425bb36;hp=98a2caa7f2e78a31b34f5a757600aa3309eeb377;hpb=b91ce252d94a8876097b939e129dc33264cef2f5;p=Sone.git diff --git a/src/main/kotlin/net/pterodactylus/sone/core/DefaultElementLoader.kt b/src/main/kotlin/net/pterodactylus/sone/core/DefaultElementLoader.kt index 98a2caa..3f33efe 100644 --- a/src/main/kotlin/net/pterodactylus/sone/core/DefaultElementLoader.kt +++ b/src/main/kotlin/net/pterodactylus/sone/core/DefaultElementLoader.kt @@ -11,34 +11,41 @@ import javax.inject.Inject /** * [ElementLoader] implementation that uses a simple Guava [com.google.common.cache.Cache]. */ -class DefaultElementLoader @Inject constructor(private val freenetInterface: FreenetInterface, ticker: Ticker = Ticker.systemTicker()) : ElementLoader { +class DefaultElementLoader(private val freenetInterface: FreenetInterface, ticker: Ticker) : ElementLoader { + + @Inject constructor(freenetInterface: FreenetInterface) : this(freenetInterface, Ticker.systemTicker()) private val loadingLinks = CacheBuilder.newBuilder().build() private val failureCache = CacheBuilder.newBuilder().ticker(ticker).expireAfterWrite(30, MINUTES).build() private val imageCache = CacheBuilder.newBuilder().build() private val callback = object : FreenetInterface.BackgroundFetchCallback { override fun cancelForMimeType(uri: FreenetURI, mimeType: String): Boolean { + println("Got MIME Type “$mimeType” for $uri.") return !mimeType.startsWith("image/") } override fun loaded(uri: FreenetURI, mimeType: String, data: ByteArray) { + println("Got ${data.size} Bytes with MIME Type $mimeType for $uri.") if (!mimeType.startsWith("image/")) { return } ByteArrayInputStream(data).use { ImageIO.read(it) }?.let { + println("Successfully parsed images from $uri.") imageCache.get(uri.toString()) { LinkedElement(uri.toString()) } } removeLoadingLink(uri) } override fun failed(uri: FreenetURI) { + println("Failed to load $uri.") failureCache.put(uri.toString(), true) removeLoadingLink(uri) } private fun removeLoadingLink(uri: FreenetURI) { + println("Not loading anymore: $uri.") synchronized(loadingLinks) { loadingLinks.invalidate(uri.toString()) } @@ -46,18 +53,23 @@ class DefaultElementLoader @Inject constructor(private val freenetInterface: Fre } override fun loadElement(link: String): LinkedElement { + println("Checking for $link...") synchronized(loadingLinks) { imageCache.getIfPresent(link)?.run { + println("In the Image Cache: $link") return this } failureCache.getIfPresent(link)?.run { + println("In the Failure Cache: $link") return LinkedElement(link, failed = true) } if (loadingLinks.getIfPresent(link) == null) { + println("Not loading: $link") loadingLinks.put(link, true) freenetInterface.startFetch(FreenetURI(link), callback) } } + println("Returning loading element: $link") return LinkedElement(link, loading = true) }