Golang實現簡單http伺服器的範例詳解

func main() {
   // 對8080埠進行監聽
   l, _ := net.Listen("tcp", ":8080")
   // 獲取和埠8080完成三次握手的tcp連線
   conn, _ := l.Accept()
   // 此時便能夠使用該連線和使用者端進行通訊
   data := make([]byte, 4096)
   // 可以從conn讀取資料緩衝區當中
   conn.Read(data)
   // 將緩衝區的資料列印處理
   print(string(data))
}

type Conn interface {
   // Read reads data from the connection.
   // Read can be made to time out and return an error after a fixed
   // time limit; see SetDeadline and SetReadDeadline.
   Read(b []byte) (n int, err error)

   // Write writes data to the connection.
   // Write can be made to time out and return an error after a fixed
   // time limit; see SetDeadline and SetWriteDeadline.
   Write(b []byte) (n int, err error)

   // Close closes the connection.
   // Any blocked Read or Write operations will be unblocked and return errors.
   Close() error
}

type Conn struct {
   rwc     net.Conn
   // bufio.Reader 提供了快取的功能，以及一些函數，方便我們解析HTTP請求體
   br      *bufio.Reader
   bw      *bufio.Writer
   // 是否已經寫入響應頭
   wroteHaeder bool
 }
 func NewConn(rwc net.Conn) *Conn {
   return &Conn{
      rwc:     rwc,
      br:      bufio.NewReader(rwc),
      bw:      bufio.NewWriter(rwc),
   }
}

type Request struct {
   // 儲存請求方法,上面例子便是GET 
   method string
   // 用於儲存請求的uri
   uri    string
   // 用於儲存http版本
   proto  string
   // http首部欄位
   Header map[string]string
}

func (c *Conn) readRequest() (*Request, error) {
   req := NewRequest()
   // 現在只支援Get請求，讀取第一行內容
   // GET /ping HTTP1.1
   line, err := c.br.ReadBytes('n')
   if err != nil {
      return req, err
   }
   var f []string
   // 按空格來進行分割,將請求行分割為三部分
   if f = strings.Split(string(line), " "); len(f) != 3 {
      return req, errors.New("http Header error")
   }
   // 獲取到GET, /ping, HTTP/1.1
   req.method, req.url, req.proto = f[0], f[1], f[2]
   // 解析請求體首部欄位
   for {   
      line, err = c.br.ReadBytes('n')
      if err != nil {
         return nil, err
      }
      // 當讀取到空行時，說明已經首部欄位已經讀取完了
      if len(strings.TrimSpace(string(line))) == 0 {
         break
      }
      //舉例，讀取connection: keep-alive,獲取第一個空格的下標
      i := bytes.IndexByte(line, ' ')
      if i == -1 {
         return nil, errors.New("header is error")
      }
      // 此時獲取到請求首部key，為connection
      key := string(line[:i-1])
      // 讀取到對應的值，這裡讀取到keep-alive
      value := strings.TrimSpace(string(line[i:]))
      // 簡單讀取頭部欄位即可
      req.Header[key] = value
   }
}

func main() {
   // 對8080埠進行監聽
   l, _ := net.Listen("tcp", ":8080")
   // 獲取和埠8080完成三次握手的tcp連線
   conn, _ := l.Accept()
   // 獲取到conn連線
   c := NewConn(conn, handler)
   // 讀取到請求體
   req, _ := c.readRequest()
   if request.uri == "/hello" {
     ....
   }else{
     .... 
   }
 }

const (
   StatusOK = 200
)
var statusText = map[int]string{
   StatusOK:                   "OK",
}

// 返回響應行
// 構造響應行 HTTP/1.1 200 OK
func (c *Conn) writeHeader(status int) error {
   if c.wroteHeader {
      return errors.New("code has been set")
   }
   c.wroteHeader = true
   var proto string
   //GET /hello HTTP/1.1
   proto = "HTTP/1.1"
   // 獲取文字描述,這裡為OK
   text, ok := statusText[status]
   if !ok {
      text = "status code " + strconv.Itoa(status)
   }
   // 寫入資料 HTTP1.1 200 OKrn
   c.bw.WriteString(proto + " " + strconv.Itoa(status) + " " + text + "rn")
   // 寫入空行
   c.bw.Write("rn")
   return nil
}
// 寫入響應資料
func (c *Conn) WriteData(data []byte) error {
   // 還沒寫入請求頭 
   if !c.wroteHeader {
       //預設狀態碼是200 OK
       c.writeHeader(StatusOK)
   }
   c.bw.Write(data)
   return nil
}

func main() {
   // 對8080埠進行監聽
   l, _ := net.Listen("tcp", ":8080")
   // 獲取和埠8080完成三次握手的tcp連線
   conn, _ := l.Accept()
   // 獲取到conn連線
   c := NewConn(conn)
   // 讀取到請求體
   req, _ := c.readRequest()
   if request.uri == "hello" {
      c.WriteData("hello")
   }else{
      c.WriteData("hello world")
   }
   // 在最後,需要將緩衝區的資料進行清空
   c.bw.Flush()
   // 因為響應沒有設定content-length,所以只有連線斷開後，
   // 瀏覽器才知道資料讀取完成了,此處需要斷開連線
   c.rwc.Close()
 }