implement close()

MoeOrganization · Apr 4, 2013 · 9033d41 · 9033d41
1 parent 8832e2f
commit 9033d41
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/main/scala/org/moe/runtime/builtins/IOClass.scala b/src/main/scala/org/moe/runtime/builtins/IOClass.scala
@@ -137,6 +137,15 @@ object IOClass {
       )
     )
 
+    ioClass.addMethod(
+      new MoeMethod(
+        "close",
+        new MoeSignature(),
+        env,
+        (e) => self(e).close(r)
+      )
+    )
+
     /**
      * NOTE:
      * This list needs some work, I have been punting on

diff --git a/src/main/scala/org/moe/runtime/nativeobjects/MoeIOObject.scala b/src/main/scala/org/moe/runtime/nativeobjects/MoeIOObject.scala
@@ -56,6 +56,12 @@ class MoeIOObject(
     Source.fromFile(file).mkString
   )
 
+  def close (r: MoeRuntime): MoeUndefObject = {
+    reader.close()
+    writer.close()
+    r.NativeObjects.getUndef
+  }
+
   // MoeNativeObject overrides
 
   override def copy = new MoeIOObject(