fix: fix the recorder

parent 60bd34d8
...@@ -10,13 +10,14 @@ import socket ...@@ -10,13 +10,14 @@ import socket
import json import json
from jms.service import AppService from jms.service import AppService
import jms_storage
from .config import Config from .config import Config
from .sshd import SSHServer from .sshd import SSHServer
from .httpd import HttpServer from .httpd import HttpServer
from .logger import create_logger from .logger import create_logger
from .tasks import TaskHandler from .tasks import TaskHandler
from .recorder import get_command_recorder_class, get_replay_recorder_class from .recorder import get_command_recorder_class, ServerReplayRecorder
from .utils import get_logger from .utils import get_logger
...@@ -109,7 +110,8 @@ class Coco: ...@@ -109,7 +110,8 @@ class Coco:
self.config.update(configs) self.config.update(configs)
def get_recorder_class(self): def get_recorder_class(self):
self.replay_recorder_class = get_replay_recorder_class(self.config) self.replay_recorder_class = ServerReplayRecorder
self.replay_recorder_class.client = jms_storage.init(self.config["REPLAY_STORAGE"])
self.command_recorder_class = get_command_recorder_class(self.config) self.command_recorder_class = get_command_recorder_class(self.config)
def new_command_recorder(self): def new_command_recorder(self):
......
...@@ -290,7 +290,7 @@ def get_command_recorder_class(config): ...@@ -290,7 +290,7 @@ def get_command_recorder_class(config):
else: else:
return ServerCommandRecorder return ServerCommandRecorder
#
def get_replay_recorder_class(config): # def get_replay_recorder_class(config):
ServerReplayRecorder.client = jms_storage.init(config["REPLAY_STORAGE"]) # ServerReplayRecorder.client = jms_storage.init(config["REPLAY_STORAGE"])
return ServerReplayRecorder # return ServerReplayRecorder
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment