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

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mysql-async_2.13 Show documentation

mysql-async

The newest version!

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
  }

}