Fixes FLINK-1276

Closes #229
[FLINK-1249] [APIs] [compiler] Add custom partitioner for CoGroup

This closes #228.