public virtual async Task <ExecuteResult> ExecuteSql_InsertUpdate(DbCommand command, string sql, int readMaxRows, int changeMaxRows)
        {
            var           m      = GetInsertUpdateRegex().Match(sql);
            ExecuteResult result = new ExecuteResult();

            if (m.Success)
            {
                result.IsException = true;
                result.Exception   = "此SQL语句包含“" + m.Value + "”关键字!";
                return(result);
            }
            var list    = SqlStringList.SplitString(sql);
            var sqlList = list.SplitBySemi();

            try {
                result.Result = new List <ExecuteResultItem>();
                bool isRollback = false;
                using (var tran = await command.Connection.BeginTransactionAsync()) {
                    foreach (var item in sqlList)
                    {
                        ExecuteResultItem sqlResult = new ExecuteResultItem();
                        result.Result.Add(sqlResult);
                        var b = await CheckUpdateDelete(command, sqlResult, item, changeMaxRows);

                        if (b)
                        {
                            b = await ExecuteSql(command, sqlResult, item, readMaxRows);
                        }
                        if (b == false)
                        {
                            result.IsException = true;
                            result.Exception   = sqlResult.Exception;
                            result.Result      = null;
                            isRollback         = true;
                            tran.Rollback();
                            break;
                        }
                    }
                    if (isRollback == false)
                    {
                        await tran.CommitAsync();
                    }
                }
            } catch (Exception x) {
                result.IsException = true;
                result.Exception   = x.Message;
                result.Result      = null;
            }
            return(result);
        }
        public virtual async Task <ExecuteResult> ExecuteSql_AllPermissions(DbCommand command, string sql)
        {
            var           list    = SqlStringList.SplitString(sql);
            var           sqlList = list.SplitBySemi();
            ExecuteResult result  = new ExecuteResult();

            try {
                result.Result = new List <ExecuteResultItem>();
                bool isRollback = false;
                using (var tran = await command.Connection.BeginTransactionAsync()) {
                    foreach (var item in sqlList)
                    {
                        ExecuteResultItem sqlResult = new ExecuteResultItem();
                        result.Result.Add(sqlResult);

                        var b = await ExecuteSql(command, sqlResult, item, int.MaxValue);

                        if (b == false)
                        {
                            result.IsException = true;
                            result.Exception   = sqlResult.Exception;
                            result.Result      = null;
                            isRollback         = true;
                            tran.Rollback();
                            break;
                        }
                    }
                    if (isRollback == false)
                    {
                        await tran.CommitAsync();
                    }
                }
            } catch (Exception x) {
                result.IsException = true;
                result.Exception   = x.Message;
                result.Result      = null;
            }
            return(result);
        }
        public virtual async Task <ExecuteResult> ExecuteSql_Select(DbCommand command, string sql, int readMaxRows)
        {
            var           m      = GetSelectRegex().Match(sql);
            ExecuteResult result = new ExecuteResult();

            if (m.Success)
            {
                result.IsException = true;
                result.Exception   = "此SQL语句包含“" + m.Value + "”关键字!";
                return(result);
            }
            var list    = SqlStringList.SplitString(sql);
            var sqlList = list.SplitBySemi();

            try {
                result.Result = new List <ExecuteResultItem>();
                foreach (var item in sqlList)
                {
                    ExecuteResultItem sqlResult = new ExecuteResultItem();
                    result.Result.Add(sqlResult);
                    var b = await ExecuteSql(command, sqlResult, item, readMaxRows);

                    if (b == false)
                    {
                        result.IsException = true;
                        result.Exception   = sqlResult.Exception;
                        result.Result      = null;
                        break;
                    }
                }
            } catch (Exception x) {
                result.IsException = true;
                result.Exception   = x.Message;
                result.Result      = null;
            }
            return(result);
        }
