c.readRequest(ctx)讀請(qǐng)求
先配置Header最長(zhǎng)讀取時(shí)間肪获、req最長(zhǎng)讀取時(shí)間、req最大讀取長(zhǎng)度默認(rèn)6M柒傻。
RFC7230禁止\r\n參數(shù)孝赫,Url中只允許包含英文字母(a-zA-Z)、數(shù)字(0-9)红符、-_.~4個(gè)特殊字符以及所有保留字符青柄。但go net/http包放寬了這個(gè)要求劫映。
讀出Request
先構(gòu)建newTextprotoReader,由于緩沖區(qū)是對(duì)象復(fù)用的刹前,用完后要defer put泳赋。共完以以下解析任務(wù):
- 協(xié)議第一行
- URL
- Header
- Body
TextprotoReader數(shù)據(jù)結(jié)構(gòu),將字節(jié)碼Reader轉(zhuǎn)成文本Reader喇喉。
type Reader struct
R *bufio.Reader
dot *dotReader
buf []byte // a re-usable buffer for readContinuedLineSlice
}
type dotReader struct {
r *Reader
state int
}
//struct之間相互引用
第一步祖今,從第一行解析出method uri prototype。
第二步解析URL拣技。url.URL數(shù)據(jù)結(jié)構(gòu):
type URL struct {
Scheme string
Opaque string
User *Userinfo
Host string
Path string
RawPath string
ForceQuery bool
RawQuery string
Fragment string
}
解析Scheme千诬,協(xié)議前綴(小寫(xiě))。有查詢參數(shù)?膏斤,則配置url.ForceQuery url.RawQuery徐绑。有認(rèn)證信息///...//,則解析url.User url.Host莫辨。最后配置url.Path和url.RawPath傲茄,如果Path==RawPath,則RawPath=""沮榜。
第三步解析MIMEHeader盘榨。
第四步readTransfer。重新配置如下參數(shù):RequestMethod ProtoMajor ProtoMinor Header Trailer ContentLength Close蟆融。對(duì)于Body草巡,如果encodings支持chunked,讀取流用chunkedReader包裹型酥。默認(rèn)情況用LimitedReader山憨,無(wú)body賦空的struct{}。
Request檢查并補(bǔ)全
以下情況返回非空err弥喉,示得到正確的請(qǐng)求:
- 請(qǐng)求太大郁竟,超過(guò)了readLimitSize。
- 不支持的0.x和2+的HTTP協(xié)議版本档桃。
- 請(qǐng)求Header中沒(méi)有Host枪孩、超過(guò)1個(gè)或不合法。合法字符集為httpguts.validHostByte藻肄。
- Header的Key或Value不合法。Key的要求必須一httpguts.isTokenTable字符集拒担,Value的要求不可以是控制字符集嘹屯。
最后配置req.ctx req.RemoteAddr req.TLS body.doEarlyClose = true。
構(gòu)建Response并包裝chunkWriter
構(gòu)建Response:
w = &response{
conn: c,
cancelCtx: cancelCtx,
req: req,
reqBody: req.Body,
handlerHeader: make(Header),
contentLength: -1,
closeNotifyCh: make(chan bool, 1),
wants10KeepAlive: req.wantsHttp10KeepAlive(),
wantsClose: req.wantsClose(),
}
其中closeNotifyCh必須在構(gòu)建時(shí)初始化从撼,沒(méi)有content所以先置contentLength為-1州弟。
配置w.cw并被w.w包裹钧栖。w.cw緩沖默認(rèn)大小2M。
請(qǐng)求未正確獲取的情況
獲取Request可能出現(xiàn)如下錯(cuò)誤:
- 請(qǐng)求超過(guò)readLimitSize婆翔,返回431錯(cuò)誤拯杠,并刷新conn的讀寫(xiě)緩沖。
- basRequest錯(cuò)誤啃奴,返回400錯(cuò)誤潭陪。包括網(wǎng)絡(luò)原因?qū)е碌淖x取錯(cuò)誤。
w.finishRequest()
先上響應(yīng)數(shù)據(jù)結(jié)構(gòu):
type response struct {
conn *conn
req *Request
reqBody io.ReadCloser
cancelCtx context.CancelFunc
wroteHeader bool
wroteContinue bool
wants10KeepAlive bool
wantsClose bool
w *bufio.Writer
cw chunkWriter
handlerHeader Header
calledHeader bool
written int64
contentLength int64
status int
closeAfterReply bool
requestBodyLimitHit bool
trailers []string
handlerDone atomicBool
dateBuf [len(TimeFormat)]byte
clenBuf [10]byte
statusBuf [3]byte
closeNotifyCh chan bool
didCloseNotify int32
}
response字段可以分類(lèi)為:大對(duì)象最蕾、緩沖依溯、KV對(duì)或bool型的狀態(tài)參數(shù)。
大對(duì)象有:
- conn為T(mén)CP連接指針
- req為對(duì)應(yīng)的請(qǐng)求指針
- reqBody為req.Body
- cancelCtx為連接上下文context
狀態(tài)字段:
- requestBodyLimitHit瘟则,讀req.Body使用了maxBytesReader保護(hù)機(jī)制黎炉,當(dāng)請(qǐng)求信息過(guò)大達(dá)到上限,就會(huì)在這個(gè)參數(shù)上反映醋拧。
- 4個(gè)Header參數(shù):
- wroteHeader響應(yīng)Header寫(xiě)入完成
- wroteContinue響應(yīng)加入了100 Continue
- wants10KeepAlive為Connection "keep-alive"連接重用參數(shù)
- wantsClose為Connection "close"連接重用參數(shù)
- 3個(gè)數(shù)據(jù): dateBuf clenBuf statusBuf對(duì)應(yīng)Date Content-Length status cod
- 3個(gè)Body相關(guān)參數(shù):contentLength written status三個(gè)字段用來(lái)校驗(yàn)響應(yīng)內(nèi)容是否寫(xiě)入完整慷嗜,如果未寫(xiě)完則不對(duì)此連接進(jìn)行復(fù)用。定位到上一篇文章serve()方法中有提到丹壕。
- trailers:指寫(xiě)完響應(yīng)之后洪添,還需要向TCP連接寫(xiě)入的數(shù)據(jù)
- handlerDone 表示handler是否存在
- handlerHeader calledHeader保存Header
緩沖區(qū)字段
chunkWriter數(shù)據(jù)結(jié)構(gòu):
type chunkWriter struct {
res *response
header Header
wroteHeader bool
chunking bool
}
chunkWriter包裹了Response,功能之一是完成Header設(shè)置雀费,包括Content-Type Content-Length chunk-header干奢。bufio.Writer是chunkWriter是緩沖包裹。
邏輯流
共完成三層緩沖寫(xiě)入:resp.w->resp.cw->conn.bufw
handler將響應(yīng)寫(xiě)入到response.w盏袄。
調(diào)用w.w.Flush()將w寫(xiě)入到cw忿峻,注意到Flush()操作,如果未刷空緩存并報(bào)錯(cuò)辕羽,觸發(fā)拷貝操作逛尚。報(bào)錯(cuò)不會(huì)退回已寫(xiě)出的數(shù)據(jù)。
copy(b.buf[0:b.n-n], b.buf[n:b.n]
進(jìn)而調(diào)用cw.Write()刁愿,根據(jù)cw.chunking參數(shù)绰寞。
- 使能,則全部數(shù)據(jù)寫(xiě)入到TCP緩沖铣口。
- 否則返回實(shí)際向TCP緩沖的寫(xiě)入量滤钱。
putBufioWriter(w.w)清空resp.w緩沖,如果池化放回sync.pool脑题。
根據(jù)chunkWriter的定義件缸,w.cw.close()負(fù)責(zé)cw的結(jié)束工作:寫(xiě)入換行符和resp.trailers數(shù)據(jù)。
最后刷新TCP緩沖w.conn.bufw.Flush()叔遂,完成響應(yīng)包發(fā)送他炊。并正確關(guān)閉request争剿。