import com.google.common.cache.CacheBuilder
import freenet.keys.FreenetURI
import java.io.ByteArrayInputStream
import com.google.common.cache.CacheBuilder
import freenet.keys.FreenetURI
import java.io.ByteArrayInputStream
import java.util.concurrent.TimeUnit.MINUTES
import javax.imageio.ImageIO
import javax.inject.Inject
import java.util.concurrent.TimeUnit.MINUTES
import javax.imageio.ImageIO
import javax.inject.Inject
-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<String, Boolean>()
private val failureCache = CacheBuilder.newBuilder().ticker(ticker).expireAfterWrite(30, MINUTES).build<String, Boolean>()
private val imageCache = CacheBuilder.newBuilder().build<String, LinkedElement>()
private val callback = object : FreenetInterface.BackgroundFetchCallback {
private val loadingLinks = CacheBuilder.newBuilder().build<String, Boolean>()
private val failureCache = CacheBuilder.newBuilder().ticker(ticker).expireAfterWrite(30, MINUTES).build<String, Boolean>()
private val imageCache = CacheBuilder.newBuilder().build<String, LinkedElement>()
private val callback = object : FreenetInterface.BackgroundFetchCallback {
- override fun cancelForMimeType(uri: FreenetURI, mimeType: String): Boolean {
- return !mimeType.startsWith("image/")
+ override fun shouldCancel(uri: FreenetURI, mimeType: String, size: Long): Boolean {
+ return !mimeType.startsWith("image/") || (size > 2097152)
- if (loadingLinks.getIfPresent(link) == null) {
- loadingLinks.put(link, true)
+ if (loadingLinks.getIfPresent(normalizedLink) == null) {
+ loadingLinks.put(normalizedLink, true)
freenetInterface.startFetch(FreenetURI(link), callback)
}
}
return LinkedElement(link, loading = true)
}
freenetInterface.startFetch(FreenetURI(link), callback)
}
}
return LinkedElement(link, loading = true)
}