mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-27 21:46:04 +02:00
ef4c5eb65e
Please note we have force-pushed a new initial commit in order to remove some publicly-available Twitter user information. Note that this process may be required in the future.
44 lines
1.8 KiB
Scala
44 lines
1.8 KiB
Scala
package com.twitter.home_mixer.util
|
|
|
|
import com.twitter.bijection.Injection
|
|
import com.twitter.io.Buf
|
|
import com.twitter.servo.util.Transformer
|
|
import com.twitter.storage.client.manhattan.bijections.Bijections
|
|
import com.twitter.util.Return
|
|
import com.twitter.util.Try
|
|
import java.nio.ByteBuffer
|
|
|
|
object InjectionTransformerImplicits {
|
|
implicit class ByteArrayInjectionToByteBufferTransformer[A](baInj: Injection[A, Array[Byte]]) {
|
|
|
|
private val bbInj: Injection[A, ByteBuffer] = baInj
|
|
.andThen(Bijections.byteArray2Buf)
|
|
.andThen(Bijections.byteBuffer2Buf.inverse)
|
|
|
|
def toByteBufferTransformer(): Transformer[A, ByteBuffer] = new InjectionTransformer(bbInj)
|
|
def toByteArrayTransformer(): Transformer[A, Array[Byte]] = new InjectionTransformer(baInj)
|
|
}
|
|
|
|
implicit class BufInjectionToByteBufferTransformer[A](bufInj: Injection[A, Buf]) {
|
|
|
|
private val bbInj: Injection[A, ByteBuffer] = bufInj.andThen(Bijections.byteBuffer2Buf.inverse)
|
|
private val baInj: Injection[A, Array[Byte]] = bufInj.andThen(Bijections.byteArray2Buf.inverse)
|
|
|
|
def toByteBufferTransformer(): Transformer[A, ByteBuffer] = new InjectionTransformer(bbInj)
|
|
def toByteArrayTransformer(): Transformer[A, Array[Byte]] = new InjectionTransformer(baInj)
|
|
}
|
|
|
|
implicit class ByteBufferInjectionToByteBufferTransformer[A](bbInj: Injection[A, ByteBuffer]) {
|
|
|
|
private val baInj: Injection[A, Array[Byte]] = bbInj.andThen(Bijections.bb2ba)
|
|
|
|
def toByteBufferTransformer(): Transformer[A, ByteBuffer] = new InjectionTransformer(bbInj)
|
|
def toByteArrayTransformer(): Transformer[A, Array[Byte]] = new InjectionTransformer(baInj)
|
|
}
|
|
}
|
|
|
|
class InjectionTransformer[A, B](inj: Injection[A, B]) extends Transformer[A, B] {
|
|
override def to(a: A): Try[B] = Return(inj(a))
|
|
override def from(b: B): Try[A] = Try.fromScala(inj.invert(b))
|
|
}
|