Skip to content
Toggle navigation
P
Projects
G
Groups
S
Snippets
Help
CIRCLE3
/
agentdriver
This project
Loading...
Sign in
Toggle navigation
Go to a project
Project
Repository
Issues
0
Merge Requests
0
Pipelines
Wiki
Snippets
Members
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Commit
edddef0f
authored
Dec 10, 2025
by
Szeberényi Imre
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
celery fix
parent
8a0740e8
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
72 additions
and
34 deletions
+72
-34
agentmancelery.py
+42
-12
protocol.py
+19
-21
serializers.py
+11
-1
No files found.
agentmancelery.py
View file @
edddef0f
...
...
@@ -3,35 +3,65 @@ import serializers
from
kombu
import
Queue
,
Exchange
from
os
import
getenv
QUEUE_NAME
=
"localho
ts
.man"
QUEUE_NAME
=
"localho
st
.man"
AMQP_URI
=
getenv
(
'AMQP_URI'
)
celery
=
Celery
(
'agentman'
,
broker
=
AMQP_URI
)
celery
m
=
Celery
(
'agentman'
,
broker
=
AMQP_URI
)
celery
.
config_from_object
(
'celeryconfig'
)
celery
m
.
config_from_object
(
'celeryconfig'
)
celery
.
conf
.
update
(
CELERY_QUEUES
=
(
Queue
(
QUEUE_NAME
,
Exchange
(
'manager'
,
type
=
'direct'
),
routing_key
=
'manager'
),
),
task_protocol
=
1
,
# Celery 3 compatibility
)
celerym
.
conf
.
update
(
# Queue declaration
CELERY_QUEUES
=
(
Queue
(
QUEUE_NAME
,
Exchange
(
'manager'
,
type
=
'direct'
),
routing_key
=
'manager'
,
),
),
task_protocol
=
1
,
# Celery 3 compatibility
@celery.task
(
name
=
'vm.tasks.local_agent_tasks.renew'
)
# ROUTING: which task which queue / exchange / routing_key
CELERY_ROUTES
=
{
'vm.tasks.local_agent_tasks.renew'
:
{
'queue'
:
QUEUE_NAME
,
'exchange'
:
'manager'
,
'routing_key'
:
'manager'
,
},
'vm.tasks.local_agent_tasks.agent_started'
:
{
'queue'
:
QUEUE_NAME
,
'exchange'
:
'manager'
,
'routing_key'
:
'manager'
,
},
'vm.tasks.local_agent_tasks.agent_stopped'
:
{
'queue'
:
QUEUE_NAME
,
'exchange'
:
'manager'
,
'routing_key'
:
'manager'
,
},
'vm.tasks.local_agent_tasks.agent_ok'
:
{
'queue'
:
QUEUE_NAME
,
'exchange'
:
'manager'
,
'routing_key'
:
'manager'
,
},
},
)
@celerym.task
(
name
=
'vm.tasks.local_agent_tasks.renew'
)
def
renew
(
vm
):
pass
@celery.task
(
name
=
'vm.tasks.local_agent_tasks.agent_started'
)
@celery
m
.task
(
name
=
'vm.tasks.local_agent_tasks.agent_started'
)
def
agent_started
(
vm
,
version
,
system
):
pass
@celery.task
(
name
=
'vm.tasks.local_agent_tasks.agent_stopped'
)
@celery
m
.task
(
name
=
'vm.tasks.local_agent_tasks.agent_stopped'
)
def
agent_stopped
(
vm
):
pass
@celery.task
(
name
=
'vm.tasks.local_agent_tasks.agent_ok'
)
@celery
m
.task
(
name
=
'vm.tasks.local_agent_tasks.agent_ok'
)
def
agent_ok
(
vm
):
pass
protocol.py
View file @
edddef0f
...
...
@@ -17,8 +17,8 @@ reactor.connections = {}
def
numObjsByName
(
name
):
num
=
0
for
ob
in
gc
.
get_objects
():
if
isinstance
(
ob
,
name
):
num
+=
1
if
isinstance
(
ob
,
name
):
num
+=
1
return
num
class
GraphiteClientProtocol
(
protocol
.
Protocol
):
...
...
@@ -64,7 +64,7 @@ def inotify_handler(self, file, mask):
del
_p
reactor
.
running_tasks
[
vm
][
'started'
]
.
pop
(
addr
)
logger
.
info
(
'NumOBJSerialLineReceiverFactory2:
%
s'
,
numObjsByName
(
SerialLineReceiverFactory
))
logger
.
info
(
'reacror_running1
%
s'
,
reactor
.
running_tasks
)
logger
.
debug
(
'reacror_running1
%
s'
,
reactor
.
running_tasks
)
return
elif
reactor
.
running_tasks
[
vm
][
'started'
][
addr
]:
return
...
...
@@ -72,8 +72,8 @@ def inotify_handler(self, file, mask):
logger
.
info
(
"connecting to
%
s (
%
s)"
,
vm
,
file
.
path
)
ic
=
reactor
.
connectUNIX
(
file
.
path
,
serial
,
10
)
logger
.
info
(
'IConnector state:
%
s'
,
ic
.
state
)
logger
.
info
(
'reacror_running2
%
s'
,
reactor
.
running_tasks
)
# logger.info
('NumOBJSerialLineReceiverFactory2: %s' , numObjsByName(SerialLineReceiverFactory))
logger
.
debug
(
'reacror_running2
%
s'
,
reactor
.
running_tasks
)
logger
.
debug
(
'NumOBJSerialLineReceiverFactory2:
%
s'
,
numObjsByName
(
SerialLineReceiverFactory
))
class
SerialLineReceiver
(
SerialLineReceiverBase
):
...
...
@@ -88,31 +88,27 @@ class SerialLineReceiver(SerialLineReceiverBase):
def
handle_command
(
self
,
command
,
args
):
logger
.
info
(
'serial_command:
%
s
%
s'
,
command
,
args
)
from
kombu
import
Exchange
,
Queue
if
command
==
'agent_stopped'
:
agent_stopped
.
apply_async
(
queue
=
'localhost.man'
,
args
=
(
self
.
factory
.
vm
,
))
logger
.
debug
(
'agent_stopped
%
r'
,
self
.
factory
.
vm
)
agent_stopped
.
apply_async
(
args
=
(
self
.
factory
.
vm
,
))
elif
command
==
'agent_started'
:
version
=
args
.
get
(
'version'
,
None
)
system
=
args
.
get
(
'system'
,
None
)
from
kombu
import
Exchange
,
Queue
st
=
agent_started
.
apply_async
(
args
=
(
self
.
factory
.
vm
,
version
,
system
),
exchange
=
'manager'
,
routing_key
=
'manager'
,
serializer
=
'json'
,
declare
=
[
Queue
(
'localhost.man'
,
Exchange
(
'manager'
,
'direct'
),
routing_key
=
'manager'
)],
args
=
(
self
.
factory
.
vm
,
version
,
system
),
)
# logger.debug("apply_async %r", st.id)
# logger.debug("apply_st6 %r", st.status)
elif
command
==
'renew'
:
renew
.
apply_async
(
queue
=
'localhost.man'
,
args
=
(
self
.
factory
.
vm
,
))
renew
.
apply_async
(
args
=
(
self
.
factory
.
vm
,
))
elif
command
==
'ping'
:
self
.
send_response
(
response
=
'pong'
,
args
=
args
)
self
.
send_response
(
response
=
'pong'
,
args
=
args
)
def
handle_response
(
self
,
response
,
args
):
logger
.
debug
(
'handle_response:
%
s
%
s'
,
response
,
args
)
logger
.
info
(
'handle_response:
%
s
%
s'
,
response
,
args
)
vm
=
self
.
factory
.
vm
if
response
==
'status'
:
self
.
send_to_graphite
(
args
)
...
...
@@ -128,7 +124,7 @@ class SerialLineReceiver(SerialLineReceiverBase):
def
connectionMade
(
self
):
logger
.
info
(
"connected to
%
s (
%
s)"
,
self
.
factory
.
vm
,
self
.
transport
.
addr
)
logger
.
info
(
"reactor connections:
%
s"
,
reactor
.
connections
)
logger
.
debug
(
"reactor connections:
%
s"
,
reactor
.
connections
)
if
self
.
factory
.
vm
not
in
reactor
.
connections
:
reactor
.
connections
[
self
.
factory
.
vm
]
=
set
()
logger
.
info
(
"reactor connections factory:
%
s"
,
reactor
.
connections
[
self
.
factory
.
vm
])
...
...
@@ -166,7 +162,7 @@ class SerialLineReceiverFactory(protocol.ClientFactory):
reactor
.
running_tasks
[
vm
][
'started'
][
addr
]
=
self
logger
.
info
(
'NumOBJSerialLineReceiverFactory:
%
s'
,
numObjsByName
(
SerialLineReceiverFactory
))
logger
.
info
(
'NumOBJSerialLineReceiver:
%
s'
,
numObjsByName
(
SerialLineReceiver
))
logger
.
info
(
"NumConnetions:
%
s"
,
reactor
.
running_tasks
)
logger
.
debug
(
"NumConnetions:
%
s"
,
reactor
.
running_tasks
)
def
clientrConnectionLost
(
self
,
connector
):
vm
=
self
.
vm
...
...
@@ -186,3 +182,5 @@ class SerialLineReceiverFactory(protocol.ClientFactory):
reactor
.
running_tasks
[
vm
][
'started'
]
.
pop
(
addr
)
logger
.
info
(
"active connetions:
%
s"
,
reactor
.
running_tasks
[
vm
])
serializers.py
View file @
edddef0f
# serializers.py
from
kombu.serialization
import
register
from
kombu.serialization
import
register
,
enable_insecure_serializers
,
pickle_protocol
import
pickle
import
logging
# PROTOKOLL 2 → Py2.7 is képes lesz deszerializálni
pickle_protocol
=
2
logger
=
logging
.
getLogger
(
__name__
)
def
pickle_v2_dumps
(
obj
):
logger
.
debug
(
"pickle_v2_dumps CALLED type=
%
r repr=
%
r"
,
type
(
obj
),
repr
(
obj
))
return
pickle
.
dumps
(
obj
,
protocol
=
2
)
def
pickle_v2_loads
(
s
):
logger
.
debug
(
"pickle_v2_loads CALLED len=
%
r"
,
len
(
s
))
return
pickle
.
loads
(
s
)
register
(
...
...
@@ -16,3 +24,5 @@ register(
content_encoding
=
'binary'
,
)
#enable_insecure_serializers()
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment