• 附录B：有趣的代码片段
    • 自重写程序
    • 三元表达式
    • 禁止 main 函数退出的方法
    • 基于管道的随机数生成器
    • Assert测试断言

    附录B：有趣的代码片段

    这里收集一些比较有意思的Go程序片段。

    自重写程序

    UNIX/Go语言之父 Ken Thompson 在1983年的图灵奖演讲 Reflections on Trusting Trust 就给出了一个C语言的自重写程序。

    最短的C语言自重写程序是 Vlad Taeerov 和 Rashit Fakhreyev 的版本：

    1. main(a){printf(a="main(a){printf(a=%c%s%c,34,a,34);}",34,a,34);}

    下面的Go语言版本自重写程序是 rsc 提供的：

    1. /* Go quine */
    2. package main
    4. import "fmt"
    6. func main() {
    7.     fmt.Printf("%s%c%s%c\n", q, 0x60, q, 0x60)
    8. }
    10. var q = `/* Go quine */
    11. package main
    13. import "fmt"
    15. func main() {
    16.     fmt.Printf("%s%c%s%c\n", q, 0x60, q, 0x60)
    17. }
    19. var q = `

    在 golang-nuts 中还有很多版本：

    1. package main;func main(){c:="package main;func main(){c:=%q;print(c,c)}";print(c,c)}
    1. package main;func main(){print(c+"\x60"+c+"\x60")};var c=`package main;func main(){print(c+"\x60"+c+"\x60")};var c=`

    如果有更短的版本欢迎告诉我们。

    三元表达式

    1. func If(condition bool, trueVal, falseVal interface{}) interface{} {
    2.     if condition {
    3.         return trueVal
    4.     }
    5.     return falseVal
    6. }
    8. a, b := 2, 3
    9. max := If(a > b, a, b).(int)
    10. println(max)

    禁止 main 函数退出的方法

    1. func main() {
    2.     defer func() { for {} }()
    3. }
    5. func main() {
    6.     defer func() { select {} }()
    7. }
    9. func main() {
    10.     defer func() { <-make(chan bool) }()
    11. }

    基于管道的随机数生成器

    随机数的一个特点是不好预测。如果一个随机数的输出是可以简单预测的，那么一般会称为伪随机数。

    1. func main() {
    2.     for i := range random(100) {
    3.         fmt.Println(i)
    4.     }
    5. }
    7. func random(n int) <-chan int {
    8.     c := make(chan int)
    9.     go func() {
    10.         defer close(c)
    11.         for i := 0; i < n; i++ {
    12.             select {
    13.             case c <- 0:
    14.             case c <- 1:
    15.             }
    16.         }
    17.     }()
    18.     return c
    19. }

    基于select语言特性构造的随机数生成器。

    Assert测试断言

    1. type testing_TBHelper interface {
    2.     Helper()
    3. }
    5. func Assert(tb testing.TB, condition bool, args ...interface{}) {
    6.     if x, ok := tb.(testing_TBHelper); ok {
    7.         x.Helper() // Go1.9+
    8.     }
    9.     if !condition {
    10.         if msg := fmt.Sprint(args...); msg != "" {
    11.             tb.Fatalf("Assert failed, %s", msg)
    12.         } else {
    13.             tb.Fatalf("Assert failed")
    14.         }
    15.     }
    16. }
    18. func Assertf(tb testing.TB, condition bool, format string, a ...interface{}) {
    19.     if x, ok := tb.(testing_TBHelper); ok {
    20.         x.Helper() // Go1.9+
    21.     }
    22.     if !condition {
    23.         if msg := fmt.Sprintf(format, a...); msg != "" {
    24.             tb.Fatalf("Assertf failed, %s", msg)
    25.         } else {
    26.             tb.Fatalf("Assertf failed")
    27.         }
    28.     }
    29. }
    31. func AssertFunc(tb testing.TB, fn func() error) {
    32.     if x, ok := tb.(testing_TBHelper); ok {
    33.         x.Helper() // Go1.9+
    34.     }
    35.     if err := fn(); err != nil {
    36.         tb.Fatalf("AssertFunc failed, %v", err)
    37.     }
    38. }