com.github.mauricio.async.db.mysql.encoder.AuthenticationSwitchResponseEncoder.scala Maven / Gradle / Ivy

Go to download

package com.github.mauricio.async.db.mysql.encoder

import com.github.mauricio.async.db.mysql.message.client.{AuthenticationSwitchResponse, ClientMessage}
import io.netty.buffer.ByteBuf
import com.github.mauricio.async.db.exceptions.UnsupportedAuthenticationMethodException
import com.github.mauricio.async.db.mysql.encoder.auth.AuthenticationMethod
import java.nio.charset.Charset
import com.github.mauricio.async.db.util.ByteBufferUtils

class AuthenticationSwitchResponseEncoder( charset : Charset ) extends MessageEncoder {

  def encode(message: ClientMessage): ByteBuf = {
    val switch = message.asInstanceOf[AuthenticationSwitchResponse]

    val method = switch.request.method
    val authenticator = AuthenticationMethod.Availables.getOrElse(
    method, { throw new UnsupportedAuthenticationMethodException(method) })

    val buffer = ByteBufferUtils.packetBuffer()

    val bytes = authenticator.generateAuthentication(charset, switch.password, switch.request.seed.getBytes(charset) )
    buffer.writeBytes(bytes)

    buffer
  }

}