Browse Source

[Feature] 支持es存储命令

pull/968/head
ibuler 7 years ago
parent
commit
67001dd99f
  1. 3
      apps/jumpserver/settings.py
  2. 27
      apps/terminal/backends/command/es.py

3
apps/jumpserver/settings.py

@ -373,7 +373,8 @@ CAPTCHA_FOREGROUND_COLOR = '#001100'
CAPTCHA_NOISE_FUNCTIONS = ('captcha.helpers.noise_dots',) CAPTCHA_NOISE_FUNCTIONS = ('captcha.helpers.noise_dots',)
CAPTCHA_TEST_MODE = CONFIG.CAPTCHA_TEST_MODE CAPTCHA_TEST_MODE = CONFIG.CAPTCHA_TEST_MODE
COMMAND_STORAGE_BACKEND = 'terminal.backends.command.db' #COMMAND_STORAGE_BACKEND = 'terminal.backends.command.db'
COMMAND_STORAGE_BACKEND = 'terminal.backends.command.es'
# Django bootstrap3 setting, more see http://django-bootstrap3.readthedocs.io/en/latest/settings.html # Django bootstrap3 setting, more see http://django-bootstrap3.readthedocs.io/en/latest/settings.html
BOOTSTRAP3 = { BOOTSTRAP3 = {

27
apps/terminal/backends/command/es.py

@ -0,0 +1,27 @@
# -*- coding: utf-8 -*-
#
from jms_es_storage import ESStore
from .base import CommandBase
class CommandStore(CommandBase, ESStore):
def __init__(self):
ESStore.__init__(self, hosts=["http://elastic:changeme@localhost:9200"])
def save(self, command):
return ESStore.save(self, command)
def bulk_save(self, commands):
return ESStore.bulk_save(self, commands)
def filter(self, date_from=None, date_to=None,
user=None, asset=None, system_user=None,
input=None, session=None):
data = ESStore.filter(
self, date_from=date_from, date_to=date_to,
user=user, asset=asset, system_user=system_user,
input=input, session=session
)
return [item["_source"] for item in data["hits"] if item]
Loading…
Cancel
Save