use Scala Option and Stream in Scala future results

This commit is contained in:
Albert Shift 2015-06-23 08:34:20 -07:00
parent ba416ba1ff
commit 071f6466ae
2 changed files with 62 additions and 21 deletions

View file

@ -17,6 +17,9 @@ package com.noorq.casser.core.operation;
import java.util.Optional; import java.util.Optional;
import scala.None;
import scala.Option;
import scala.Some;
import scala.concurrent.Future; import scala.concurrent.Future;
import com.datastax.driver.core.PreparedStatement; import com.datastax.driver.core.PreparedStatement;
@ -83,24 +86,45 @@ public abstract class AbstractOptionalOperation<E, O extends AbstractOptionalOpe
return future; return future;
} }
public Future<Optional<E>> future() { public ListenableFuture<Option<E>> asyncForScala() {
return Scala.asFuture(async());
ResultSetFuture resultSetFuture = sessionOps.executeAsync(options(buildStatement()), showValues);
ListenableFuture<Option<E>> future = Futures.transform(resultSetFuture, new Function<ResultSet, Option<E>>() {
@Override
public Option<E> apply(ResultSet resultSet) {
Optional<E> optional = transform(resultSet);
if (optional.isPresent()) {
return new Some<E>(optional.get());
}
else {
return Option.empty();
}
}
}, sessionOps.getExecutor());
return future;
}
public Future<Option<E>> future() {
return Scala.asFuture(asyncForScala());
} }
public <A> Future<Fun.Tuple2<Optional<E>, A>> future(A a) { public <A> Future<Fun.Tuple2<Option<E>, A>> future(A a) {
return Scala.asFuture(async(), a); return Scala.asFuture(asyncForScala(), a);
} }
public <A, B> Future<Fun.Tuple3<Optional<E>, A, B>> future(A a, B b) { public <A, B> Future<Fun.Tuple3<Option<E>, A, B>> future(A a, B b) {
return Scala.asFuture(async(), a, b); return Scala.asFuture(asyncForScala(), a, b);
} }
public <A, B, C> Future<Fun.Tuple4<Optional<E>, A, B, C>> future(A a, B b, C c) { public <A, B, C> Future<Fun.Tuple4<Option<E>, A, B, C>> future(A a, B b, C c) {
return Scala.asFuture(async(), a, b, c); return Scala.asFuture(asyncForScala(), a, b, c);
} }
public <A, B, C, D> Future<Fun.Tuple5<Optional<E>, A, B, C, D>> future(A a, B b, C c, D d) { public <A, B, C, D> Future<Fun.Tuple5<Option<E>, A, B, C, D>> future(A a, B b, C c, D d) {
return Scala.asFuture(async(), a, b, c, d); return Scala.asFuture(asyncForScala(), a, b, c, d);
} }
} }

View file

@ -83,24 +83,41 @@ public abstract class AbstractStreamOperation<E, O extends AbstractStreamOperati
return future; return future;
} }
public Future<Stream<E>> future() { public ListenableFuture<scala.collection.immutable.Stream<E>> asyncForScala() {
return Scala.asFuture(async());
ResultSetFuture resultSetFuture = sessionOps.executeAsync(options(buildStatement()), showValues);
ListenableFuture<scala.collection.immutable.Stream<E>> future = Futures.transform(resultSetFuture, new Function<ResultSet, scala.collection.immutable.Stream<E>>() {
@Override
public scala.collection.immutable.Stream<E> apply(ResultSet resultSet) {
Stream<E> stream = transform(resultSet);
return scala.collection.JavaConversions.asScalaIterator(stream.iterator()).toStream();
}
}, sessionOps.getExecutor());
return future;
} }
public <A> Future<Fun.Tuple2<Stream<E>, A>> future(A a) { public Future<scala.collection.immutable.Stream<E>> future() {
return Scala.asFuture(async(), a); return Scala.asFuture(asyncForScala());
} }
public <A, B> Future<Fun.Tuple3<Stream<E>, A, B>> future(A a, B b) { public <A> Future<Fun.Tuple2<scala.collection.immutable.Stream<E>, A>> future(A a) {
return Scala.asFuture(async(), a, b); return Scala.asFuture(asyncForScala(), a);
} }
public <A, B, C> Future<Fun.Tuple4<Stream<E>, A, B, C>> future(A a, B b, C c) { public <A, B> Future<Fun.Tuple3<scala.collection.immutable.Stream<E>, A, B>> future(A a, B b) {
return Scala.asFuture(async(), a, b, c); return Scala.asFuture(asyncForScala(), a, b);
} }
public <A, B, C, D> Future<Fun.Tuple5<Stream<E>, A, B, C, D>> future(A a, B b, C c, D d) { public <A, B, C> Future<Fun.Tuple4<scala.collection.immutable.Stream<E>, A, B, C>> future(A a, B b, C c) {
return Scala.asFuture(async(), a, b, c, d); return Scala.asFuture(asyncForScala(), a, b, c);
}
public <A, B, C, D> Future<Fun.Tuple5<scala.collection.immutable.Stream<E>, A, B, C, D>> future(A a, B b, C c, D d) {
return Scala.asFuture(asyncForScala(), a, b, c, d);
} }
} }