mirror of
https://github.com/ckaczor/sqltoolsservice.git
synced 2026-01-16 17:23:38 -05:00
Move managed parser into its own project (test code coverage) (#774)
* Created New ManagedBatchParser project in .NetStandard * Addressing PR Comments * Resolve 'No Repository' warning. * Move batch parser tests to integrations test project * Fix SLN file
This commit is contained in:
@@ -0,0 +1,97 @@
|
||||
//
|
||||
// Copyright (c) Microsoft. All rights reserved.
|
||||
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
|
||||
//
|
||||
|
||||
using System;
|
||||
using System.Data.SqlClient;
|
||||
|
||||
namespace Microsoft.SqlTools.ServiceLayer.Connection.ReliableConnection
|
||||
{
|
||||
public abstract partial class RetryPolicy
|
||||
{
|
||||
public interface IErrorDetectionStrategy
|
||||
{
|
||||
/// <summary>
|
||||
/// Determines whether the specified exception represents a temporary failure that can be compensated by a retry.
|
||||
/// </summary>
|
||||
/// <param name="ex">The exception object to be verified.</param>
|
||||
/// <returns>True if the specified exception is considered as temporary, otherwise false.</returns>
|
||||
bool CanRetry(Exception ex);
|
||||
|
||||
/// <summary>
|
||||
/// Determines whether the specified exception can be ignored.
|
||||
/// </summary>
|
||||
/// <param name="ex">The exception object to be verified.</param>
|
||||
/// <returns>True if the specified exception is considered as non-harmful.</returns>
|
||||
bool ShouldIgnoreError(Exception ex);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Base class with common retry logic. The core behavior for retrying non SqlExceptions is the same
|
||||
/// across retry policies
|
||||
/// </summary>
|
||||
public abstract class ErrorDetectionStrategyBase : IErrorDetectionStrategy
|
||||
{
|
||||
public bool CanRetry(Exception ex)
|
||||
{
|
||||
if (ex != null)
|
||||
{
|
||||
SqlException sqlException;
|
||||
if ((sqlException = ex as SqlException) != null)
|
||||
{
|
||||
return CanRetrySqlException(sqlException);
|
||||
}
|
||||
if (ex is InvalidOperationException)
|
||||
{
|
||||
// Operations can throw this exception if the connection is killed before the write starts to the server
|
||||
// However if there's an inner SqlException it may be a CLR load failure or other non-transient error
|
||||
if (ex.InnerException != null
|
||||
&& ex.InnerException is SqlException)
|
||||
{
|
||||
return CanRetry(ex.InnerException);
|
||||
}
|
||||
return true;
|
||||
}
|
||||
if (ex is TimeoutException)
|
||||
{
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
public bool ShouldIgnoreError(Exception ex)
|
||||
{
|
||||
if (ex != null)
|
||||
{
|
||||
SqlException sqlException;
|
||||
if ((sqlException = ex as SqlException) != null)
|
||||
{
|
||||
return ShouldIgnoreSqlException(sqlException);
|
||||
}
|
||||
if (ex is InvalidOperationException)
|
||||
{
|
||||
// Operations can throw this exception if the connection is killed before the write starts to the server
|
||||
// However if there's an inner SqlException it may be a CLR load failure or other non-transient error
|
||||
if (ex.InnerException != null
|
||||
&& ex.InnerException is SqlException)
|
||||
{
|
||||
return ShouldIgnoreError(ex.InnerException);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
protected virtual bool ShouldIgnoreSqlException(SqlException sqlException)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
protected abstract bool CanRetrySqlException(SqlException sqlException);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user