def write_data(key, path): data_info = get_data_info(key) if not data_info: return None if isinstance(data_info['data'], pd.DataFrame): _write_dataframe(data_info['data'], path) gateway.update_context_type(data_info['status'], path) elif psdf and isinstance(data_info['data'], psdf.DataFrame): dataframe = data_info['data'].toPandas() _write_dataframe(dataframe, path) gateway.update_context_type(data_info['status'], path) else: KVStoreClient.set(key, data_info['data']) gateway.update_context_type(data_info['status'], path, 'kv_store') gateway.notify_data_updated(key, data_info['status']) data_info['data'] = None # Delete data from data dictionary after write return data_info['status'].toJson()
def read_redis(key): return KVStoreClient.get(key)