// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

using System.Data.ProviderBase;
using System.Diagnostics;

namespace System.Data.Odbc
{
    internal sealed class OdbcReferenceCollection : DbReferenceCollection
    {
        internal const int Closing = 0;
        internal const int Recover = 1;

        internal const int CommandTag = 1;

        public override void Add(object value, int tag)
        {
            base.AddItem(value, tag);
        }

        protected override void NotifyItem(int message, int tag, object value)
        {
            switch (message)
            {
                case Recover:
                    if (CommandTag == tag)
                    {
                        ((OdbcCommand)value).RecoverFromConnection();
                    }
                    else
                    {
                        Debug.Fail("shouldn't be here");
                    }
                    break;
                case Closing:
                    if (CommandTag == tag)
                    {
                        ((OdbcCommand)value).CloseFromConnection();
                    }
                    else
                    {
                        Debug.Fail("shouldn't be here");
                    }
                    break;
                default:
                    Debug.Fail("shouldn't be here");
                    break;
            }
        }

        public override void Remove(object value)
        {
            base.RemoveItem(value);
        }
    }
}