// StringBuilder s = new StringBuilder();
// stmt.accept(new MySqlOutputAstVisitor(s));
// s.toString();
/// <exception cref="System.Exception" />
public override string GetSql()
{
var sql = "select id,member_id,gmt_create from offer where member_id in ('22')";
stmt = SqlParserDelegate.Parse(sql);
return("select id,member_id,gmt_create from offer where member_id in ('1','22','333','1124','4525')");
}
// StringBuilder s = new StringBuilder();
// stmt.accept(new MySqlOutputAstVisitor(s));
// s.toString();
/// <exception cref="System.Exception" />
public override string GetSql()
{
var sql = "insert into xoffer (member_id, gmt_create) values ('1','2001-09-13 20:20:33')";
stmt = SqlParserDelegate.Parse(sql);
return("insert into xoffer (member_id, gmt_create) values ('1','2001-09-13 20:20:33')");
}
//private static readonly Logger Logger = Logger.GetLogger(typeof(ServerRouter));
public static RouteResultset Route(SchemaConfig schema,
string stmt,
string charset,
object info)
{
var rrs = new RouteResultset(stmt);
// 检查是否含有cobar hint
var prefixIndex = HintRouter.IndexOfPrefix(stmt);
if (prefixIndex >= 0)
{
HintRouter.RouteFromHint(info, schema, rrs, prefixIndex, stmt);
return(rrs);
}
// 检查schema是否含有拆分库
if (schema.IsNoSharding)
{
if (schema.IsKeepSqlSchema)
{
var ast = SqlParserDelegate.Parse(stmt, charset ?? MySqlParser.DefaultCharset);
var visitor = new PartitionKeyVisitor(schema.Tables);
visitor.SetTrimSchema(schema.Name);
ast.Accept(visitor);
if (visitor.IsSchemaTrimmed())
{
stmt = GenSql(ast, stmt);
}
}
var nodes = new RouteResultsetNode[1];
nodes[0] = new RouteResultsetNode(schema.DataNode, stmt);
rrs.Nodes = nodes;
return(rrs);
}
// 生成和展开AST
var ast1 = SqlParserDelegate.Parse(stmt, charset ?? MySqlParser.DefaultCharset);
var visitor1 = new PartitionKeyVisitor(schema.Tables);
visitor1.SetTrimSchema(schema.IsKeepSqlSchema ? schema.Name : null);
ast1.Accept(visitor1);
// 如果sql包含用户自定义的schema，则路由到default节点
if (schema.IsKeepSqlSchema && visitor1.IsCustomedSchema())
{
if (visitor1.IsSchemaTrimmed())
{
stmt = GenSql(ast1, stmt);
}
var nodes = new RouteResultsetNode[1];
nodes[0] = new RouteResultsetNode(schema.DataNode, stmt);
rrs.Nodes = nodes;
return(rrs);
}
// 元数据语句路由
if (visitor1.IsTableMetaRead())
{
MetaRouter.RouteForTableMeta(rrs, schema, ast1, visitor1, stmt);
if (visitor1.IsNeedRewriteField())
{
rrs.Flag = RouteResultset.RewriteField;
}
return(rrs);
}
// 匹配规则
TableConfig matchedTable = null;
RuleConfig rule = null;
IDictionary <string, IList <object> > columnValues = null;
var astExt = visitor1.GetColumnValue();
var tables = schema.Tables;
foreach (var e in astExt)
{
var col2Val = e.Value;
var tc = tables.GetValue(e.Key);
if (tc == null)
{
continue;
}
if (matchedTable == null)
{
matchedTable = tc;
}
if (col2Val == null || col2Val.IsEmpty())
{
continue;
}
var tr = tc.Rule;
if (tr != null)
{
foreach (var rc in tr.Rules)
{
var match = true;
foreach (var ruleColumn in rc.Columns)
{
match &= col2Val.ContainsKey(ruleColumn);
}
if (match)
{
columnValues = col2Val;
rule = rc;
matchedTable = tc;
goto ft_break;
}
}
}
}
ft_break:
;
// 规则匹配处理，表级别和列级别。
if (matchedTable == null)
{
var sql = visitor1.IsSchemaTrimmed() ? GenSql(ast1, stmt) : stmt;
var rn = new RouteResultsetNode[1];
if (string.Empty.Equals(schema.DataNode) && IsSystemReadSql(ast1))
{
rn[0] = new RouteResultsetNode(schema.RandomDataNode, sql);
}
else
{
rn[0] = new RouteResultsetNode(schema.DataNode, sql);
}
rrs.Nodes = rn;
return(rrs);
}
if (rule == null)
{
if (matchedTable.IsRuleRequired)
{
throw new ArgumentException(string.Format("route rule for table {0} is required: {1}",
matchedTable.Name, stmt));
}
var dataNodes = matchedTable.DataNodes;
var sql = visitor1.IsSchemaTrimmed() ? GenSql(ast1, stmt) : stmt;
var rn = new RouteResultsetNode[dataNodes.Length];
for (var i = 0; i < dataNodes.Length; ++i)
{
rn[i] = new RouteResultsetNode(dataNodes[i], sql);
}
rrs.Nodes = rn;
SetGroupFlagAndLimit(rrs, visitor1);
return(rrs);
}
// 规则计算
ValidateAst(ast1, matchedTable, rule, visitor1);
var dnMap = RuleCalculate(matchedTable, rule, columnValues);
if (dnMap == null || dnMap.IsEmpty())
{
throw new ArgumentException("No target dataNode for rule " + rule);
}
// 判断路由结果是单库还是多库
if (dnMap.Count == 1)
{
var dataNode = matchedTable.DataNodes[dnMap.Keys.FirstOrDefault()];
//string dataNode = matchedTable.GetDataNodes()[dnMap.Keys.GetEnumerator().Current];
var sql = visitor1.IsSchemaTrimmed() ? GenSql(ast1, stmt) : stmt;
var rn = new RouteResultsetNode[1];
rn[0] = new RouteResultsetNode(dataNode, sql);
rrs.Nodes = rn;
}
else
{
var rn = new RouteResultsetNode[dnMap.Count];
if (ast1 is DmlInsertReplaceStatement)
{
var ir = (DmlInsertReplaceStatement)ast1;
DispatchInsertReplace(rn, ir, rule.Columns, dnMap, matchedTable, stmt, visitor1);
}
else
{
DispatchWhereBasedStmt(rn, ast1, rule.Columns, dnMap, matchedTable, stmt, visitor1);
}
rrs.Nodes = rn;
SetGroupFlagAndLimit(rrs, visitor1);
}
return(rrs);
}
