def setUpClass(cls) -> None:
if is_debug:
cls.ctx = get_debug_test_context()
else:
if is_standalone:
cls.ctx = get_standalone_context()
else:
cls.ctx = get_cluster_context()
set_default_option('include_key', False)
set_default_option('total_partitions', total_partitions)
def get_cluster_context(role, options: dict = None, props_file=default_props_file, party_id=None):
if options is None:
options = {}
rp_context = rpta.get_cluster_context(options=options)
rs_options = get_option(role, props_file)
if party_id:
rs_options['self_party_id'] = str(party_id)
rs_context = RollSiteContext(roll_site_session_id, rp_ctx=rp_context,
options=rs_options)
return rs_context
def get_cluster_context(role,
options: dict = None,
props_file=default_props_file,
party_id=None,
roll_site_session_id=None):
if options is None:
options = {}
options['session_id'] = '_'.join([roll_site_session_id, role, party_id])
print("session_id: ", options['session_id'])
options[CoreConfKeys.STATIC_CONF_PATH] = props_file
rp_context = rpta.get_cluster_context(options=options)
rs_options = get_option(role, self_party_id=party_id, conf_file=props_file)
rs_context = RollSiteContext(roll_site_session_id,
rp_ctx=rp_context,
options=rs_options)
return rs_context
def setUpClass(cls) -> None:
opts = {"eggroll.session.processors.per.node": "3"}
#opts = {}
cls.ctx = get_cluster_context(options=opts)
def setUp(self):
self.ctx = get_cluster_context()