Trino lambda表达式使用学习小结
Trino lambda表达式使用学习小结
Trino中使用了很多的lambda表达式的写法，这点与Impala非常不同。这里简单学习了一些场景下的lambda表达式的用法。具体的例子如下：
private final JdbcApiStats getTableNames = new JdbcApiStats();
public JdbcApiStats getGetTableNames()
{
return getTableNames;
}
public List<SchemaTableName> getTableNames(ConnectorSession session, Optional<String> schema)
{
return stats.getGetTableNames().wrap(() -> delegate().getTableNames(session, schema));
}函数本体只有一行代码，但是却完成了多个动作，这里主要用到的就是lambda表达式的写法，新手阅读起来会非常困难，我们一步一步拆解来看。首先通过getGetTableNames()方法获取到了一个JdbcApiStats类型的变量，然后调用其wrap方法。传入的参数是一个lambda表达式，即() -> delegate().getTableNames(session, schema)。这个表达式不需要参数，返回值是一个List集合。我们看下JdbcApiStats这个类的内容，如下所示：
@ThreadSafe
public class JdbcApiStats
{
private final TimeStat time = new TimeStat(MILLISECONDS);
private final CounterStat failures = new CounterStat();
public <V, E extends Exception> V wrap(ThrowingCallable<V, E> callable)
throws E
{
try (TimeStat.BlockTimer ignored = time.time()) {
return callable.call();
}
catch (Exception e) {
failures.update(1);
throw e;
}
}
public <E extends Exception> void wrap(ThrowingRunnable<E> callable)
throws E
{
try (TimeStat.BlockTimer ignored = time.time()) {
callable.run();
}
catch (Exception e) {
failures.update(1);
throw e;
}
}
@Managed
@Nested
public TimeStat getTime()
{
return time;
}
@Managed
@Nested
public CounterStat getFailures()
{
return failures;
}
public interface ThrowingCallable<V, E extends Exception>
{
V call() throws E;
}
public interface ThrowingRunnable<E extends Exception>
{
void run() throws E;
}
}这里针对有返回值和无返回值的情况，定义了两个interface，分别是：ThrowingCallable和ThrowingRunnable，分别只有一个abstract method，这就是我们通常所说的函数式接口，有且仅有一个抽象方法。然后用这两个interface分别作为参数，构造了两个wrap函数。同时，进行了一些时间和失败次数的状态统计。再回到最初的labmda代码，我们其实就可以拆解为如下的几行代码：
public List<SchemaTableName> getTableNames(ConnectorSession session, Optional<String> schema)
{
JdbcApiStats.ThrowingCallable callable = () -> delegate().getTableNames(session, schema);
List<SchemaTableName> list = (List<SchemaTableName>) stats.getGetTableNames().wrap(callable);
return list;
}可以看到，这里将lambda显示转换为了ThrowingCallable这个interface类型的变量，然后传入到wrap函数。最后，将返回值显示转换为List，这里对应的就是ThrowingCallable中的V。在原先的写法中，编译器可以直接就从delegate().getTableNames()推导出返回值的类型，因此不需要显示转换。而ThrowingCallable中的call也就对应了delegate().getTableNames()，这里函数名称不一定必须是call，也可以换成其他的名称，但是只能有一个abstract method。如果我们再增加一个abstract method，例如V call2() throws E，编辑器就会报错，如下所示：
这样的使用好处就是，对于不同动作的相同wrap，就可以直接通过同一个JdbcApiStats来实现，非常方便：
@Override
public Optional<JdbcTableHandle> getTableHandle(ConnectorSession session, SchemaTableName schemaTableName)
{
return stats.getGetTableHandle().wrap(() -> delegate().getTableHandle(session, schemaTableName));
}
@Override
public List<JdbcColumnHandle> getColumns(ConnectorSession session, JdbcTableHandle tableHandle)
{
return stats.getGetColumns().wrap(() -> delegate().getColumns(session, tableHandle));
}Trino的代码中用了很多函数式编程和lambda特性，后续有时间再继续研究学习。
腾讯云开发者
