com.johnsnowlabs.util.FileHelper.scala Maven / Gradle / Ivy
/*
* Copyright 2017-2022 John Snow Labs
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.johnsnowlabs.util
import org.apache.commons.io.FileUtils
import java.io.{File, IOException}
import java.nio.charset.Charset
import java.nio.file.{Files, Paths}
import java.security.MessageDigest
import java.text.DecimalFormat
object FileHelper {
def writeLines(file: String, lines: Seq[String], encoding: String = "UTF-8"): Unit = {
val writer = Files.newBufferedWriter(Paths.get(file), Charset.forName(encoding))
try {
var cnt = 0
for (line <- lines) {
writer.write(line)
if (cnt > 0)
writer.write(System.lineSeparator())
cnt += 1
}
} catch {
case ex: IOException =>
ex.printStackTrace()
} finally if (writer != null) writer.close()
}
def delete(file: String, throwOnError: Boolean = false): Unit = {
val f = new File(file)
if (f.exists()) {
try {
if (f.isDirectory)
FileUtils.deleteDirectory(f)
else
FileUtils.deleteQuietly(f)
} catch {
case e: Exception =>
if (throwOnError)
throw e
else
FileUtils.forceDeleteOnExit(f)
}
}
}
def generateChecksum(path: String): String = {
val arr = Files readAllBytes (Paths get path)
val checksum = MessageDigest.getInstance("MD5") digest arr
checksum.map("%02X" format _).mkString
}
def getHumanReadableFileSize(size: Long): String = {
if (size <= 0) return "0"
val units = Array[String]("B", "KB", "MB", "GB", "TB", "PB", "EB")
val digitGroups = (Math.log10(size) / Math.log10(1024)).toInt
new DecimalFormat("#,##0.#").format(size / Math.pow(1024, digitGroups)) + " " + units(
digitGroups)
}
}