recover.go

package util

import (
	"errors"
	"fmt"
	"log"
	"os"
	"runtime/debug"
	"strings"
)

func Recover() {
	if r := recover(); r != nil {
		s, ok := r.(string)
		if ok {
			LogError(errors.New(s))
			return
		}
		err := r.(error)
		LogError(err)
	}
}

func RecoverAndExit() {
	if r := recover(); r != nil {
		s, ok := r.(string)
		if ok {
			LogError(errors.New(s))
			return
		}
		err := r.(error)
		LogError(err)
		os.Exit(1)
	}
}

var ErrorLogger = func(s string) {
	log.Println(s)
}

func getStackLines(stack []byte, from int, to int) []string {
	lines := strings.Split(string(stack), "\n")
	if len(lines) < 2 {
		return []string{}
	}
	return lines[from:to]
}
func GetStackLines(from int, to int) []string {
	return getStackLines(debug.Stack(), from, to)
}
func LogError(err error) {
	if IsErrorIgnored(err) == false {
		lines := strings.Split(string(debug.Stack()), "\n")
		length := len(lines)
		maxLength := LoggerMaxLength*2 + 7
		if length > maxLength {
			length = maxLength
		}
		var output = make([]string, length-6)
		panicid := IDGenerator()
		var panicidinfo string
		if panicid != "" {
			panicidinfo = fmt.Sprintf("[PanicID:%s] - ", panicid)
		}
		output[0] = fmt.Sprintf("Panic: %s%s", panicidinfo, err.Error())
		output[0] += "\n" + lines[0]
		copy(output[1:], lines[7:])
		ErrorLogger(strings.Join(output, "\n"))
	}
}

func LogIfError(err error) {
	if err != nil {
		LogError(err)
	}
}