summaryrefslogtreecommitdiffhomepage
path: root/go/response.go
diff options
context:
space:
mode:
authorMax Romanov <max.romanov@nginx.com>2019-12-24 17:59:52 +0300
committerMax Romanov <max.romanov@nginx.com>2019-12-24 17:59:52 +0300
commitf5a2984acf9a562b6b1eb45a21d9af2be2415659 (patch)
tree5e61bb9fef4adc52f8e9a46803dc065b4b0faaf2 /go/response.go
parenta427ecd4c547958079cce2dae13060b2d60aa4d4 (diff)
downloadunit-f5a2984acf9a562b6b1eb45a21d9af2be2415659.tar.gz
unit-f5a2984acf9a562b6b1eb45a21d9af2be2415659.tar.bz2
Go: moving source files to the root of the project.
This patch includes packaging changes related to files move.
Diffstat (limited to 'go/response.go')
-rw-r--r--go/response.go87
1 files changed, 87 insertions, 0 deletions
diff --git a/go/response.go b/go/response.go
new file mode 100644
index 00000000..767d66b7
--- /dev/null
+++ b/go/response.go
@@ -0,0 +1,87 @@
+/*
+ * Copyright (C) Max Romanov
+ * Copyright (C) NGINX, Inc.
+ */
+
+package unit
+
+/*
+#include "nxt_cgo_lib.h"
+*/
+import "C"
+
+import (
+ "net/http"
+)
+
+type response struct {
+ header http.Header
+ headerSent bool
+ req *http.Request
+ c_req C.uintptr_t
+}
+
+func new_response(c_req C.uintptr_t, req *http.Request) *response {
+ resp := &response{
+ header: http.Header{},
+ req: req,
+ c_req: c_req,
+ }
+
+ return resp
+}
+
+func (r *response) Header() http.Header {
+ return r.header
+}
+
+func (r *response) Write(p []byte) (n int, err error) {
+ if !r.headerSent {
+ r.WriteHeader(http.StatusOK)
+ }
+
+ res := C.nxt_cgo_response_write(r.c_req, buf_ref(p), C.uint32_t(len(p)))
+ return int(res), nil
+}
+
+func (r *response) WriteHeader(code int) {
+ if r.headerSent {
+ // Note: explicitly using Stderr, as Stdout is our HTTP output.
+ nxt_go_warn("multiple response.WriteHeader calls")
+ return
+ }
+ r.headerSent = true
+
+ // Set a default Content-Type
+ if _, hasType := r.header["Content-Type"]; !hasType {
+ r.header.Add("Content-Type", "text/html; charset=utf-8")
+ }
+
+ fields := 0
+ fields_size := 0
+
+ for k, vv := range r.header {
+ for _, v := range vv {
+ fields++
+ fields_size += len(k) + len(v)
+ }
+ }
+
+ C.nxt_cgo_response_create(r.c_req, C.int(code), C.int(fields),
+ C.uint32_t(fields_size))
+
+ for k, vv := range r.header {
+ for _, v := range vv {
+ C.nxt_cgo_response_add_field(r.c_req, str_ref(k), C.uint8_t(len(k)),
+ str_ref(v), C.uint32_t(len(v)))
+ }
+ }
+
+ C.nxt_cgo_response_send(r.c_req)
+}
+
+func (r *response) Flush() {
+ if !r.headerSent {
+ r.WriteHeader(http.StatusOK)
+ }
+}