com.amazon.sqs.javamessaging.util.SQSMessagingClientThreadFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of amazon-sqs-java-messaging-lib Show documentation

The Amazon SQS Java Messaging Library holds the Java Message Service compatible classes, that are used for communicating with Amazon Simple Queue Service.

There is a newer version: 2.1.4

Show newest version

/*
 * Copyright 2010-2014 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License").
 * You may not use this file except in compliance with the License.
 * A copy of the License is located at
 *
 *  http://aws.amazon.com/apache2.0
 *
 * or in the "license" file accompanying this file. This file 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.amazon.sqs.javamessaging.util;

import java.util.concurrent.ThreadFactory;
import java.util.concurrent.atomic.AtomicInteger;

/** Simple thread factory that supports ThreadGroups */

public class SQSMessagingClientThreadFactory implements ThreadFactory {

    private final String threadBaseName;

    private final AtomicInteger threadCounter;

    private final boolean isDaemon;

    private ThreadGroup threadGroup;

    public SQSMessagingClientThreadFactory(String taskName, boolean isDaemon) {
        this(taskName, isDaemon, false);
    }

    public SQSMessagingClientThreadFactory(String taskName, boolean isDaemon, boolean createWithThreadGroup) {
        this.threadBaseName = taskName + "Thread-";
        this.threadCounter = new AtomicInteger(0);
        this.isDaemon = isDaemon;
        if (createWithThreadGroup) {
            threadGroup = new ThreadGroup(taskName + "ThreadGroup");
            threadGroup.setDaemon(isDaemon);
        }
    }

    public SQSMessagingClientThreadFactory(String taskName, ThreadGroup threadGroup) {
        this.threadBaseName = taskName + "Thread-";
        this.threadCounter = new AtomicInteger(0);
        this.isDaemon = threadGroup.isDaemon();
        this.threadGroup = threadGroup;
    }
    
    /**
     * Constructs a new Thread. Initializes name, daemon status, and ThreadGroup
     * if there is any.
     * 
     * @param r
     *            A runnable to be executed by new thread instance
     * @return The constructed thread
     */
    public Thread newThread(Runnable r) {
        Thread t;
        if (threadGroup == null) {
            t = new Thread(r, threadBaseName + threadCounter.incrementAndGet());
            t.setDaemon(isDaemon);
        } else {
            t = new Thread(threadGroup, r, threadBaseName + threadCounter.incrementAndGet());
            t.setDaemon(isDaemon);
        }
        return t;
    }
    
    /**
     * Checks if the thread is member of the thread group
     * 
     * @param thread
     * @return True If there is a thread group and the given thread is member of
     *         the group
     */
    public boolean wasThreadCreatedWithThisThreadGroup(Thread thread) {
        if (threadGroup == null) {
            return false;
        }
        return thread.getThreadGroup() == threadGroup;
    }
  
}