Exemplo n.º 4
0
        public static SqlStringList SplitString(string sql)
        {
            SqlStringList list = new SqlStringList();

            bool isInText        = false;
            bool isInComment     = false;
            bool isInLineComment = false;
            var  c     = 'a';
            var  index = 0;
            var  text  = "";

            while (index < sql.Length)
            {
                var ch = sql[index++];
                if (isInLineComment)
                {
                    if (ch == '\r' && index < sql.Length && sql[index] == '\n')
                    {
                        isInLineComment = false;
                        list.Add(text, SqlToken.Comment);
                        list.Add("\r\n", SqlToken.Whitespace);
                        text = "";
                    }
                    else if (ch == '\r' || ch == '\n')
                    {
                        isInLineComment = false;
                        list.Add(text, SqlToken.Comment);
                        list.Add(ch.ToString(), SqlToken.Whitespace);
                        text = "";
                    }
                    else
                    {
                        text += ch;
                    }
                }
                else if (isInComment)
                {
                    text += ch;
                    if (ch == '*' && index < sql.Length && sql[index] == '/')
                    {
                        isInComment = false; list.Add(text + "/", SqlToken.Comment); text = ""; index++;
                    }
                }
                else if (isInText)
                {
                    text += ch;
                    if (c == '\\')
                    {
                        text += sql[index++];
                    }
                    else if (c == '[')
                    {
                        if (ch == ']')
                        {
                            isInText = false; list.Add(text, SqlToken.Field); text = "";
                        }
                    }
                    else if (c == '\'')
                    {
                        if (ch == '\'' && index < sql.Length && sql[index] == '\'')
                        {
                            text += sql[index++]; break;
                        }
                        if (ch == '\'')
                        {
                            isInText = false; list.Add(text, SqlToken.Field); text = "";
                        }
                    }
                    else if (c == '"')
                    {
                        if (ch == '"')
                        {
                            isInText = false; list.Add(text, SqlToken.Field); text = "";
                        }
                    }
                    else if (c == '`')
                    {
                        if (ch == '`')
                        {
                            isInText = false; list.Add(text, SqlToken.Field); text = "";
                        }
                    }
                }
                else if (ch == '#')
                {
                    isInLineComment = true; list.Add(text, SqlToken.Field); text = "#";
                }
                else if (ch == '-' && index < sql.Length && sql[index] == '-')
                {
                    isInLineComment = true; list.Add(text, SqlToken.Field); text = "--"; index++;
                }
                else if (ch == '/' && index < sql.Length && sql[index] == '*')
                {
                    isInComment = true; list.Add(text, SqlToken.Field); text = "/*"; index++;
                }
                else if (ch == '[' || ch == '\'' || ch == '"' || ch == '`')
                {
                    c = ch; isInText = true; list.Add(text, SqlToken.Field); text = ch.ToString();
                }
                else if (ch == '(' || ch == '{')
                {
                    list.Add(text, SqlToken.BracketOpen); text = ch.ToString();
                }
                else if (ch == ')' || ch == '}')
                {
                    list.Add(text, SqlToken.BracketClose); text = ch.ToString();
                }
                else if (ch == ',' || ch == '.' || ch == ':')
                {
                    list.Add(text, SqlToken.Field); list.Add(ch.ToString(), SqlToken.Other); text = "";
                }
                else if (ch == ';')
                {
                    list.Add(text, SqlToken.Field); list.Add(ch.ToString(), SqlToken.Semi); text = "";
                }
                else if (ch == '+' || ch == '-' || ch == '*' || ch == '/' || ch == '%' || ch == '=' || ch == ':' || ch == '~' || ch == '|' || ch == '&' || ch == '^' || ch == '<' || ch == '>' || ch == '!')
                {
                    list.Add(text, SqlToken.Field); list.Add(ch.ToString(), SqlToken.Operator); text = "";
                }
                else if (ch == ' ' || ch == '\t' || ch == '\r' || ch == '\n')
                {
                    list.Add(text, SqlToken.Field); list.Add(ch.ToString(), SqlToken.Whitespace); text = "";
                }
                else
                {
                    text += ch;
                }
            }
            if (isInComment || isInLineComment)
            {
                list.Add(text, SqlToken.Comment);
            }
            else
            {
                list.Add(text, SqlToken.Field);
            }

            list.RemoveAll(q => string.IsNullOrEmpty(q.SqlField));
            list.SetLayer();
            return(list);
        }