Method Close

internal/caches/writer_memory.go:114–158

Close 关闭

()

// Close 关闭
func (this *MemoryWriter) Close() error {
	// 需要在Locker之外
	defer this.once.Do(func() {
		this.endFunc(this.item)
	})

	if this.item == nil {
		return nil
	}

	// check content length
	if this.expectedBodySize > 0 && this.bodySize != this.expectedBodySize {
		this.storage.locker.Lock()
		delete(this.storage.valuesMap, this.hash)
		this.storage.locker.Unlock()
		return ErrUnexpectedContentLength
	}

	this.storage.locker.Lock()
	this.item.IsDone = true
	var err error
	if this.isDirty {
		if this.storage.parentStorage != nil {
			this.storage.valuesMap[this.hash] = this.item

			select {
			case this.storage.dirtyChan <- types.String(this.bodySize) + "@" + this.key:
				atomic.AddInt64(&this.storage.totalDirtySize, this.bodySize)
			default:
				// remove from values map
				delete(this.storage.valuesMap, this.hash)

				err = ErrWritingQueueFull
			}
		} else {
			this.storage.valuesMap[this.hash] = this.item
		}
	} else {
		this.storage.valuesMap[this.hash] = this.item
	}

	this.storage.locker.Unlock()

	return err
}

// Discard 丢弃
func (this *MemoryWriter) Discard() error {