147be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org/* 247be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org * Copyright 2004 The WebRTC Project Authors. All rights reserved. 347be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org * 447be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org * Use of this source code is governed by a BSD-style license 547be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org * that can be found in the LICENSE file in the root of the source 647be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org * tree. An additional intellectual property rights grant can be found 747be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org * in the file PATENTS. All contributing project authors may 847be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org * be found in the AUTHORS file in the root of the source tree. 947be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org */ 1047be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org 1147be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org#include "webrtc/base/worker.h" 1247be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org 1347be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org#include "webrtc/base/common.h" 1447be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org#include "webrtc/base/logging.h" 1547be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org#include "webrtc/base/thread.h" 1647be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org 1747be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.orgnamespace rtc { 1847be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org 1947be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.orgenum { 2047be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org MSG_HAVEWORK = 0, 2147be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org}; 2247be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org 2347be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.orgWorker::Worker() : worker_thread_(NULL) {} 2447be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org 2547be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.orgWorker::~Worker() { 2647be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org // We need to already be stopped before being destroyed. We cannot call 2747be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org // StopWork() from here because the subclass's data has already been 2847be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org // destructed, so OnStop() cannot be called. 2947be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org ASSERT(!worker_thread_); 3047be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org} 3147be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org 3247be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.orgbool Worker::StartWork() { 3347be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org rtc::Thread *me = rtc::Thread::Current(); 3447be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org if (worker_thread_) { 3547be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org if (worker_thread_ == me) { 3647be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org // Already working on this thread, so nothing to do. 3747be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org return true; 3847be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org } else { 3947be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org LOG(LS_ERROR) << "Automatically switching threads is not supported"; 4047be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org ASSERT(false); 4147be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org return false; 4247be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org } 4347be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org } 4447be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org worker_thread_ = me; 4547be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org OnStart(); 4647be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org return true; 4747be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org} 4847be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org 4947be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.orgbool Worker::StopWork() { 5047be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org if (!worker_thread_) { 5147be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org // Already not working, so nothing to do. 5247be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org return true; 5347be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org } else if (worker_thread_ != rtc::Thread::Current()) { 5447be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org LOG(LS_ERROR) << "Stopping from a different thread is not supported"; 5547be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org ASSERT(false); 5647be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org return false; 5747be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org } 5847be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org OnStop(); 5947be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org worker_thread_->Clear(this, MSG_HAVEWORK); 6047be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org worker_thread_ = NULL; 6147be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org return true; 6247be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org} 6347be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org 6447be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.orgvoid Worker::HaveWork() { 6547be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org ASSERT(worker_thread_ != NULL); 6647be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org worker_thread_->Post(this, MSG_HAVEWORK); 6747be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org} 6847be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org 6947be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.orgvoid Worker::OnMessage(rtc::Message *msg) { 7047be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org ASSERT(msg->message_id == MSG_HAVEWORK); 7147be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org ASSERT(worker_thread_ == rtc::Thread::Current()); 7247be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org OnHaveWork(); 7347be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org} 7447be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org 7547be73b8629244d6bb63a28198f97f040ce53d21henrike@webrtc.org} // namespace rtc 76