com.github.mauricio.async.db.util.ChannelFutureTransformer.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of db-async-common_2.13.0-M5 Show documentation

db-async-common

The newest version!

/*
 * Copyright 2013 Maurício Linhares
 *
 * Maurício Linhares 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 com.github.mauricio.async.db.util

import io.netty.channel.{ChannelFutureListener, ChannelFuture}
import scala.concurrent.{Promise, Future}
import com.github.mauricio.async.db.exceptions.CanceledChannelFutureException
import scala.language.implicitConversions

object ChannelFutureTransformer {

  implicit class ChannelFutureSyntax(channelFuture: ChannelFuture) {
    def asScala = {
      val promise = Promise[ChannelFuture]
      channelFuture.addListener(new ChannelFutureListener {
        def operationComplete(future: ChannelFuture) {
          if ( future.isSuccess ) {
            promise.success(future)
          } else {
            val exception = if ( future.cause == null ) {
              new CanceledChannelFutureException(future)
                .fillInStackTrace()
            } else {
              future.cause
            }

            promise.failure(exception)

          }
        }
      })
      promise.future
    }

  }
}