diff options
author | vlad <vlad@driver.xyz> | 2017-06-13 16:12:20 -0700 |
---|---|---|
committer | vlad <vlad@driver.xyz> | 2017-06-13 16:12:20 -0700 |
commit | cd1b635b2ae90d9ac2d8b1779183a1fbd8c5fd5c (patch) | |
tree | 062e8dad1a1513e26b0fd08b1742d6ff2ee874f7 /src/main/scala/xyz/driver/pdsuicommon/resources | |
parent | 0000a65ab4479a2a40e2d6468036438e9705b4aa (diff) | |
download | rest-query-cd1b635b2ae90d9ac2d8b1779183a1fbd8c5fd5c.tar.gz rest-query-cd1b635b2ae90d9ac2d8b1779183a1fbd8c5fd5c.tar.bz2 rest-query-cd1b635b2ae90d9ac2d8b1779183a1fbd8c5fd5c.zip |
Adding domain entitiesv0.1.0
Diffstat (limited to 'src/main/scala/xyz/driver/pdsuicommon/resources')
-rw-r--r-- | src/main/scala/xyz/driver/pdsuicommon/resources/ResourcesStorage.scala | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/src/main/scala/xyz/driver/pdsuicommon/resources/ResourcesStorage.scala b/src/main/scala/xyz/driver/pdsuicommon/resources/ResourcesStorage.scala new file mode 100644 index 0000000..2be7ed7 --- /dev/null +++ b/src/main/scala/xyz/driver/pdsuicommon/resources/ResourcesStorage.scala @@ -0,0 +1,36 @@ +package xyz.driver.pdsuicommon.resources + +import scala.io.{Codec, Source} + +trait ResourcesStorage { + + /** + * @param resourcePath Don't forget / at start + */ + def getFirstLine(resourcePath: String): String +} + +object RealResourcesStorage extends ResourcesStorage { + + def getFirstLine(resourcePath: String): String = { + val resourceUrl = getClass.getResource(resourcePath) + Option(resourceUrl) match { + case Some(url) => + val source = Source.fromURL(resourceUrl)(Codec.UTF8) + try { + val lines = source.getLines() + if (lines.isEmpty) throw new RuntimeException(s"'$resourcePath' is empty") + else lines.next() + } finally { + source.close() + } + case None => + throw new RuntimeException(s"Can not find the '$resourcePath'!") + } + } +} + +object FakeResourcesStorage extends ResourcesStorage { + + def getFirstLine(resourcePath: String): String = "" +} |