using System.Collections.Concurrent;

namespace Gott.Services;

public class StreamingTracker
{
    private readonly ConcurrentDictionary<string, string> _streaming = new ConcurrentDictionary<string, string>();

    public void Start(string sessionId, string chainId) => _streaming[sessionId] = chainId;

    public void Stop(string sessionId) => _streaming.TryRemove(sessionId, out _);

    public bool IsStreaming(string sessionId) => _streaming.ContainsKey(sessionId);

    public string GetChainId(string sessionId) => _streaming.TryGetValue(sessionId, out var chainId) ? chainId : null;
}