diff options
Diffstat (limited to 'src/main/scala/xyz/driver/pdsuicommon/utils/WritesUtils.scala')
-rw-r--r-- | src/main/scala/xyz/driver/pdsuicommon/utils/WritesUtils.scala | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/src/main/scala/xyz/driver/pdsuicommon/utils/WritesUtils.scala b/src/main/scala/xyz/driver/pdsuicommon/utils/WritesUtils.scala new file mode 100644 index 0000000..fa05e96 --- /dev/null +++ b/src/main/scala/xyz/driver/pdsuicommon/utils/WritesUtils.scala @@ -0,0 +1,21 @@ +package xyz.driver.pdsuicommon.utils + +import play.api.libs.json._ + +object WritesUtils { + + def filterKeys[T](p: String => Boolean)(implicit w: Writes[T]): Writes[T] = { + filter { + case (key, _) => p(key) + } + } + + def filter[T](p: (String, JsValue) => Boolean)(implicit w: Writes[T]): Writes[T] = { + w.transform { input: JsValue => + input match { + case JsObject(map) => JsObject(map.filter(Function.tupled(p))) + case x => x + } + } + } +} |