From 1722e8314cd959f86beb1585a5e280de4604cef2 Mon Sep 17 00:00:00 2001
From: mbalassi <mbalassi@apache.org>
Date: Thu, 30 Apr 2015 16:49:11 +0200
Subject: [PATCH] [streaming] Duplicate userfunction eliminated from StreamMap

---
 .../org/apache/flink/streaming/api/operators/StreamMap.java | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/flink-staging/flink-streaming/flink-streaming-core/src/main/java/org/apache/flink/streaming/api/operators/StreamMap.java b/flink-staging/flink-streaming/flink-streaming-core/src/main/java/org/apache/flink/streaming/api/operators/StreamMap.java
index 9f1db1ec032..f421f5ddbac 100644
--- a/flink-staging/flink-streaming/flink-streaming-core/src/main/java/org/apache/flink/streaming/api/operators/StreamMap.java
+++ b/flink-staging/flink-streaming/flink-streaming-core/src/main/java/org/apache/flink/streaming/api/operators/StreamMap.java
@@ -22,11 +22,8 @@ import org.apache.flink.api.common.functions.MapFunction;
 public class StreamMap<IN, OUT> extends ChainableStreamOperator<IN, OUT> {
 	private static final long serialVersionUID = 1L;
 
-	private MapFunction<IN, OUT> mapper;
-
 	public StreamMap(MapFunction<IN, OUT> mapper) {
 		super(mapper);
-		this.mapper = mapper;
 	}
 
 	@Override
@@ -37,8 +34,9 @@ public class StreamMap<IN, OUT> extends ChainableStreamOperator<IN, OUT> {
 	}
 
 	@Override
+	@SuppressWarnings("unchecked")
 	protected void callUserFunction() throws Exception {
-		collector.collect(mapper.map(nextObject));
+		collector.collect(((MapFunction<IN, OUT>) userFunction).map(nextObject));
 	}
 
 }
-- 
GitLab