如何使用DacFX注释TsqlFragment/TSqlScript中的语句
如果我有一个带有一组语句的TqlFragment或TsqlScript，如何使用DacFX注释特定的语句？没有CommentStatement或类似的东西。如何将语法树中的语句替换为注释版本的语句？
我知道这可以通过纯文本编辑或regex来完成，但此时我正在使用DacFx访问者模式扫描某些语句。所以我需要继续使用这个约束。
回答 2
Stack Overflow用户
发布于 2018-03-08 18:07:42
正如您已经发现的，注释没有脚本dom语句，我们所拥有的是一个令牌流，它类似于一个语句，但在一个稍低的级别上。
我不能百分之百确定你想要的是什么，但是如果我假设你有一个脚本，并且想要评论一个声明，那么就有可能是这样的。您说过您需要停留在访问者模式内，但是访问者不会返回评论，所以这应该可以做您想做的事情，您可以再次重新解析脚本，或者将它保留为sql，然后使用它执行您喜欢的操作：
static void Main(string[] args)
{
var sqlText = @"
create procedure something
as
select 100;
select 200
exec sp_who2;
";
var sql = new StringReader(sqlText);
var parser = new TSql140Parser(false);
IList<ParseError> errors;
var script = parser.Parse(sql, out errors);
var visitor = new visitor();
script.Accept(visitor);
TSqlParserToken startComment = new TSqlParserToken(TSqlTokenType.SingleLineComment, "/*");
TSqlParserToken endComment = new TSqlParserToken(TSqlTokenType.SingleLineComment, "*/");
var newScriptSql = "";
for (var i = 0; i < script.LastTokenIndex; i++)
{
if (i == visitor.Statement.FirstTokenIndex)
{
newScriptSql += startComment.Text;
}
if (i == visitor.Statement.LastTokenIndex)
{
newScriptSql += endComment.Text;
}
newScriptSql += script.ScriptTokenStream[i].Text;
}
script = parser.Parse(new StringReader(newScriptSql), out errors);
Console.WriteLine(newScriptSql);
}
class visitor : TSqlFragmentVisitor
{
public ExecuteStatement Statement;
public override void Visit(ExecuteStatement s)
{
Statement = s;
}
}
}这样做是传递上面的过程(将其放入代码文件或其他东西中)，并使用访问者模式查找exec，一旦我们注意到语句在脚本令牌流中开始和结束的位置，然后迭代这些令牌，将它们转换为this，当我们到达语句时，我们插入一个startComment，在语句之后插入一个endToken (注意，它在前面；所以您可能需要添加一些额外的逻辑来处理这个问题)。
我只是尝试更改原始脚本的ScriptTokenStream，但是即使它允许您使用它，如果您对它做了任何操作，那么它就会被忽略，所以认为它是不可变的(除非我遗漏了什么)。据我所知，这不起作用，但如果可以的话，它将往返到文本/解析本质：
/* - changes the token stream but if you use a generator on script, they don't appear
script.ScriptTokenStream.Insert(visitor.Statement.LastTokenIndex, endComment);
script.ScriptTokenStream.Insert(visitor.Statement.FirstTokenIndex, startComment);
*/希望能帮上忙！
边缘
Stack Overflow用户
发布于 2018-03-12 16:48:49
艾德，我重新整理了你的答案。
- 修改访问者以处理多个语句
- 更改代码，使其在访问者语句中向后循环，这样就不会损坏位置。
- 修改代码以插入注释标记
意见：
- 正如您推断的那样，sql生成器在试图写出令牌流时愉快地删除了注释。
- 以这种方式添加注释将注释中行尾的分号排除在外。
代码：
class Program
{
static void Main(string[] args)
{
var sqlText = @"
create procedure something
as
select 100;
select 200
exec sp_who;
exec sp_who2;
";
var sql = new StringReader(sqlText);
var parser = new TSql140Parser(false);
var script = parser.Parse(sql, out IList<ParseError> errors);
var visitor = new ExecVisitor();
script.Accept(visitor);
TSqlParserToken startComment = new TSqlParserToken(TSqlTokenType.SingleLineComment, "/*");
TSqlParserToken endComment = new TSqlParserToken(TSqlTokenType.SingleLineComment, "*/");
var newScriptTokenStream = new List<TSqlParserToken>(script.ScriptTokenStream);
for(var i = visitor.Statements.Count - 1; i >= 0; i--)
{
var stmt = visitor.Statements[i];
newScriptTokenStream.Insert(stmt.LastTokenIndex, endComment);
newScriptTokenStream.Insert(stmt.FirstTokenIndex, startComment);
}
var newFragment = parser.Parse(newScriptTokenStream, out errors);
Console.WriteLine(GetScript(newFragment.ScriptTokenStream));
}
private static string GetScript(IList<TSqlParserToken> tokenStream)
{
var sb = new StringBuilder();
foreach(var t in tokenStream)
{
sb.Append(t.Text);
}
return sb.ToString();
}
}
class ExecVisitor : TSqlFragmentVisitor
{
public IList<ExecuteStatement> Statements { get; set; } = new List<ExecuteStatement>();
public override void Visit(ExecuteStatement s)
{
Statements.Add(s);
}
}结果：
create procedure something
as
select 100;
select 200
/*exec sp_who*/;
/*exec sp_who2*/;https://stackoverflow.com/questions/49178729
复制相似问题
腾讯云开发者
