optimize type assertion (#324)

bytedance · Oct 31, 2022 · 2866800 · 2866800
1 parent 78c2ade
commit 2866800
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/ast/node.go b/ast/node.go
@@ -695,14 +695,14 @@ func (self *Node) GetByPath(path ...interface{}) *Node {
     }
     var s = self
     for _, p := range path {
-        switch p.(type) {
+        switch p := p.(type) {
         case int:
-            s = s.Index(p.(int))
+            s = s.Index(p)
             if !s.Valid() {
                 return s
             }
         case string:
-            s = s.Get(p.(string))
+            s = s.Get(p)
             if !s.Valid() {
                 return s
             }

diff --git a/ast/search.go b/ast/search.go
@@ -40,13 +40,13 @@ func (self *Searcher) GetByPath(path ...interface{}) (Node, error) {
 
     var err types.ParsingError
     for _, p := range path {
-        switch p.(type) {
+        switch p := p.(type) {
         case int:
-            if err = self.parser.searchIndex(p.(int)); err != 0 {
+            if err = self.parser.searchIndex(p); err != 0 {
                 return Node{}, self.parser.ExportError(err)
             }
         case string:
-            if err = self.parser.searchKey(p.(string)); err != 0 {
+            if err = self.parser.searchKey(p); err != 0 {
                 return Node{}, self.parser.ExportError(err)
             }
         default: