add main.scala

Author: daewon

.gitignore

@@ -1,5 +1,6 @@
 *.class
 *.log
+.ensime*
 
 # sbt specific
 dist/*

build.sbt

@@ -0,0 +1,7 @@
+name := "daewonHttp"
+ 
+version := "1.0"
+ 
+scalaVersion := "2.9.1"
+ 
+

project/plugins.sbt

@@ -0,0 +1 @@
+addSbtPlugin("org.ensime" % "ensime-sbt-cmd" % "0.1.0")

src/main/scala/common/main.scala

@@ -0,0 +1,171 @@
+/**
+ * la-scala static http server
+ * 참고: http://twitter.github.com/scala_school/concurrency.html
+ */
+package daewon.http
+
+import java.net._
+import java.util.concurrent._
+import java.util.Date
+import scala.io._
+import java.io._
+
+/**
+ * 서버 메인
+ * Excutors 스레드 풀 사용
+ */ 
+class HttpServer(docRoot: String, port: Int) {
+  val serverSocket = new ServerSocket(port) 
+  val pool: ExecutorService = Executors.newCachedThreadPool 
+	
+  def run() {
+    try {
+      while (true) {
+        val socket = serverSocket.accept() // block here
+        pool.execute(new Handler(docRoot, socket))
+      }
+    } finally {
+      pool.shutdown()
+    }
+  }
+}
+
+/**
+ * Http 상수 모음
+ */
+object HttpConst {
+	private val htmlExt = Set("html", "htm")
+  val SP = " "
+  val CRLF = "\r\n"
+  val BODYDELIMITER = CRLF + CRLF
+
+	object Method {
+		val GET = "GET"
+		val POST = "POST"
+		val DELETE = "DELETE"
+		val PUT = "PUT"
+	}
+
+	def isHTMLPage(path: String) = htmlExt(path.toLowerCase().split("\\.").last)
+}
+
+/**
+ * 응답 핸들러
+ * 200 OK, 404 NotFound 에 대한 처리를 담당
+ */
+class Handler(docRoot: String, socket: Socket) extends Runnable {
+	val iterator = io.Source.fromInputStream(socket.getInputStream)(Codec.UTF8).getLines()
+
+  def run() {
+		try {
+			if (iterator.isEmpty){
+				log("cannot read input stream"); return
+			}
+			// 01. input stream으로부터 첫줄 읽어내서 메소드, 경로 등 파싱
+			val Array(method, path, version) = iterator.next().split(HttpConst.SP)
+
+			log(method, path, version)
+
+			val file = new File(docRoot + path)
+			val os = socket.getOutputStream
+
+			method match {
+				case HttpConst.Method.GET => {
+					if (!file.exists()) NotFound(os)
+					else {
+						if (HttpConst.isHTMLPage(path)) OK(file, os)
+						else OKBinary(file, os)
+					}
+				}
+				case _ => throw new Error("unsupported method ")
+			}
+		} finally {
+			socket.close
+		}
+	}
+}
+
+/*
+ * 응답 모음
+ * 응답 헤더 모음 쪽으로 리팩터링 필요
+ */ 
+abstract class Response {
+	val BUF_SIZE = 1024 * 1024
+
+	/**
+	 * 날자는 매 응답마다 변경되어야 한다.
+	 */
+	def date = "Date: " + new Date().toString + HttpConst.CRLF
+
+	/**
+	 * 파일을 메모리에 들지 않기 위해 사용
+	 */
+	def writeFileAsStream(file: File, os: OutputStream){
+		val fs = new FileInputStream(file)	
+		val buf: Array[Byte] = new Array[Byte](BUF_SIZE)
+
+		try {
+			Stream.continually(fs.read(buf, 0, BUF_SIZE)).takeWhile(_ != -1).foreach( cnt => os.write(buf, 0, cnt))
+		} catch{
+			case e: IOException => log("can't write outputstream")
+		} finally{
+			fs.close
+		}
+	}
+}
+
+/**
+ * 404 에러 반환
+ */ 
+case class NotFound(os: OutputStream) extends Response {
+val head = """HTTP/1.1 404 Not Found
+Connection: Close""" + HttpConst.BODYDELIMITER
+
+	log(head)
+
+	os.write( (head + date) getBytes )
+}
+
+/**
+ * 200 OK with HTML 페이지
+ */
+case class OK(file: File, os: OutputStream) extends Response {
+val head = """HTTP/1.1 200 OK
+Connection: Close
+Content-Type: text/html; charset=utf-8
+""" + "Content-Length: " + file.length + HttpConst.BODYDELIMITER 
+
+	log(head)
+
+	os.write(head getBytes)
+	this.writeFileAsStream(file, os)
+}
+
+/**
+ * 200 OK 바이너리 (파일 다운로드)
+ */ 
+case class OKBinary(file: File, os: OutputStream) extends Response {
+val head = """HTTP/1.1 200 OK
+Connection: Close
+""" +  "Content-type : application/octet-stream; " + 	file.getName + HttpConst.CRLF + "Content-Length: " + file.length + HttpConst.CRLF + "Content-Disposition: attachment; " + file.getName + HttpConst.BODYDELIMITER
+
+	log(head)
+
+	os.write(head getBytes)
+	this.writeFileAsStream(file, os)
+}
+
+/**
+ * 추후 로그 변경을 위해서
+ */
+case class log(msg: String*) {
+	println( msg.mkString(", ") )
+}
+
+/**
+ * Main
+ */ 
+object Main extends App {
+	val docRoot = "."
+	(new HttpServer(docRoot, 8080)).run	
+}

src/main/scala/common/package.scala

@@ -0,0 +1,37 @@
+import java.io.File
+
+package object common {
+  /** `???` can be used for marking methods that remain to be implemented.
+   *  @throws  An `Error`
+   */
+  def ??? : Nothing = throw new Error("an implementation is missing")
+
+  type ??? = Nothing
+  type *** = Any
+
+  
+  /**
+   * Get a child of a file. For example,
+   * 
+   *   subFile(homeDir, "b", "c")
+   * 
+   * corresponds to ~/b/c
+   */
+  def subFile(file: File, children: String*) = {
+    children.foldLeft(file)((file, child) => new File(file, child))
+  }
+
+  /**
+   * Get a resource from the `src/main/resources` directory. Eclipse does not copy
+   * resources to the output directory, then the class loader cannot find them.
+   */
+  def resourceAsStreamFromSrc(resourcePath: List[String]): Option[java.io.InputStream] = {
+    val classesDir = new File(getClass.getResource(".").toURI)
+    val projectDir = classesDir.getParentFile.getParentFile.getParentFile.getParentFile
+    val resourceFile = subFile(projectDir, ("src" :: "main" :: "resources" :: resourcePath): _*)
+    if (resourceFile.exists)
+      Some(new java.io.FileInputStream(resourceFile))
+    else
+      None
+  }
+}