funcTools.go

package go_utils

import (
	"bytes"
	"context"
	"encoding/json"
	"log"
	"os"
	"runtime"
	"strings"
)

type RegFuncs struct {
	FuncList []func()
}

// 并发执行所有func，并等待他们执行完
func WaitFunc(a ...func()) {
	var wg = NewSizedWaitGroup(len(a))
	WaitFunc4Wg(&wg, a...)
	wg.Wait()
}

// 并行执行方法，并将使用 wg 计数器
func WaitFunc4Wg(wg *SizedWaitGroup, a ...func()) {
	for _, x := range a {
		wg.Add(1)
		go func(cbk func()) {
			defer wg.Done()
			cbk()
		}(x)
	}
}

// 并行执行方法，并将使用 wg 计数器
// 同时传入参数parms
func WaitFunc4WgParms[T any](wg *SizedWaitGroup, parms []T, a ...func(x ...T)) {
	for _, x := range a {
		wg.Add(1)
		go func(cbk func(...T)) {
			defer wg.Done()
			cbk(parms...)
		}(x)
	}
}

// map format out
func OutMap(m *map[string]interface{}) {
	if data, err := Json.Marshal(m); nil == err {
		var out bytes.Buffer
		err = json.Indent(&out, data, "", "  ")
		if err != nil {
			log.Fatalln(err)
		}

		out.WriteTo(os.Stdout)
	}
}
func WaitOneFunc4WgParmsChan[T any](wg *SizedWaitGroup, cbk func(x T), cT chan struct{}, parms ...T) {
	for _, x := range parms {
		cT <- struct{}{}
		wg.Add(1)
		go func(p1 T) {
			defer func() {
				wg.Done()
				<-cT
			}()
			cbk(p1)
		}(x)
	}
}

// 通用的、常规小工具参数接收和处理
// 接收命令行 os.Args[1:]
// 接收管道命令，按行输入
func DoCommontools4Chan() chan *string {
	var a []string
	var out = make(chan *string)
	if 1 < len(os.Args) {
		a = os.Args[1:]
		go func() {
			for _, x := range a {
				out <- &x
			}
			close(out)
		}()
	} else {
		// ReadStdIn 内 close
		go ReadStdIn(out) // 必须 移步
	}
	return out
}

// 通用的、常规小工具参数接收和处理
// 接收命令行 os.Args[1:]
// 接收管道命令，按行输入
func DoCommontools(cbk func(string, *SizedWaitGroup), cTs ...*chan struct{}) {
	var out = DoCommontools4Chan()
	var wg = NewSizedWaitGroup(2000)
	bC := nil != cTs && 0 < len(cTs)
	for x := range out {
		if bC {
			*cTs[0] <- struct{}{}
		}
		WaitFunc4WgParms(&wg, []any{*x}, func(x2 ...any) {
			if bC {
				defer func() {
					<-*cTs[0]
				}()
			}
			cbk(x2[0].(string), &wg)
		})
	}
	wg.Wait()
}

// 迭代所有的参数
func WaitOneFunc4WgParms[T any](wg *SizedWaitGroup, cbk func(x T), parms ...T) {
	for _, x := range parms {
		wg.Add(1)
		go func(p1 T) {
			defer wg.Done()
			cbk(p1)
		}(x)
	}
}

func RegFuncWithCtx(ctx context.Context, cancel context.CancelFunc, cbk ...func(context.Context, context.CancelFunc)) {
	for _, x := range cbk {
		go x(ctx, cancel)
	}
}
func RegFunc(cbk ...func(context.Context, context.CancelFunc)) (context.Context, context.CancelFunc) {
	var ctx, cancel = context.WithCancel(context.Background())
	RegFuncWithCtx(ctx, cancel, cbk...)
	return ctx, cancel
}

// 注册
func (r *RegFuncs) RegFunc(fn func()) {
	r.FuncList = append(r.FuncList, fn)
}

// tick 检测
var TickFunc = new(RegFuncs)
var ReleaseFunc = new(RegFuncs)

// 串行tick
func (r *RegFuncs) DoFunc() {
	for _, c := range r.FuncList {
		c()
	}
}

// Catch Panic
//
//	in your func: defer CatchPanic()
func CatchPanic() {
	if o := recover(); nil != o {
		log.Println(o)
	}
}

// 将该方法放到方法中运行，就可以打印出所有调用该方法的链路出来
func PrintCaller() {
	var i = 0
	for {
		i++
		if pc, file, line, ok := runtime.Caller(i); ok {
			fc := runtime.FuncForPC(pc)
			log.Printf("<-%s %s file:%s (line:%d)\n", strings.Repeat(":", i-1), fc.Name(), file, line) // , runtime.CallersFrames([]uintptr{pc})
			if "main.main" == fc.Name() {
				break
			}
		} else {
			break
		}
	}
}