dongyonglie5132 2015-05-12 16:10
浏览 189
已采纳

如何比较两个源代码文件/ ast树?

I'm generating some source code using the templates package( is there a better method? )and part of the testing I need to check if the output matches the expected source code.

  • I tried a string comparison but it fails due the extra spaces / new lines generated by the templates package. I've also tried format.Source with not success. ( FAIL)
  • I tried to parse the ast of the both sources (see bellow) but the ast doesn't match either even if the code is basically same except the new lines / spaces. (FAIL)

    package main

    import (
        "fmt"
        "go/parser"
        "go/token"
        "reflect"
    )
    
    func main() {
        stub1 := `package main
         func myfunc(s string) error {
            return nil  
        }`
        stub2 := `package main
    
         func myfunc(s string) error {
    
            return nil
    
        }`
        fset := token.NewFileSet()
        r1, err := parser.ParseFile(fset, "", stub1, parser.AllErrors)
        if err != nil {
            panic(err)
        }
        fset = token.NewFileSet()
        r2, err := parser.ParseFile(fset, "", stub2, parser.AllErrors)
        if err != nil {
            panic(err)
        }
        if !reflect.DeepEqual(r1, r2) {
            fmt.Printf("e %v, r %s, ", r1, r2)
        }
    }
    

Playground

  • 写回答

2条回答 默认 最新

  • dragon2025 2015-05-15 13:35
    关注

    This was easier than I thought. All I had to do was to remove the empty new lines(after formatting). Below is the code.

        package main
    
        import (
            "fmt"
            "go/format"
            "strings"
        )
    
        func main() {
            a, err := fmtSource(stub1)
            if err != nil {
                panic(err)
            }
            b, err := fmtSource(stub2)
            if err != nil {
                panic(err)
            }
            if a != b {
                fmt.Printf("a %v, 
     b %v", a, b)
            }
        }
    
    func fmtSource(source string) (string, error) {
        if !strings.Contains(source, "package") {
            source = "package main
    " + source
        }
        b, err := format.Source([]byte(source))
        if err != nil {
            return "", err
        }
        // cleanLine replaces double space with one space
        cleanLine := func(s string)string{
            sa := strings.Fields(s)
            return strings.Join(sa, " ")
        }
        lines := strings.Split(string(b), "
    ")
        n := 0
        var startLn *int
        for _, line := range lines {
            if line != "" {
                line = cleanLine(line)
                lines[n] = line
                if startLn == nil {
                    x := n
                    startLn = &x
                }
                n++
            }
        }
        lines = lines[*startLn:n]
        // Add final "" entry to get trailing newline from Join.
        if n > 0 && lines[n-1] != "" {
            lines = append(lines, "")
        }
    
    
        // Make it pretty 
        b, err = format.Source([]byte(strings.Join(lines, "
    ")))
        if err != nil {
            return "", err
        }
        return string(b), nil
    }
    
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题?

悬赏问题

  • ¥15 python天天向上类似问题,但没有清零
  • ¥30 3天&7天&&15天&销量如何统计同一行
  • ¥30 帮我写一段可以读取LD2450数据并计算距离的Arduino代码
  • ¥15 C#调用python代码(python带有库)
  • ¥15 矩阵加法的规则是两个矩阵中对应位置的数的绝对值进行加和
  • ¥15 活动选择题。最多可以参加几个项目?
  • ¥15 飞机曲面部件如机翼,壁板等具体的孔位模型
  • ¥15 vs2019中数据导出问题
  • ¥20 云服务Linux系统TCP-MSS值修改?
  • ¥20 关于#单片机#的问题:项目:使用模拟iic与ov2640通讯环境:F407问题:读取的ID号总是0xff,自己调了调发现在读从机数据时,SDA线上并未有信号变化(语言-c语言)