附录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. }