优化Partial Content缓存

internal/caches/partial_ranges.go

@@ -88,6 +88,23 @@ func (this *PartialRanges) Contains(begin int64, end int64) bool {
 	return false
 }
 
+// Nearest 查找最近的某个范围
+func (this *PartialRanges) Nearest(begin int64, end int64) (r [2]int64, ok bool) {
+	if len(this.Ranges) == 0 {
+		return
+	}
+
+	// TODO 使用二分法查找改进性能
+	for _, r2 := range this.Ranges {
+		if r2[0] <= begin && r2[1] > begin {
+			r = [2]int64{begin, this.min(end, r2[1])}
+			ok = true
+			return
+		}
+	}
+	return
+}
+
 // AsJSON 转换为JSON
 func (this *PartialRanges) AsJSON() ([]byte, error) {
 	return json.Marshal(this)

internal/caches/partial_ranges_test.go

@@ -97,6 +97,34 @@ func TestNewPartialRanges5(t *testing.T) {
 	logs.PrintAsJSON(r.Ranges, t)
 }
 
+func TestNewPartialRanges_Nearest(t *testing.T) {
+	{
+		// nearby
+		var r = caches.NewPartialRanges()
+		r.Add(301, 400)
+		r.Add(401, 500)
+		r.Add(501, 600)
+
+		t.Log(r.Nearest(100, 200))
+		t.Log(r.Nearest(300, 350))
+		t.Log(r.Nearest(302, 350))
+	}
+
+	{
+		// nearby
+		var r = caches.NewPartialRanges()
+		r.Add(301, 400)
+		r.Add(450, 500)
+		r.Add(550, 600)
+
+		t.Log(r.Nearest(100, 200))
+		t.Log(r.Nearest(300, 350))
+		t.Log(r.Nearest(302, 350))
+		t.Log(r.Nearest(302, 440))
+		t.Log(r.Nearest(302, 1000))
+	}
+}
+
 func TestNewPartialRanges_AsJSON(t *testing.T) {
 	var r = caches.NewPartialRanges()
 	for j := 0; j < 1000; j++ {

internal/caches/reader.go

@@ -39,7 +39,7 @@ type Reader interface {
 	BodySize() int64
 
 	// ContainsRange 是否包含某个区间内容
-	ContainsRange(r rangeutils.Range) bool
+	ContainsRange(r rangeutils.Range) (r2 rangeutils.Range, ok bool)
 
 	// Close 关闭
 	Close() error

internal/caches/reader_file.go

@@ -334,8 +334,8 @@ func (this *FileReader) ReadBodyRange(buf []byte, start int64, end int64, callba
 }
 
 // ContainsRange 是否包含某些区间内容
-func (this *FileReader) ContainsRange(r rangeutils.Range) bool {
-	return true
+func (this *FileReader) ContainsRange(r rangeutils.Range) (r2 rangeutils.Range, ok bool) {
+	return r, true
 }
 
 func (this *FileReader) Close() error {

internal/caches/reader_memory.go

@@ -199,8 +199,8 @@ func (this *MemoryReader) ReadBodyRange(buf []byte, start int64, end int64, call
 }
 
 // ContainsRange 是否包含某些区间内容
-func (this *MemoryReader) ContainsRange(r rangeutils.Range) bool {
-	return true
+func (this *MemoryReader) ContainsRange(r rangeutils.Range) (r2 rangeutils.Range, ok bool) {
+	return r, true
 }
 
 func (this *MemoryReader) Close() error {

internal/caches/reader_partial_file.go

@@ -113,8 +113,8 @@ func (this *PartialFileReader) InitAutoDiscard(autoDiscard bool) error {
 
 // ContainsRange 是否包含某些区间内容
 // 这里的 r 是已经经过格式化的
-func (this *PartialFileReader) ContainsRange(r rangeutils.Range) bool {
-	return this.ranges.Contains(r.Start(), r.End())
+func (this *PartialFileReader) ContainsRange(r rangeutils.Range) (r2 rangeutils.Range, ok bool) {
+	return this.ranges.Nearest(r.Start(), r.End())
 }
 
 // MaxLength 获取区间最大长度