| 460 | } |
| 461 | |
| 462 | func TestHintFilePartialRead(t *testing.T) { |
| 463 | tmpDir, err := os.MkdirTemp("", "hintfile-test") |
| 464 | if err != nil { |
| 465 | t.Fatalf("Failed to create temp dir: %v", err) |
| 466 | } |
| 467 | defer func() { _ = os.RemoveAll(tmpDir) }() |
| 468 | |
| 469 | hintPath := filepath.Join(tmpDir, "partial.hint") |
| 470 | |
| 471 | // Create a hint file with multiple entries |
| 472 | writer := &HintFileWriter{} |
| 473 | err = writer.Create(hintPath) |
| 474 | if err != nil { |
| 475 | t.Fatalf("Failed to create hint file: %v", err) |
| 476 | } |
| 477 | |
| 478 | entries := []*HintEntry{ |
| 479 | { |
| 480 | BucketId: 1, |
| 481 | KeySize: 3, |
| 482 | ValueSize: 5, |
| 483 | Timestamp: 1234567890, |
| 484 | TTL: 3600, |
| 485 | Flag: DataSetFlag, |
| 486 | Status: Committed, |
| 487 | Ds: DataStructureBTree, |
| 488 | DataPos: 100, |
| 489 | FileID: 1, |
| 490 | Key: []byte("key1"), |
| 491 | }, |
| 492 | { |
| 493 | BucketId: 2, |
| 494 | KeySize: 3, |
| 495 | ValueSize: 5, |
| 496 | Timestamp: 1234567891, |
| 497 | TTL: 3600, |
| 498 | Flag: DataSetFlag, |
| 499 | Status: Committed, |
| 500 | Ds: DataStructureBTree, |
| 501 | DataPos: 200, |
| 502 | FileID: 1, |
| 503 | Key: []byte("key2"), |
| 504 | }, |
| 505 | } |
| 506 | |
| 507 | for _, entry := range entries { |
| 508 | err = writer.Write(entry) |
| 509 | if err != nil { |
| 510 | t.Fatalf("Failed to write hint entry: %v", err) |
| 511 | } |
| 512 | } |
| 513 | |
| 514 | err = writer.Close() |
| 515 | if err != nil { |
| 516 | t.Fatalf("Failed to close hint file: %v", err) |
| 517 | } |
| 518 | |
| 519 | // Read only the first entry |