146 lines
2.6 KiB
Go
146 lines
2.6 KiB
Go
package utils
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/md5"
|
|
"database/sql/driver"
|
|
"encoding/base64"
|
|
"fmt"
|
|
"image"
|
|
"image/png"
|
|
"net"
|
|
"os"
|
|
"sort"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gofrs/uuid"
|
|
"golang.org/x/crypto/bcrypt"
|
|
)
|
|
|
|
type JsonTime struct {
|
|
time.Time
|
|
}
|
|
|
|
func NewJsonTime(t time.Time) JsonTime {
|
|
return JsonTime{
|
|
Time: t,
|
|
}
|
|
}
|
|
|
|
func NowJsonTime() JsonTime {
|
|
return JsonTime{
|
|
Time: time.Now(),
|
|
}
|
|
}
|
|
|
|
func (t JsonTime) MarshalJSON() ([]byte, error) {
|
|
var stamp = fmt.Sprintf("\"%s\"", t.Format("2006-01-02 15:04:05"))
|
|
return []byte(stamp), nil
|
|
}
|
|
|
|
func (t JsonTime) Value() (driver.Value, error) {
|
|
var zeroTime time.Time
|
|
if t.Time.UnixNano() == zeroTime.UnixNano() {
|
|
return nil, nil
|
|
}
|
|
return t.Time, nil
|
|
}
|
|
|
|
func (t *JsonTime) Scan(v interface{}) error {
|
|
value, ok := v.(time.Time)
|
|
if ok {
|
|
*t = JsonTime{Time: value}
|
|
return nil
|
|
}
|
|
return fmt.Errorf("can not convert %v to timestamp", v)
|
|
}
|
|
|
|
type Bcrypt struct {
|
|
cost int
|
|
}
|
|
|
|
func (b *Bcrypt) Encode(password []byte) ([]byte, error) {
|
|
return bcrypt.GenerateFromPassword(password, b.cost)
|
|
}
|
|
|
|
func (b *Bcrypt) Match(hashedPassword, password []byte) error {
|
|
return bcrypt.CompareHashAndPassword(hashedPassword, password)
|
|
}
|
|
|
|
var Encoder = Bcrypt{
|
|
cost: bcrypt.DefaultCost,
|
|
}
|
|
|
|
func UUID() string {
|
|
v4, _ := uuid.NewV4()
|
|
return v4.String()
|
|
}
|
|
|
|
func Tcping(ip string, port int) bool {
|
|
var conn net.Conn
|
|
var err error
|
|
|
|
if conn, err = net.DialTimeout("tcp", ip+":"+strconv.Itoa(port), 2*time.Second); err != nil {
|
|
return false
|
|
}
|
|
defer conn.Close()
|
|
return true
|
|
}
|
|
|
|
func ImageToBase64Encode(img image.Image) (string, error) {
|
|
var buf bytes.Buffer
|
|
if err := png.Encode(&buf, img); err != nil {
|
|
return "", err
|
|
}
|
|
return base64.StdEncoding.EncodeToString(buf.Bytes()), nil
|
|
}
|
|
|
|
// 判断所给路径文件/文件夹是否存在
|
|
func FileExists(path string) bool {
|
|
_, err := os.Stat(path) //os.Stat获取文件信息
|
|
if err != nil {
|
|
if os.IsExist(err) {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
// 判断所给路径是否为文件夹
|
|
func IsDir(path string) bool {
|
|
s, err := os.Stat(path)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
return s.IsDir()
|
|
}
|
|
|
|
// 判断所给路径是否为文件
|
|
func IsFile(path string) bool {
|
|
return !IsDir(path)
|
|
}
|
|
|
|
// 去除重复元素
|
|
func Distinct(a []string) []string {
|
|
result := make([]string, 0, len(a))
|
|
temp := map[string]struct{}{}
|
|
for _, item := range a {
|
|
if _, ok := temp[item]; !ok {
|
|
temp[item] = struct{}{}
|
|
result = append(result, item)
|
|
}
|
|
}
|
|
return result
|
|
}
|
|
|
|
// 排序+拼接+摘要
|
|
func Sign(a []string) string {
|
|
sort.Strings(a)
|
|
data := []byte(strings.Join(a, ""))
|
|
has := md5.Sum(data)
|
|
return fmt.Sprintf("%x", has)
|
|
}
|