ohUrlShortener/utils/utils.go

// Copyright (c) [2022] [巴拉迪维 BaratSemet]
// [ohUrlShortener] is licensed under Mulan PSL v2.
// You can use this software according to the terms and conditions of the Mulan PSL v2.
// You may obtain a copy of Mulan PSL v2 at:
// 				 http://license.coscl.org.cn/MulanPSL2
// THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
// See the Mulan PSL v2 for more details.

package utils

import (
	"crypto/sha256"
	"fmt"
	"log"
	"math/rand"
	"os"
	"strings"
	"time"

	"github.com/btcsuite/btcd/btcutil/base58"
)

// ExitOnError 退出程序
func ExitOnError(message string, err error) {
	if err != nil {
		log.Printf("[%s] - %s", message, err)
		os.Exit(-1)
	}
}

// PrintOnError 打印错误
func PrintOnError(message string, err error) {
	if err != nil {
		log.Printf("[%s] - %s", message, err)
	}
}

// RaiseError 返回错误
func RaiseError(message string) error {
	if !EmptyString(message) {
		return fmt.Errorf(message)
	}
	return nil
}

// EmptyString 判断字符串是否为空
func EmptyString(str string) bool {
	str = strings.TrimSpace(str)
	return strings.EqualFold(str, "")
}

// UserAgentIpHash 生成用户代理和IP的哈希值
func UserAgentIpHash(useragent string, ip string) string {
	input := fmt.Sprintf("%s-%s-%s-%d", useragent, ip, time.Now().String(), rand.Int())
	data, _ := Sha256Of(input)
	str := Base58Encode(data)
	return str[:10]
}

// Sha256Of 计算字符串的哈希值
func Sha256Of(input string) ([]byte, error) {
	algorithm := sha256.New()
	_, err := algorithm.Write([]byte(strings.TrimSpace(input)))
	if err != nil {
		return nil, err
	}
	return algorithm.Sum(nil), nil
}

// Base58Encode base58编码
func Base58Encode(data []byte) string {
	return base58.Encode(data)
}