Go 如何让函数只能被调用一次

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_36431213/article/details/83277009
某些时候我们会有让函数只能被调用一次的需求
  • 完成这个功能可以使用 Gosync.Once 方法
package main

import (
	"fmt"
	"sync"
)

type SayHello struct {
	once sync.Once
}

func (sayHello *SayHello) print() {
	sayHello.once.Do(func() {
		fmt.Println("hello")
	})
}

func main() {

	var a SayHello
	a.print()
	a.print()
	a.print()
	a.print()
	a.print()
	a.print()
}

输出:
hello
  • 当然也可以自己使用和 标志位 flagcount 计数器完成这个功能
package main

import (
	"fmt"
	"sync"
)

type SayHello struct {
	mutex  sync.Mutex
	isUsed bool
}

func (once *SayHello) Print() {
	once.mutex.Lock()
	if !once.isUsed {
		fmt.Println("hello")
		once.isUsed = true
	}
	once.mutex.Unlock()
}

func main() {
	a := SayHello{}
	a.Print()
	a.Print()
	a.Print()
	a.Print()
	a.Print()
	a.Print()
}

输出:
hello

猜你喜欢

转载自blog.csdn.net/qq_36431213/article/details/83277009