Example #1
0
        public IZooKeeper GetZooKeeper()
        {
            if (SessionFailRetryLoop.SessionForThreadHasFailed())
            {
                throw new SessionFailRetryLoop.SessionFailedException();
            }

            Exception exception;

            backgroundExceptions.TryDequeue(out exception);

            if (exception != null)
            {
                tracer.AddCount("background-exceptions", 1);
                throw exception;
            }

            bool localIsConnected = isConnected;

            //Interlocked. (ref localIsConnected, isConnected);

            if (!localIsConnected)
            {
                CheckTimeouts();
            }

            return(zooKeeper.GetZooKeeper());
        }
        /**
         * Convenience utility: creates a "session fail" retry loop calling the given proc
         *
         * @param client Zookeeper
         * @param mode how to handle session failures
         * @param proc procedure to call with retry
         * @param <T> return type
         * @return procedure result
         * @throws Exception any non-retriable errors
         */

        public static T CallWithRetry <T>(CuratorZookeeperClient client, Mode mode, Func <T> proc)
        {
            T result = default(T);
            SessionFailRetryLoop retryLoop = client.NewSessionFailRetryLoop(mode);

            retryLoop.Start();
            try{
                while (retryLoop.ShouldContinue())
                {
                    try
                    {
                        client.InternalBlockUntilConnectedOrTimedOut();

                        result = proc();
                        retryLoop.MarkComplete();
                    }
                    catch (Exception e)
                    {
                        retryLoop.TakeException(e);
                    }
                }
            }
            finally{
                retryLoop.Dispose();
            }
            return(result);
        }