blob: 2d0f199712b27d627bd50808eb56e6cfa896cfac [file] [log] [blame]
#!/usr/bin/env python3
# -*- encoding: utf-8 -*-
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF 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
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
'''module for example topology: CustomGroupingTopology'''
import logging
from heronpy.api.custom_grouping import ICustomGrouping
import heronpy.api.api_constants as constants
from heronpy.api.topology import Topology
from import Grouping
from examples.src.python.spout import WordSpout
from examples.src.python.bolt import ConsumeBolt
# pylint: disable=unused-argument
class SampleCustomGrouping(ICustomGrouping):
def prepare(self, context, component, stream, target_tasks):
logging.getLogger().info("In prepare of SampleCustomGrouping, "
"with src component: %s, "
"with stream id: %s, "
"with target tasks: %s"
, component, stream, str(target_tasks))
self.target_tasks = target_tasks
def choose_tasks(self, values):
# only emits to the first task id
return [self.target_tasks[0]]
class CustomGrouping(Topology):
word_spout = WordSpout.spec(par=1)
consume_bolt = ConsumeBolt.spec(par=3,
inputs={word_spout: Grouping.custom(SampleCustomGrouping())},
config={constants.TOPOLOGY_TICK_TUPLE_FREQ_SECS: 10})