Function NewBadgerCache

internal/cache/badger_cache.go:23–89

NewBadgerCache creates a new Badger-based cache.

(dir string)

// NewBadgerCache creates a new Badger-based cache.
func NewBadgerCache(dir string) (*BadgerCache, error) {
	// Ensure the directory exists
	if err := os.MkdirAll(dir, 0o750); err != nil {
		return nil, fmt.Errorf("failed to create badger directory: %w", err)
	}

	// If there's a lock file but no process using it, remove it
	lockFilePath := dir + "/LOCK"
	if _, err := os.Stat(lockFilePath); err == nil {
		// Lock file exists, check if it's stale
		if isStale, err := isLockFileStale(lockFilePath); err != nil {
			getCacheLogger().Debug("Failed to check if lock file is stale: %v", err)
		} else if isStale {
			getCacheLogger().Debug("Removing stale lock file")

			if err := os.Remove(lockFilePath); err != nil {
				return nil, fmt.Errorf("failed to remove stale lock file: %w", err)
			}
		}
	}

	// Open the Badger database with default options
	opts := badger.DefaultOptions(dir)
	// Reduce logging noise
	opts.Logger = nil
	// Use a smaller value size to reduce memory usage
	opts.ValueLogFileSize = 1 << 20 // 1MB
	// Use memory mapping for values to improve performance
	// Note: In badger v4, these options may have been renamed or removed
	// opts.ValueLogLoadingMode = badger.MemoryMap

	db, err := badger.Open(opts)
	if err != nil {
		// Check if the error is due to resource temporarily unavailable
		if os.IsExist(err) || isErrorTemporarilyUnavailable(err) {
			return nil, fmt.Errorf("failed to open badger database (likely another process is using it): %w", err)
		}

		return nil, fmt.Errorf("failed to open badger database: %w", err)
	}

	cache := &BadgerCache{
		db:     db,
		stopGC: make(chan struct{}),
	}

	// Run garbage collection in the background with proper cleanup
	go func() {
		ticker := time.NewTicker(5 * time.Minute)
		defer ticker.Stop()

		for {
			select {
			case <-ticker.C:
				err := db.RunValueLogGC(0.5) // Run GC if 50% or more space can be reclaimed
				if err != nil && err != badger.ErrNoRewrite {
					getCacheLogger().Debug("Badger value log GC failed: %v", err)
				}