org.jetbrains.kotlin.konan.file.ZipFileSystemAccessor.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kotlin-compiler-embeddable Show documentation

the Kotlin compiler embeddable

There is a newer version: 2.1.0-RC

Show newest version

/*
 * Copyright 2010-2023 JetBrains s.r.o. and Kotlin Programming Language contributors.
 * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
 */

package org.jetbrains.kotlin.konan.file

import java.nio.file.FileSystem

interface ZipFileSystemAccessor {
    fun  withZipFileSystem(zipFile: File, action: (FileSystem) -> T): T
}

object ZipFileSystemInPlaceAccessor : ZipFileSystemAccessor {
    override fun  withZipFileSystem(zipFile: File, action: (FileSystem) -> T): T {
        return zipFile.withZipFileSystem(action)
    }
}

class ZipFileSystemCacheableAccessor(private val cacheLimit: Int) : ZipFileSystemAccessor {
    private val loadFactor = 0.75f
    private val initialCapacity = (1f + cacheLimit.toFloat() / loadFactor).toInt()

    private val openedFileSystems = object : LinkedHashMap(initialCapacity, loadFactor, true) {
        override fun removeEldestEntry(eldest: Map.Entry?): Boolean {
            if (size > cacheLimit) {
                eldest?.value?.close()
                return true
            }
            return false
        }
    }

    override fun  withZipFileSystem(zipFile: File, action: (FileSystem) -> T): T {
        val fileSystem = openedFileSystems.getOrPut(zipFile) { zipFile.zipFileSystem() }
        return action(fileSystem)
    }

    fun reset() {
        var lastException: Exception? = null
        for (fileSystem in openedFileSystems.values) {
            try {
                fileSystem.close()
            } catch (e: Exception) {
                lastException = e
            }
        }
        openedFileSystems.clear()

        lastException?.let {
            throw it
        }
    }
}