Exemplo n.º 1
0
        ///  <summary>
        ///     原子计数器递增
        ///  </summary>
        ///  <param name="tableName">表名</param>
        ///  <param name="rowKey">rowKey</param>
        ///  <param name="column">列名</param>
        /// <param name="iep">对应的Region的服务器地址</param>
        /// <param name="value">递增值</param>
        ///  <returns>递增后的结果</returns>
        internal long AtomicIncrement(string tableName, byte[] rowKey, string column, IPEndPoint iep, long @value = 1)
        {
            if (string.IsNullOrEmpty(tableName))
            {
                throw new ArgumentNullException("tableName");
            }
            IThriftConnectionAgent agent = _connectionPool.GetChannel(iep, "RegionServer", _protocolStack, _transactionManager);

            if (agent == null)
            {
                throw new NoConnectionException();
            }
            Exception ex     = null;
            long      result = 0;
            ThriftMessageTransaction transaction    = agent.CreateTransaction();
            AutoResetEvent           autoResetEvent = new AutoResetEvent(false);

            transaction.ResponseArrived += delegate(object sender, LightSingleArgEventArgs <ThriftMessage> e)
            {
                AtomicIncrementResponseMessage rspMsg = (AtomicIncrementResponseMessage)e.Target;
                if (rspMsg.IOErrorMessage != null)
                {
                    ex = new IOErrorException(rspMsg.IOErrorMessage.Reason);
                }
                else if (rspMsg.IllegalArgumentErrorMessage != null)
                {
                    ex = new IOErrorException(rspMsg.IllegalArgumentErrorMessage.Reason);
                }
                result = rspMsg.Success;
                autoResetEvent.Set();
            };
            transaction.Timeout += delegate
            {
                ex = new CommunicationTimeoutException(transaction.SequenceId);
                autoResetEvent.Set();
            };
            transaction.Failed += delegate
            {
                ex = new CommunicationFailException(transaction.SequenceId);
                autoResetEvent.Set();
            };
            AtomicIncrementRequestMessage reqMsg = new AtomicIncrementRequestMessage {
                TableName = tableName, RowKey = rowKey, Column = column, Value = @value
            };

            transaction.SendRequest(reqMsg);
            autoResetEvent.WaitOne();
            if (ex != null)
            {
                throw ex;
            }
            return(result);
        }
Exemplo n.º 2
0
 ///  <summary>
 /// 	原子计数器递增
 ///  </summary>
 ///  <param name="tableName">表名</param>
 ///  <param name="rowKey">rowKey</param>
 ///  <param name="column">列名</param>
 /// <param name="iep">对应的Region的服务器地址</param>
 /// <param name="value">递增值</param>
 ///  <returns>递增后的结果</returns>
 internal long AtomicIncrement(string tableName, byte[] rowKey, string column, IPEndPoint iep, long @value = 1)
 {
     if (string.IsNullOrEmpty(tableName)) throw new ArgumentNullException("tableName");
     IThriftConnectionAgent agent = _connectionPool.GetChannel(iep, "RegionServer", _protocolStack, _transactionManager);
     if (agent == null) throw new NoConnectionException();
     Exception ex = null;
     long result = 0;
     ThriftMessageTransaction transaction = agent.CreateTransaction();
     AutoResetEvent autoResetEvent = new AutoResetEvent(false);
     transaction.ResponseArrived += delegate(object sender, LightSingleArgEventArgs<ThriftMessage> e)
     {
         AtomicIncrementResponseMessage rspMsg = (AtomicIncrementResponseMessage)e.Target;
         if (rspMsg.IOErrorMessage != null) ex = new IOErrorException(rspMsg.IOErrorMessage.Reason);
         else if (rspMsg.IllegalArgumentErrorMessage != null) ex = new IOErrorException(rspMsg.IllegalArgumentErrorMessage.Reason);
         result = rspMsg.Success;
         autoResetEvent.Set();
     };
     transaction.Timeout += delegate
     {
         ex = new CommunicationTimeoutException(transaction.SequenceId);
         autoResetEvent.Set();
     };
     transaction.Failed += delegate
     {
         ex = new CommunicationFailException(transaction.SequenceId);
         autoResetEvent.Set();
     };
     AtomicIncrementRequestMessage reqMsg = new AtomicIncrementRequestMessage { TableName = tableName, RowKey = rowKey, Column = column, Value = @value};
     transaction.SendRequest(reqMsg);
     autoResetEvent.WaitOne();
     if (ex != null) throw ex;
     return result;
 }