| /** |
| * 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 com.alibaba.jstorm.callback; |
| |
| import java.lang.Thread.UncaughtExceptionHandler; |
| |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| import backtype.storm.utils.Time; |
| |
| import com.alibaba.jstorm.utils.JStormUtils; |
| import com.alibaba.jstorm.utils.SmartThread; |
| |
| /** |
| * Wrapper Timer thread Every several seconds execute afn, if something is run, run kill_fn |
| * |
| * |
| * @author yannian |
| * |
| */ |
| public class AsyncLoopThread implements SmartThread { |
| private static final Logger LOG = LoggerFactory.getLogger(AsyncLoopThread.class); |
| |
| private Thread thread; |
| |
| private RunnableCallback afn; |
| |
| public AsyncLoopThread(RunnableCallback afn) { |
| this.init(afn, false, Thread.NORM_PRIORITY, true); |
| } |
| |
| public AsyncLoopThread(RunnableCallback afn, boolean daemon, int priority, boolean start) { |
| this.init(afn, daemon, priority, start); |
| } |
| |
| public AsyncLoopThread(RunnableCallback afn, boolean daemon, RunnableCallback kill_fn, int priority, boolean start) { |
| this.init(afn, daemon, kill_fn, priority, start); |
| } |
| |
| public void init(RunnableCallback afn, boolean daemon, int priority, boolean start) { |
| RunnableCallback kill_fn = new AsyncLoopDefaultKill(); |
| this.init(afn, daemon, kill_fn, priority, start); |
| } |
| |
| /** |
| * |
| * @param afn |
| * @param daemon |
| * @param kill_fn (Exception e) |
| * @param priority |
| * @param args_fn |
| * @param start |
| */ |
| private void init(RunnableCallback afn, boolean daemon, RunnableCallback kill_fn, int priority, boolean start) { |
| if (kill_fn == null) { |
| kill_fn = new AsyncLoopDefaultKill(); |
| } |
| |
| Runnable runable = new AsyncLoopRunnable(afn, kill_fn); |
| thread = new Thread(runable); |
| String threadName = afn.getThreadName(); |
| if (threadName == null) { |
| threadName = afn.getClass().getSimpleName(); |
| } |
| thread.setName(threadName); |
| thread.setDaemon(daemon); |
| thread.setPriority(priority); |
| thread.setUncaughtExceptionHandler(new UncaughtExceptionHandler() { |
| @Override |
| public void uncaughtException(Thread t, Throwable e) { |
| LOG.error("UncaughtException", e); |
| JStormUtils.halt_process(1, "UncaughtException"); |
| } |
| }); |
| |
| this.afn = afn; |
| |
| if (start) { |
| thread.start(); |
| } |
| |
| } |
| |
| @Override |
| public void start() { |
| thread.start(); |
| } |
| |
| @Override |
| public void join() throws InterruptedException { |
| thread.join(); |
| } |
| |
| // for test |
| public void join(int times) throws InterruptedException { |
| thread.join(times); |
| } |
| |
| @Override |
| public void interrupt() { |
| thread.interrupt(); |
| } |
| |
| @Override |
| public Boolean isSleeping() { |
| return Time.isThreadWaiting(thread); |
| } |
| |
| public Thread getThread() { |
| return thread; |
| } |
| |
| @Override |
| public void cleanup() { |
| // TODO Auto-generated method stub |
| afn.shutdown(); |
| } |
| } |