优化HTTP缓存，主要是并发冲突、缓存写入不全等问题

internal/caches/writer_memory.go

@@ -14,11 +14,12 @@ type MemoryWriter struct {
 	bodySize   int64
 	status     int
 
-	hash uint64
-	item *MemoryItem
+	hash    uint64
+	item    *MemoryItem
+	endFunc func()
 }
 
-func NewMemoryWriter(m map[uint64]*MemoryItem, key string, expiredAt int64, status int, locker *sync.RWMutex) *MemoryWriter {
+func NewMemoryWriter(m map[uint64]*MemoryItem, key string, expiredAt int64, status int, locker *sync.RWMutex, endFunc func()) *MemoryWriter {
 	w := &MemoryWriter{
 		m:         m,
 		key:       key,
@@ -28,38 +29,43 @@ func NewMemoryWriter(m map[uint64]*MemoryItem, key string, expiredAt int64, stat
 			ExpiredAt: expiredAt,
 			Status:    status,
 		},
-		status: status,
+		status:  status,
+		endFunc: endFunc,
 	}
 	w.hash = w.calculateHash(key)
 
 	return w
 }
 
-// 写入数据
+// WriteHeader 写入数据
 func (this *MemoryWriter) WriteHeader(data []byte) (n int, err error) {
 	this.headerSize += int64(len(data))
 	this.item.HeaderValue = append(this.item.HeaderValue, data...)
 	return len(data), nil
 }
 
-// 写入数据
+// Write 写入数据
 func (this *MemoryWriter) Write(data []byte) (n int, err error) {
 	this.bodySize += int64(len(data))
 	this.item.BodyValue = append(this.item.BodyValue, data...)
 	return len(data), nil
 }
 
-// 数据尺寸
+// HeaderSize 数据尺寸
 func (this *MemoryWriter) HeaderSize() int64 {
 	return this.headerSize
 }
 
+// BodySize 主体内容尺寸
 func (this *MemoryWriter) BodySize() int64 {
 	return this.bodySize
 }
 
-// 关闭
+// Close 关闭
 func (this *MemoryWriter) Close() error {
+	// 需要在Locker之外
+	defer this.endFunc()
+
 	if this.item == nil {
 		return nil
 	}
@@ -72,25 +78,28 @@ func (this *MemoryWriter) Close() error {
 	return nil
 }
 
-// 丢弃
+// Discard 丢弃
 func (this *MemoryWriter) Discard() error {
+	// 需要在Locker之外
+	defer this.endFunc()
+	
 	this.locker.Lock()
 	delete(this.m, this.hash)
 	this.locker.Unlock()
 	return nil
 }
 
-// Key
+// Key 获取Key
 func (this *MemoryWriter) Key() string {
 	return this.key
 }
 
-// 过期时间
+// ExpiredAt 过期时间
 func (this *MemoryWriter) ExpiredAt() int64 {
 	return this.expiredAt
 }
 
-// 内容类型
+// ItemType 内容类型
 func (this *MemoryWriter) ItemType() ItemType {
 	return ItemTypeMemory
 }