Index Hints in query not handled correctly about parser HOT 4 CLOSED

// Restore implements Node interface.
func (n *TableSource) Restore(ctx *RestoreCtx) error {
    needParen := false
    switch n.Source.(type) {
    case *SelectStmt, *UnionStmt:
        needParen = true
    }
    if needParen {
        ctx.WritePlain("(")
    }
    if err := n.Source.Restore(ctx); err != nil {
        return errors.Annotate(err, "An error occurred while restore TableSource.Source")
    }
    if needParen {
        ctx.WritePlain(")")
    }
    if asName := n.AsName.String(); asName != "" {
        ctx.WriteKeyWord(" AS ")
        ctx.WriteName(asName)
    }

    return nil
}

// Restore implements Node interface.
func (n *TableName) Restore(ctx *RestoreCtx) error {
    if n.Schema.String() != "" {
        ctx.WriteName(n.Schema.String())
        ctx.WritePlain(".")
    }
    ctx.WriteName(n.Name.String())
    if len(n.PartitionNames) > 0 {
        ctx.WriteKeyWord(" PARTITION")
        ctx.WritePlain("(")
        for i, v := range n.PartitionNames {
            if i != 0 {
                ctx.WritePlain(", ")
            }
            ctx.WriteName(v.String())
        }
        ctx.WritePlain(")")
    }
    for _, value := range n.IndexHints {
        ctx.WritePlain(" ")
        if err := value.Restore(ctx); err != nil {
            return errors.Annotate(err, "An error occurred while splicing IndexHints")
        }
    }

    return nil
}

diff --git a/vendor/github.com/pingcap/parser/ast/dml.go b/vendor/github.com/pingcap/parser/ast/dml.go
index 483018f..95cf7d2 100644
--- a/vendor/github.com/pingcap/parser/ast/dml.go
+++ b/vendor/github.com/pingcap/parser/ast/dml.go
@@ -205,13 +205,6 @@ func (n *TableName) Restore(ctx *RestoreCtx) error {
                }
                ctx.WritePlain(")")
        }
-       for _, value := range n.IndexHints {
-               ctx.WritePlain(" ")
-               if err := value.Restore(ctx); err != nil {
-                       return errors.Annotate(err, "An error occurred while splicing IndexHints")
-               }
-       }
-
        return nil
 }

@@ -395,6 +388,14 @@ func (n *TableSource) Restore(ctx *RestoreCtx) error {
                ctx.WriteKeyWord(" AS ")
                ctx.WriteName(asName)
        }
+       if tn, ok := n.Source.(*TableName); ok {
+               for _, value := range tn.IndexHints {
+                       ctx.WritePlain(" ")
+                       if err := value.Restore(ctx); err != nil {
+                               return errors.Annotate(err, "An error occurred while splicing IndexHints")
+                       }
+               }
+       }

        return nil
 }