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
b40e6ffa
authored
Nov 02, 2025
by
Szeberényi Imre
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
python3.9
parent
f79359f8
Hide whitespace changes
Inline
Side-by-side
Showing
7 changed files
with
118 additions
and
27 deletions
+118
-27
agentcelery.py
+7
-22
agentmancelery.py
+52
-0
celeryconfig.py
+10
-0
protocol.py
+26
-4
requirements/base.txt
+2
-1
requirements/install_first.txt
+3
-0
serializers.py
+18
-0
No files found.
agentcelery.py
View file @
b40e6ffa
# from twisted.internet.defer import Deferred
from
twisted.internet
import
reactor
# threads
from
celery.result
import
TimeoutError
from
celery
import
Celery
import
serializers
from
celery.result
import
TimeoutError
from
kombu
import
Queue
,
Exchange
from
os
import
getenv
from
socket
import
gethostname
...
...
@@ -17,10 +18,12 @@ HOSTNAME = gethostname().split('.')[0]
AMQP_URI
=
getenv
(
'AMQP_URI'
)
celery
=
Celery
(
'agent'
,
broker
=
AMQP_URI
)
celery
.
conf
.
update
(
CELERY_RESULT_BACKEND
=
'amqp'
,
CELERY_TASK_RESULT_EXPIRES
=
300
,
celery
.
config_from_object
(
'celeryconfig'
)
celery
.
conf
.
update
(
CELERYD_PREFETCH_MULTIPLIER
=
32
,
CELERY_QUEUES
=
(
Queue
(
HOSTNAME
+
'.agent'
,
CELERY_QUEUES
=
(
Queue
(
HOSTNAME
+
'.agent'
,
Exchange
(
'agent'
,
type
=
'direct'
),
routing_key
=
'agent'
),
))
...
...
@@ -137,21 +140,3 @@ def change_ip(vm, interfaces, dns):
send_command
(
vm
,
command
=
'change_ip'
,
interfaces
=
interfaces
,
dns
=
dns
)
@celery.task
(
name
=
'vm.tasks.local_agent_tasks.renew'
)
def
renew
(
vm
):
print
(
vm
)
@celery.task
(
name
=
'vm.tasks.local_agent_tasks.agent_started'
)
def
agent_started
(
vm
):
print
(
vm
)
@celery.task
(
name
=
'vm.tasks.local_agent_tasks.agent_stopped'
)
def
agent_stopped
(
vm
):
print
(
vm
)
@celery.task
(
name
=
'vm.tasks.local_agent_tasks.agent_ok'
)
def
agent_ok
(
vm
):
print
(
vm
)
agentmancelery.py
0 → 100644
View file @
b40e6ffa
## from twisted.internet.defer import Deferred
#from twisted.internet import reactor # threads
#from celery.result import TimeoutError
from
celery
import
Celery
import
serializers
from
kombu
import
Queue
,
Exchange
from
os
import
getenv
#from socket import gethostname
#from threading import Event
import
logging
from
celery.utils.log
import
get_task_logger
#from celery.signals import after_setup_task_logger
#from celery.app.log import TaskFormatter
logger
=
get_task_logger
(
__name__
)
QUEUE_NAME
=
"localhots.man"
AMQP_URI
=
getenv
(
'AMQP_URI'
)
celery
=
Celery
(
'agentman'
,
broker
=
AMQP_URI
)
celery
.
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
task_serializer
=
'json'
,
accept_content
=
[
'json'
],
# Ignore other content
result_serializer
=
'json'
,
timezone
=
'UTC'
,
enable_utc
=
True
,)
@celery.task
(
name
=
'vm.tasks.local_agent_tasks.renew'
)
def
renew
(
vm
):
pass
@celery.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'
)
def
agent_stopped
(
vm
):
pass
@celery.task
(
name
=
'vm.tasks.local_agent_tasks.agent_ok'
)
def
agent_ok
(
vm
):
pass
celeryconfig.py
0 → 100644
View file @
b40e6ffa
result_backend
=
'amqp://'
task_result_expires
=
300
result_expires
=
300
timezone
=
'utc'
enable_utc
=
True
accept_content
=
[
'json'
,
'pickle_v2'
]
task_serializer
=
'json'
result_serializer
=
'pickle_v2'
task_store_errors_even_if_ignored
=
True
protocol.py
View file @
b40e6ffa
...
...
@@ -8,7 +8,7 @@ import struct
from
os
import
getenv
import
gc
from
utils
import
SerialLineReceiverBase
from
agentcelery
import
agent_started
,
agent_stopped
,
renew
from
agent
man
celery
import
agent_started
,
agent_stopped
,
renew
from
celery.utils.log
import
get_task_logger
logger
=
get_task_logger
(
__name__
)
...
...
@@ -95,8 +95,30 @@ class SerialLineReceiver(SerialLineReceiverBase):
elif
command
==
'agent_started'
:
version
=
args
.
get
(
'version'
,
None
)
system
=
args
.
get
(
'system'
,
None
)
agent_started
.
apply_async
(
queue
=
'localhost.man'
,
args
=
(
self
.
factory
.
vm
,
version
,
system
))
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'
)],
)
# st = agent_started.apply_async(
# args=(self.factory.vm, version, system),
# queue='localhost.man',
# exchange='manager',
# routing_key='manager',
# serializer='json',
# )
logger
.
debug
(
"apply_async
%
r"
,
st
.
id
)
logger
.
debug
(
"apply_st
%
r"
,
st
.
status
)
time
.
sleep
(
2
)
logger
.
debug
(
"apply_st2
%
r"
,
st
.
status
)
time
.
sleep
(
8
)
logger
.
debug
(
"apply_st3
%
r"
,
st
.
status
)
elif
command
==
'renew'
:
renew
.
apply_async
(
queue
=
'localhost.man'
,
args
=
(
self
.
factory
.
vm
,
))
...
...
@@ -105,7 +127,7 @@ class SerialLineReceiver(SerialLineReceiverBase):
args
=
args
)
def
handle_response
(
self
,
response
,
args
):
logger
.
info
(
'handle_reponse:
%
s
%
s'
,
response
,
args
)
logger
.
info
(
'handle_re
s
ponse:
%
s
%
s'
,
response
,
args
)
vm
=
self
.
factory
.
vm
if
response
==
'status'
:
self
.
send_to_graphite
(
args
)
...
...
requirements/base.txt
View file @
b40e6ffa
celery==3.1.17
celery==4.4.7
#Twisted==22.10.0
Twisted==20.3.0
threadpool==1.3.2
requirements/install_first.txt
0 → 100644
View file @
b40e6ffa
setuptools>=58,<70
pip<24
serializers.py
0 → 100644
View file @
b40e6ffa
# serializers.py
from
kombu.serialization
import
register
import
pickle
def
pickle_v2_dumps
(
obj
):
return
pickle
.
dumps
(
obj
,
protocol
=
2
)
def
pickle_v2_loads
(
s
):
return
pickle
.
loads
(
s
)
register
(
'pickle_v2'
,
pickle_v2_dumps
,
pickle_v2_loads
,
content_type
=
'application/x-pickle'
,
content_encoding
=
'binary'
,
)
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