�
d�bc@s�dd
lZdd
l
Z
dd
lZddlmZ
ddlmZmZ
ddlm	Z	
ddl
mZmZm
Z

ddlmZ
ddlmZ
d�Zd	ef
d
��YZd
S(i����N(
tlogger(t	add_eventtWALAEventOperation(
tServiceStoppedError(tustrtQueuetEmpty(
tThreadHandlerInterface(
ttextutilc

Cs
t|�
S(
N(
tSendTelemetryEventsHandler(
t
protocol_util((sL/usr/lib/python2.7/site-packages/azurelinuxagent/ga/send_telemetry_events.pyt!get_send_telemetry_events_handlers
R	cBs�eZ
dZd
Zejdd�
jZdZejdd�
Z	d�Z
ed��
Zd�Z
d�Zd	�Zd
�Zd�Zd�Zd
�Zd�Zd�Zd�Zd�ZRS(s�
    This Handler takes care of sending all telemetry out of the agent to Wireserver. It sends out data as soon as
    there's any data available in the queue to send.
    tSendTelemetryHandlertsecondsiicCs1|
j�|_
t|_d|_t�|_dS(
N(tget_protocolt	_protocoltTruet
should_runtNonet_threadRt_queue(tselfR
((sL/usr/lib/python2.7/site-packages/azurelinuxagent/ga/send_telemetry_events.pyt__init__.s

	
		c
Cstj
S(
N(R	t_THREAD_NAME(((sL/usr/lib/python2.7/site-packages/azurelinuxagent/ga/send_telemetry_events.pytget_thread_name<sc

Cstj
d
�

|j�
dS(Ns#Start SendTelemetryHandler service.(Rtinfotstart(
R((sL/usr/lib/python2.7/site-packages/azurelinuxagent/ga/send_telemetry_events.pytrun@s


c

Cs|jdk	o|jj�S(
N(RRtis_alive(
R((sL/usr/lib/python2.7/site-packages/azurelinuxagent/ga/send_telemetry_events.pyRDs
c

CsOtj
d
|j�
|_|jjt�

|jj|j��

|jj�
dS(Nttarget(	t	threadingtThreadt_process_telemetry_threadRt	setDaemonRtsetNameRR(
R((sL/usr/lib/python2.7/site-packages/azurelinuxagent/ga/send_telemetry_events.pyRGs



c

Cs&t|_
|j�r"|j�
nd
S(sO
        Stop server communication and join the thread to main thread.
        N(tFalseRRtjoin(
R((sL/usr/lib/python2.7/site-packages/azurelinuxagent/ga/send_telemetry_events.pytstopMs	

c


Cs|jj
�
|jj
�
dS(
N(RR$R(
R((sL/usr/lib/python2.7/site-packages/azurelinuxagent/ga/send_telemetry_events.pyR$Us


c


Cs|jS(
N(
R(
R((sL/usr/lib/python2.7/site-packages/azurelinuxagent/ga/send_telemetry_events.pytstoppedYs
cCs||j�r*t
d
j|j��
�
�
ny|jj|
dtj�


Wn.tk
rw
}
t
djt	|�
�
�
�
n
XdS(Ns,{0} is stopped, not accepting anymore eventsttimeoutsMUnable to enqueue due to: {0}, stopping any more enqueuing until the next run(
R&RtformatRRtputR	t_MAX_TIMEOUTt	ExceptionR(Rteventterror((sL/usr/lib/python2.7/site-packages/azurelinuxagent/ga/send_telemetry_events.pyt
enqueue_event\s




c
CsGy)|jj
d
tj�
}
|jj�
Wntk
rB


d}
n
X|
S(sE

        Wait for atleast one event in Queue or timeout after SendTelemetryEventsHandler._MAX_TIMEOUT seconds.
        In case of a timeout, set the event to None.
        :return: event if an event is added to the Queue or None to signify no events were added in queue.
        This would raise in case of an error.
        R'N(RtgetR	R*t	task_doneRR(RR,((sL/usr/lib/python2.7/site-packages/azurelinuxagent/ga/send_telemetry_events.pyt_wait_for_event_in_queuels




c
Cs�tj
d
j|j��
�

yJxC|j�s?|jj�rd|j�}
|
r"|j|
�

q"q"WWnPt	k
r�
}
dj|j�t
j|�
�}tdt
jd|dt�
n
XdS(Ns#Successfully started the {0} threadsJAn unknown error occurred in the {0} thread main loop, stopping thread.{1}toptmessaget
is_success(RRR(RR&RtemptyR1t_send_events_in_queueR+Rtformat_exceptionR
RtUnhandledErrorR#(Rtfirst_eventR-terr_msg((sL/usr/lib/python2.7/site-packages/azurelinuxagent/ga/send_telemetry_events.pyR |s

 




cCs�tjj
�}x�|j�r�|jj�d
|jkr�||jtjj
�kr�tjd|jj�d
tjj
�|j	�
t
jd
�

qW|jj
|j|
�
�

dS(Ni
sMWaiting for events to batch. Total events so far: {0}, Time elapsed: {1} secs(tdatetimetutcnowR&Rtqsizet_MIN_EVENTS_TO_BATCHt_MIN_BATCH_WAIT_TIMERtverboseR
ttimetsleepRtreport_eventt_get_events_in_queue(RR9t
start_time((sL/usr/lib/python2.7/site-packages/azurelinuxagent/ga/send_telemetry_events.pyR6�s
,
	
'
ccsy|
V
xm|jj
�sty%|jj�}|jj�
|V
Wqtk
rp
}
tjd
jtj	|�
�
�

qXqWdS(Ns2Some exception when fetching event from queue: {0}(
RR5t
get_nowaitR0R+RR-R(RR7(RR9R,R-((sL/usr/lib/python2.7/site-packages/azurelinuxagent/ga/send_telemetry_events.pyRD�s






	

(t__name__t
__module__t__doc__RR;t	timedeltaR
R*R>R?RtstaticmethodRRRRR%R$R&R.R1R R6RD(((sL/usr/lib/python2.7/site-packages/azurelinuxagent/ga/send_telemetry_events.pyR	#s$


											(R;RRAtazurelinuxagent.commonRtazurelinuxagent.common.eventR
Rt azurelinuxagent.common.exceptionRtazurelinuxagent.common.futureRRRt!azurelinuxagent.common.interfacesRtazurelinuxagent.common.utilsRRR	(((sL/usr/lib/python2.7/site-packages/azurelinuxagent/ga/send_telemetry_events.pyt<module>s