[ https://issues.apache.org/jira/browse/GEARPUMP-311?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=16104298#comment-16104298 ]
ASF GitHub Bot commented on GEARPUMP-311: ----------------------------------------- Github user manuzhang commented on a diff in the pull request: https://github.com/apache/incubator-gearpump/pull/198#discussion_r129997142 --- Diff: examples/streaming/wordcount/src/main/scala/org/apache/gearpump/streaming/examples/wordcount/dsl/refactor/WordCount.scala --- @@ -0,0 +1,151 @@ +/* + * 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 + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.gearpump.streaming.examples.wordcount.dsl.refactor + +import org.apache.commons.logging.{Log, LogFactory} +import org.apache.gearpump.streaming.refactor.coder.{StringUtf8Coder, VarLongCoder} +import org.apache.gearpump.streaming.refactor.dsl.api.functions.MapWithStateFunction +import org.apache.gearpump.streaming.refactor.dsl.scalaapi.StreamApp +import org.apache.gearpump.streaming.refactor.dsl.scalaapi.functions.FlatMapWithStateFunction +import org.apache.gearpump.streaming.refactor.state.api.{StateInternals, ValueState} +import org.apache.gearpump.streaming.refactor.state.{RuntimeContext, StateNamespaces, StateTags} +import org.apache.hadoop.conf.Configuration +import org.apache.gearpump.cluster.UserConfig +import org.apache.gearpump.cluster.client.ClientContext +import org.apache.gearpump.cluster.main.{ArgumentsParser, CLIOption, ParseResult} +import org.apache.gearpump.streaming.hadoop.HadoopCheckpointStoreFactory +import org.apache.gearpump.streaming.hadoop.lib.rotation.FileSizeRotation +import org.apache.gearpump.streaming.state.impl.PersistentStateConfig +import org.apache.gearpump.streaming.{Processor, StreamApplication} +import org.apache.gearpump.util.{AkkaApp, Graph} + +/** + * + */ +object WordCount extends AkkaApp with ArgumentsParser { + + override val options: Array[(String, CLIOption[Any])] = Array.empty + + override def main(akkaConf: Config, args: Array[String]): Unit = { + val context = ClientContext(akkaConf) + + val hadoopConfig = new Configuration + val checkpointStoreFactory = new HadoopCheckpointStoreFactory("MessageConsume", hadoopConfig, + // Rotates on 1MB + new FileSizeRotation(1000000)) + val taskConfig = UserConfig.empty + .withBoolean(PersistentStateConfig.STATE_CHECKPOINT_ENABLE, true) + .withLong(PersistentStateConfig.STATE_CHECKPOINT_INTERVAL_MS, 1000L) + .withValue(PersistentStateConfig.STATE_CHECKPOINT_STORE_FACTORY, + checkpointStoreFactory)(context.system) + + val app = StreamApp("dsl", context, taskConfig) + val data = "This is a good start, bingo!! bingo!!" + app.source(data.lines.toList, 1, "source"). + // word => (word, count) + flatMapWithState(new StatefulFlatMapFunction(), "a stateful flatmap udf"). + mapWithState(new StatefulMapFunction(), ""). + // (word, count1), (word, count2) => (word, count1 + count2) + groupByKey().sum.log + + context.submit(app).waitUntilFinish() + context.close() + } + + + private class StatefulFlatMapFunction + extends FlatMapWithStateFunction[String, String] { + + private val logger: Log = LogFactory.getLog(getClass) + + private implicit val counterStateTag = "tag1" --- End diff -- We'd better refrain us from using Scala implicit in user land and provide an explicit interface instead > refactor state management > ------------------------- > > Key: GEARPUMP-311 > URL: https://issues.apache.org/jira/browse/GEARPUMP-311 > Project: Apache Gearpump > Issue Type: Improvement > Components: streaming > Reporter: yanghua > Assignee: yanghua > Priority: Minor > > inspired by Apache Beam -- This message was sent by Atlassian JIRA (v6.4.14#64029)