Skip to content
GitLab
Explore
Sign in
Register
Primary navigation
Search or go to…
Project
R
rabbitmq_agents
Manage
Activity
Members
Labels
Plan
Issues
16
Issue boards
Milestones
Wiki
Code
Merge requests
10
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Snippets
Build
Pipelines
Jobs
Pipeline schedules
Artifacts
Deploy
Releases
Package Registry
Container Registry
Model registry
Operate
Environments
Terraform modules
Monitor
Incidents
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
rc
rabbitmq_agents
Merge requests
!146
Feat group manager
Code
Review changes
Check out branch
Download
Patches
Plain diff
Merged
Feat group manager
louistw/rabbitmq_agents:feat-group-manager
into
main
Overview
0
Commits
9
Pipelines
4
Changes
2
Merged
Bo-Chun Chen
requested to merge
louistw/rabbitmq_agents:feat-group-manager
into
main
1 year ago
Overview
0
Commits
9
Pipelines
4
Changes
2
Expand
Closes
#148 (closed)
Edited
1 year ago
by
Bo-Chun Chen
👍
1
👎
0
Merge request reports
Compare
main
version 6
8ca10435
1 year ago
version 5
fb5c52a0
1 year ago
version 4
a8845167
1 year ago
version 3
ec3b6ea5
1 year ago
version 2
ec3b6ea5
1 year ago
version 1
60507f61
1 year ago
main (base)
and
latest version
latest version
d73303a2
9 commits,
1 year ago
version 6
8ca10435
7 commits,
1 year ago
version 5
fb5c52a0
4 commits,
1 year ago
version 4
a8845167
3 commits,
1 year ago
version 3
ec3b6ea5
2 commits,
1 year ago
version 2
ec3b6ea5
3 commits,
1 year ago
version 1
60507f61
1 commit,
1 year ago
2 files
+
245
−
42
Inline
Compare changes
Side-by-side
Inline
Show whitespace changes
Show one file at a time
Files
2
Search (e.g. *.vue) (Ctrl+P)
prod_rmq_agents/group_member.py
+
99
−
42
Options
#!/usr/bin/env python
import
os
import
dataset
import
json
import
pika
import
shlex
import
rc_util
from
subprocess
import
Popen
,
PIPE
from
pathlib
import
Path
from
datetime
import
datetime
from
subprocess
import
Popen
,
PIPE
from
rc_rmq
import
RCRMQ
import
rabbit_config
as
rcfg
@@ -14,85 +14,142 @@ task = "group_member"
args
=
rc_util
.
get_args
()
logger
=
rc_util
.
get_logger
(
args
)
# Initialize db
db
=
dataset
.
connect
(
f
"
sqlite:///
{
rcfg
.
db_path
}
/user_reg.db
"
)
table
=
db
[
"
groups
"
]
# Instantiate rabbitmq object
rc_rmq
=
RCRMQ
({
"
exchange
"
:
rcfg
.
Exchange
,
"
exchange_type
"
:
"
topic
"
})
def
insert_db
(
operation
,
groupname
,
msg
):
if
operation
==
"
remove
"
:
op
=
0
elif
operation
==
"
add
"
:
op
=
1
# SQL insert
table
.
insert
(
{
"
user
"
:
msg
[
"
username
"
],
"
group
"
:
groupname
,
"
operation
"
:
op
,
"
date
"
:
datetime
.
now
(),
"
host
"
:
msg
[
"
host
"
],
"
executed_by
"
:
msg
[
"
executed_by
"
],
"
interface
"
:
msg
[
"
interface
"
],
}
)
def
group_member
(
ch
,
method
,
properties
,
body
):
"""
Properties:
correlation_id (str): The UUID for the request.
reply_to (str): The RabbitMQ queue name for reply to send to.
Message(body):
username (str): The user to be added/removed from groups.
groups (dict): A dictionary with `add` or `remove` key.
add (list): A list of groups to be added for the user.
remove (list): A list of groups to be removed for the user.
executed_by (str): The user who request the change.
host (str): Hostname where the request comes from.
interface (str): whether it
'
s from CLI or WebUI.
Returns:
status (bool): Whether or not the operation executed successfully.
errmsg (str): Detailed error message if operation failed.
task (str): The task name of the agent who handle the message.
"""
msg
=
json
.
loads
(
body
)
username
=
msg
[
"
username
"
]
action
=
msg
[
"
action
"
]
msg
[
"
task
"
]
=
task
state
=
msg
[
"
state
"
]
try
:
if
'
remove
'
in
msg
[
"
groups
"
]:
if
"
remove
"
in
msg
[
"
groups
"
]:
for
each_group
in
msg
[
"
groups
"
][
"
remove
"
]:
logger
.
debug
(
f
'
Removing user
{
username
}
from group
{
each_group
}
'
)
logger
.
debug
(
f
"
Removing user
{
username
}
from group
{
each_group
}
"
)
if
str
(
rcfg
.
bright_cm_version
).
split
(
"
.
"
)[
0
]
==
"
8
"
:
grp_remove_user_cmd
=
f
'
/cm/local/apps/cmd/bin/cmsh -n -c
"
group; removefrom
{
each_group
}
groupmembers
{
username
}
; commit;
"'
grp_remove_user_cmd
=
(
'
/cm/local/apps/cmd/bin/cmsh -n -c
"
group; removefrom
'
f
'
{
each_group
}
groupmembers
{
username
}
; commit;
"'
)
else
:
grp_remove_user_cmd
=
f
'
/cm/local/apps/cmd/bin/cmsh -n -c
"
group; removefrom
{
each_group
}
members
{
username
}
; commit;
"'
logger
.
info
(
f
'
Running command:
{
grp_remove_user_cmd
}
'
)
proc
=
Popen
(
shlex
.
split
(
grp_remove_user_cmd
),
stdout
=
PIPE
,
stderr
=
PIPE
)
out
,
err
=
proc
.
communicate
()
logger
.
debug
(
f
'
Result:
{
err
}
'
)
logger
.
info
(
f
'
User
{
username
}
is removed from
{
each_group
}
group
'
)
if
'
add
'
in
msg
[
"
groups
"
]:
grp_remove_user_cmd
=
(
'
/cm/local/apps/cmd/bin/cmsh -n -c
"
group; removefrom
'
f
'
{
each_group
}
members
{
username
}
; commit;
"'
)
logger
.
info
(
f
"
Running command:
{
grp_remove_user_cmd
}
"
)
proc
=
Popen
(
shlex
.
split
(
grp_remove_user_cmd
),
stdout
=
PIPE
,
stderr
=
PIPE
)
out
,
err
=
proc
.
communicate
()
logger
.
debug
(
f
"
Result:
{
err
}
"
)
logger
.
info
(
f
"
User
{
username
}
is removed from
{
each_group
}
group
"
)
insert_db
(
"
remove
"
,
each_group
,
msg
)
if
"
add
"
in
msg
[
"
groups
"
]:
for
each_group
in
msg
[
"
groups
"
][
"
add
"
]:
logger
.
debug
(
f
'
Adding user
{
username
}
to group
{
each_group
}
'
)
logger
.
debug
(
f
"
Adding user
{
username
}
to group
{
each_group
}
"
)
if
str
(
rcfg
.
bright_cm_version
).
split
(
"
.
"
)[
0
]
==
"
8
"
:
grp_add_user_cmd
=
f
'
/cm/local/apps/cmd/bin/cmsh -n -c
"
group; append
{
each_group
}
groupmembers
{
username
}
; commit;
"'
grp_add_user_cmd
=
(
'
/cm/local/apps/cmd/bin/cmsh -n -c
"
group; append
'
f
'
{
each_group
}
groupmembers
{
username
}
; commit;
"'
)
else
:
grp_add_user_cmd
=
f
'
/cm/local/apps/cmd/bin/cmsh -n -c
"
group; append
{
each_group
}
members
{
username
}
; commit;
"'
logger
.
info
(
f
'
Running command:
{
grp_add_user_cmd
}
'
)
proc
=
Popen
(
shlex
.
split
(
grp_add_user_cmd
),
stdout
=
PIPE
,
stderr
=
PIPE
)
out
,
err
=
proc
.
communicate
()
logger
.
debug
(
f
'
Result:
{
err
}
'
)
logger
.
info
(
f
'
User
{
username
}
is added to
{
each_group
}
group
'
)
grp_add_user_cmd
=
(
'
/cm/local/apps/cmd/bin/cmsh -n -c
"
group; append
'
f
'
{
each_group
}
members
{
username
}
; commit;
"'
)
logger
.
info
(
f
"
Running command:
{
grp_add_user_cmd
}
"
)
proc
=
Popen
(
shlex
.
split
(
grp_add_user_cmd
),
stdout
=
PIPE
,
stderr
=
PIPE
)
out
,
err
=
proc
.
communicate
()
logger
.
debug
(
f
"
Result:
{
err
}
"
)
logger
.
info
(
f
"
User
{
username
}
is added to
{
each_group
}
group
"
)
insert_db
(
"
add
"
,
each_group
,
msg
)
msg
[
"
success
"
]
=
True
except
Exception
:
msg
[
"
success
"
]
=
False
msg
[
"
errmsg
"
]
=
"
Exception raised, while adding user to group {groupname}, check the logs for stack trace
"
msg
[
"
errmsg
"
]
=
(
"
Exception raised, while adding user to group {groupname}, check
"
"
the logs for stack trace
"
)
logger
.
error
(
""
,
exc_info
=
True
)
corr_id
=
properties
.
correlation_id
reply_to
=
properties
.
reply_to
logger
.
debug
(
f
'
corr_id:
{
corr_id
}
\n
reply_to:
{
reply_to
}
'
)
logger
.
debug
(
f
"
corr_id:
{
corr_id
}
\n
reply_to:
{
reply_to
}
"
)
# send response to the callback queue
if
reply_to
:
props
=
pika
.
BasicProperties
(
correlation_id
=
corr_id
)
logger
.
debug
(
"
Sending confirmation back to reply_to
"
)
rc_rmq
.
publish_msg
(
{
"
routing_key
"
:
reply_to
,
"
props
"
:
props
,
"
msg
"
:
msg
}
{
"
routing_key
"
:
reply_to
,
"
props
"
:
props
,
"
msg
"
:
msg
}
)
else
:
print
(
"
Error: no reply_to
"
)
logger
.
debug
(
f
'
User
{
username
}
confirmation sent from
{
task
}
'
)
logger
.
debug
(
f
"
User
{
username
}
confirmation sent from
{
task
}
"
)
ch
.
basic_ack
(
delivery_tag
=
method
.
delivery_tag
)
logger
.
info
(
f
"
Start listening to queue:
{
task
}
"
)
rc_rmq
.
bind_queue
(
queue
=
task
,
routing_key
=
'
group_member.*
'
,
durable
=
True
)
rc_rmq
.
bind_queue
(
queue
=
task
,
routing_key
=
"
group_member.*
"
,
durable
=
True
)
rc_rmq
.
start_consume
(
{
"
queue
"
:
task
,
"
cb
"
:
group_member
}
)
rc_rmq
.
start_consume
({
"
queue
"
:
task
,
"
cb
"
:
group_member
})
logger
.
info
(
"
Disconnected
"
)
rc_rmq
.
disconnect
()
rc_rmq
.
disconnect
()
Loading