Unload graph when session close (#146)

analytical_engine/core/grape_instance.cc

@@ -70,6 +70,8 @@ bl::result<rpc::GraphDef> GrapeInstance::loadGraph(
     graph_def.set_key(graph_name);
     graph_def.set_directed(directed);
     graph_def.set_graph_type(rpc::DYNAMIC_PROPERTY);
+    // dynamic graph doesn't have a vineyard id
+    graph_def.set_vineyard_id(-1);
     auto* schema_def = graph_def.mutable_schema_def();
 
     schema_def->set_oid_type(

coordinator/gscoordinator/coordinator.py

@@ -304,6 +304,7 @@ def RunStep(self, request, context):  # noqa: C901
                     response.graph_def.key,
                     GraphMeta(
                         response.graph_def.key,
+                        response.graph_def.vineyard_id,
                         response.graph_def.schema_def,
                         schema_path,
                     ),
@@ -566,6 +567,11 @@ def _cleanup(self, is_dangling=False):
                         s=obj.key.encode("utf-8")
                     )
                 }
+                # dynamic graph doesn't have a vineyard id
+                if obj.vineyard_id != -1:
+                    config[types_pb2.VINEYARD_ID] = attr_value_pb2.AttrValue(
+                        i=obj.vineyard_id
+                    )
 
             if unload_type:
                 dag_def = create_single_op_dag(unload_type, config)

coordinator/gscoordinator/object_manager.py

@@ -25,9 +25,10 @@ def __init__(self, key, type, lib_path):
 
 
 class GraphMeta(object):
-    def __init__(self, key, schema, schema_path=None):
+    def __init__(self, key, vineyard_id, schema, schema_path=None):
         self.key = key
         self.type = "graph"
+        self.vineyard_id = vineyard_id
         self.schema = schema
         self.schema_path = schema_path