mirror of
https://github.com/Security-Onion-Solutions/securityonion.git
synced 2025-12-06 17:22:49 +01:00
[fix] Don't overwrite mysql module
This commit is contained in:
@@ -1,35 +0,0 @@
|
|||||||
#!py
|
|
||||||
|
|
||||||
from MySQLdb import _mysql
|
|
||||||
import logging
|
|
||||||
import time
|
|
||||||
|
|
||||||
log = logging.getLogger(__name__)
|
|
||||||
|
|
||||||
|
|
||||||
def status(retry):
|
|
||||||
mainint = __salt__['pillar.get']('sensor:mainint', __salt__['pillar.get']('manager:mainint'))
|
|
||||||
mainip = __salt__['grains.get']('ip_interfaces').get(mainint)[0]
|
|
||||||
|
|
||||||
mysql_up = False
|
|
||||||
for i in range(0, retry):
|
|
||||||
log.debug(f'Connection attempt {i+1}')
|
|
||||||
try:
|
|
||||||
_mysql.connect(
|
|
||||||
host=mainip,
|
|
||||||
user="root",
|
|
||||||
passwd=__salt__['pillar.get']('secrets:mysql')
|
|
||||||
)
|
|
||||||
mysql_up = True
|
|
||||||
break
|
|
||||||
except _mysql.OperationalError as e:
|
|
||||||
log.debug(e)
|
|
||||||
except Exception as e:
|
|
||||||
log.error(e)
|
|
||||||
break
|
|
||||||
time.sleep(1)
|
|
||||||
|
|
||||||
if not mysql_up:
|
|
||||||
log.error(f'Could not connect to MySQL server on {mainip} after {retry} attempts.')
|
|
||||||
|
|
||||||
return mysql_up
|
|
||||||
@@ -1,4 +1,38 @@
|
|||||||
#!py
|
#!py
|
||||||
|
|
||||||
|
import logging
|
||||||
|
|
||||||
def status():
|
def status():
|
||||||
return __salt__['cmd.run']('/usr/sbin/so-status')
|
return __salt__['cmd.run']('/usr/sbin/so-status')
|
||||||
|
|
||||||
|
|
||||||
|
def mysql_conn(retry):
|
||||||
|
from MySQLdb import _mysql
|
||||||
|
import time
|
||||||
|
|
||||||
|
log = logging.getLogger(__name__)
|
||||||
|
mainint = __salt__['pillar.get']('sensor:mainint', __salt__['pillar.get']('manager:mainint'))
|
||||||
|
mainip = __salt__['grains.get']('ip_interfaces').get(mainint)[0]
|
||||||
|
|
||||||
|
mysql_up = False
|
||||||
|
for i in range(0, retry):
|
||||||
|
log.debug(f'Connection attempt {i+1}')
|
||||||
|
try:
|
||||||
|
_mysql.connect(
|
||||||
|
host=mainip,
|
||||||
|
user="root",
|
||||||
|
passwd=__salt__['pillar.get']('secrets:mysql')
|
||||||
|
)
|
||||||
|
mysql_up = True
|
||||||
|
break
|
||||||
|
except _mysql.OperationalError as e:
|
||||||
|
log.debug(e)
|
||||||
|
except Exception as e:
|
||||||
|
log.error(e)
|
||||||
|
break
|
||||||
|
time.sleep(1)
|
||||||
|
|
||||||
|
if not mysql_up:
|
||||||
|
log.error(f'Could not connect to MySQL server on {mainip} after {retry} attempts.')
|
||||||
|
|
||||||
|
return mysql_up
|
||||||
@@ -98,7 +98,7 @@ so-mysql:
|
|||||||
- onchanges:
|
- onchanges:
|
||||||
- docker_container: so-mysql
|
- docker_container: so-mysql
|
||||||
module.run:
|
module.run:
|
||||||
- mysql.status:
|
- so.mysql_conn:
|
||||||
- retry: 900
|
- retry: 900
|
||||||
- onchanges:
|
- onchanges:
|
||||||
- cmd: so-mysql
|
- cmd: so-mysql
|
||||||
|
|||||||
Reference in New Issue
Block a user