org.apache.activemq.apollo.broker.protocol.RawMessageCodec.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of apollo-broker Show documentation
A reliable messaging server.
The newest version!
/**
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.apache.activemq.apollo.broker.protocol

import java.nio.ByteBuffer
import org.apache.activemq.apollo.util._
import org.apache.activemq.apollo.broker._
import java.lang.{Class, String}
import org.apache.activemq.apollo.broker.store.MessageRecord
import org.fusesource.hawtbuf.{AsciiBuffer, Buffer}
import org.fusesource.hawtdispatch.transport.ProtocolCodec

object RawMessageCodecFactory extends MessageCodecFactory.Provider {
  def create = Array[MessageCodec](RawMessageCodec)
}

/**
 * 
 * 
 *
 * @author Hiram Chirino
 */
object RawMessageCodec extends MessageCodec {
  
  val PROTOCOL_ID = new AsciiBuffer(id)
  def id = "raw"

  def encode(message: Message):MessageRecord = {
    message match {
      case message:RawMessage =>
        val rc = new MessageRecord
        rc.codec = PROTOCOL_ID
        rc.buffer = message.payload
        rc
      case _ => throw new RuntimeException("Invalid message type");
    }
  }

  def decode(message: MessageRecord) = {
    assert( message.codec == PROTOCOL_ID )
    RawMessage(message.buffer)
  }
}

case class RawMessage(payload:Buffer) extends Message {

  def getBodyAs[T](toType : Class[T]) = {
    if( toType.isAssignableFrom(classOf[Buffer]) ) {
      toType.cast(payload)
    } else if( toType == classOf[Array[Byte]] ) {
      toType.cast(payload.toByteArray)
    } else if( toType == classOf[ByteBuffer] ) {
      toType.cast(payload.toByteBuffer)
    } else {
      null.asInstanceOf[T]
    }
  }

  def getLocalConnectionId = null
  def getProperty(name: String) = null
  def expiration = 0L
  def persistent = false
  def priority = 0
  def codec = RawMessageCodec
  def release() {}
  def retain() {}
  def retained() = 0
}