123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889 |
- package tools
- import (
- "archive/zip"
- "bufio"
- "bytes"
- "encoding/base64"
- "encoding/binary"
- "errors"
- "fmt"
- "io"
- "io/fs"
- "io/ioutil"
- "log"
- "net"
- "net/http"
- "net/url"
- "os"
- "os/exec"
- "path"
- "path/filepath"
- "regexp"
- "runtime"
- "scd_check_tools/conf"
- "scd_check_tools/logger"
- "strconv"
- "strings"
- "sync"
- "time"
- "unicode"
- "golang.org/x/text/transform"
- "hash/crc32"
- "github.com/astaxie/beego/orm"
- "golang.org/x/text/encoding/simplifiedchinese"
- //"golang.org/x/text/transform"
- )
- var GlobalID int64
- var MbTable = []uint16{0x0000, 0xC0C1, 0xC181, 0x0140, 0xC301, 0x03C0, 0x0280, 0xC241, 0xC601, 0x06C0, 0x0780, 0xC741, 0x0500, 0xC5C1, 0xC481, 0x0440, 0xCC01, 0x0CC0, 0x0D80, 0xCD41, 0x0F00, 0xCFC1, 0xCE81, 0x0E40, 0x0A00, 0xCAC1, 0xCB81, 0x0B40, 0xC901, 0x09C0, 0x0880, 0xC841, 0xD801, 0x18C0, 0x1980, 0xD941, 0x1B00, 0xDBC1, 0xDA81, 0x1A40, 0x1E00, 0xDEC1, 0xDF81, 0x1F40, 0xDD01, 0x1DC0, 0x1C80, 0xDC41, 0x1400, 0xD4C1, 0xD581, 0x1540, 0xD701, 0x17C0, 0x1680, 0xD641, 0xD201, 0x12C0, 0x1380, 0xD341, 0x1100, 0xD1C1, 0xD081, 0x1040, 0xF001, 0x30C0, 0x3180, 0xF141, 0x3300, 0xF3C1, 0xF281, 0x3240, 0x3600, 0xF6C1, 0xF781, 0x3740, 0xF501, 0x35C0, 0x3480, 0xF441, 0x3C00, 0xFCC1, 0xFD81, 0x3D40, 0xFF01, 0x3FC0, 0x3E80, 0xFE41, 0xFA01, 0x3AC0, 0x3B80, 0xFB41, 0x3900, 0xF9C1, 0xF881, 0x3840, 0x2800, 0xE8C1, 0xE981, 0x2940, 0xEB01, 0x2BC0, 0x2A80, 0xEA41, 0xEE01, 0x2EC0, 0x2F80, 0xEF41, 0x2D00, 0xEDC1, 0xEC81, 0x2C40, 0xE401, 0x24C0, 0x2580, 0xE541, 0x2700, 0xE7C1, 0xE681, 0x2640, 0x2200, 0xE2C1, 0xE381, 0x2340, 0xE101, 0x21C0, 0x2080, 0xE041, 0xA001, 0x60C0, 0x6180, 0xA141, 0x6300, 0xA3C1, 0xA281, 0x6240, 0x6600, 0xA6C1, 0xA781, 0x6740, 0xA501, 0x65C0, 0x6480, 0xA441, 0x6C00, 0xACC1, 0xAD81, 0x6D40, 0xAF01, 0x6FC0, 0x6E80, 0xAE41, 0xAA01, 0x6AC0, 0x6B80, 0xAB41, 0x6900, 0xA9C1, 0xA881, 0x6840, 0x7800, 0xB8C1, 0xB981, 0x7940, 0xBB01, 0x7BC0, 0x7A80, 0xBA41, 0xBE01, 0x7EC0, 0x7F80, 0xBF41, 0x7D00, 0xBDC1, 0xBC81, 0x7C40, 0xB401, 0x74C0, 0x7580, 0xB541, 0x7700, 0xB7C1, 0xB681, 0x7640, 0x7200, 0xB2C1, 0xB381, 0x7340, 0xB101, 0x71C0, 0x7080, 0xB041, 0x5000, 0x90C1, 0x9181, 0x5140, 0x9301, 0x53C0, 0x5280, 0x9241, 0x9601, 0x56C0, 0x5780, 0x9741, 0x5500, 0x95C1, 0x9481, 0x5440, 0x9C01, 0x5CC0, 0x5D80, 0x9D41, 0x5F00, 0x9FC1, 0x9E81, 0x5E40, 0x5A00, 0x9AC1, 0x9B81, 0x5B40, 0x9901, 0x59C0, 0x5880, 0x9841, 0x8801, 0x48C0, 0x4980, 0x8941, 0x4B00, 0x8BC1, 0x8A81, 0x4A40, 0x4E00, 0x8EC1, 0x8F81, 0x4F40, 0x8D01, 0x4DC0, 0x4C80, 0x8C41, 0x4400, 0x84C1, 0x8581, 0x4540, 0x8701, 0x47C0, 0x4680, 0x8641, 0x8201, 0x42C0, 0x4380, 0x8341, 0x4100, 0x81C1, 0x8081, 0x4040}
- func NowTime() string {
- //大坑一个:必须使用这个时间格式化,忘记YYYY mm ss hh mi ss这些吧
- return time.Now().Format("2006-01-02 15:04:05")
- }
- func TimeSeq(bit ...int) int {
- if bit != nil && bit[0] == 13 {
- return int(time.Now().UnixNano() / 1e6)
- }
- return int(time.Now().Unix())
- }
- func Typeof(v interface{}) string {
- return fmt.Sprintf("%T", v)
- }
- //过滤特殊字符
- //chars:过滤指定的字符,多个字符使用逗号分隔
- func FilterSpecialChar(p string, chars ...string) string {
- if p == "" {
- return ""
- }
- v := "<|>|'"
- if len(chars) > 0 {
- v = strings.ReplaceAll(chars[0], ",", "|")
- }
- return regexp.MustCompile(v).ReplaceAllString(p, "")
- }
- func IsEmpty(p interface{}, defaultvalue ...string) string {
- if p == nil {
- if defaultvalue != nil && len(defaultvalue) > 0 {
- return defaultvalue[0]
- }
- return ""
- }
- objtype := Typeof(p)
- if objtype == "int" {
- return strconv.Itoa(p.(int))
- } else if objtype == "int32" {
- return strconv.FormatInt(int64(p.(int32)), 10)
- } else if objtype == "int64" {
- return strconv.FormatInt(p.(int64), 10)
- } else if objtype == "float64" {
- return strconv.FormatFloat(p.(float64), 'f', -1, 64)
- }
- v := p.(string)
- if v == "" {
- if defaultvalue != nil && len(defaultvalue) > 0 {
- return defaultvalue[0]
- }
- }
- return strings.Trim(v, " ")
- }
- func CheckErr(err error) {
- if err != nil {
- log.Println(err.Error())
- }
- }
- //多表达式分组匹配
- //多个表达式组时使用&分隔,每个分组表达式可使用()括起来,更加易读,且内部内容必须符合正则表过式规则
- //表达式样例1:(断路器)&(跳|合|闸){1,}&(位置)
- //表达式样例2:Ia1&A&保护
- func RexGroupTestMatch(regstr, soruce string) bool {
- if regstr == "" || soruce == "" {
- return false
- }
- regstrs := strings.Split(regstr, "&")
- if len(regstrs) > 1 {
- //未分组时,直接匹配
- fcdarex, err2 := regexp.Compile(regstr)
- if err2 != nil {
- logger.Logger.Error(err2, regstr)
- return false
- }
- return fcdarex.MatchString(soruce)
- }
- wg := sync.WaitGroup{}
- wg.Add(len(regstrs))
- wgResultFalse := map[string]bool{}
- for _, item := range regstrs {
- go func(item string) {
- fcdarex, err2 := regexp.Compile(item)
- if err2 != nil {
- logger.Logger.Error(err2, item)
- wgResultFalse["false"] = false
- } else if !fcdarex.MatchString(soruce) {
- wgResultFalse["false"] = false
- }
- wg.Done()
- }(item)
- }
- wg.Wait()
- matchSuccess := true
- if _, h := wgResultFalse["false"]; h {
- matchSuccess = false
- }
- return matchSuccess
- }
- //生成全局唯一ID序列
- func GetSeqID() int64 {
- if GlobalID == 0 {
- GlobalID = 1
- } else {
- GlobalID = GlobalID + 1
- }
- return GlobalID
- }
- //不建议使用。建议直接使用log包
- func Log(logdata interface{}) {
- if conf.GlobalConfig["loglevel"] == "4" || conf.GlobalConfig["loglevel"] == "5" {
- log.Println(logdata)
- }
- }
- //不建议使用。建议直接使用log包
- func Error(logdata interface{}) {
- log.Println(logdata)
- }
- func FormatFileSize(size int64) string {
- if size == 0 {
- return "0B"
- } else if size < 1024 {
- return fmt.Sprintf("%dB", size)
- } else if size < (1024 * 1024) {
- return fmt.Sprintf("%.2fKB", float64(size)/float64(1024))
- } else if size < (1024 * 1024 * 1024) {
- return fmt.Sprintf("%.2fMB", float64(size)/float64(1024*1024))
- } else if size < (1024 * 1024 * 1024 * 1024) {
- return fmt.Sprintf("%.2fGB", float64(size)/float64(1024*1024*1024))
- } else if size < (1024 * 1024 * 1024 * 1024 * 1024) {
- return fmt.Sprintf("%.2fTB", float64(size)/float64(1024*1024*1024*1024))
- } else {
- return ""
- }
- }
- func CalCheckCRC32(data string) string {
- return fmt.Sprintf("%X", crc32.ChecksumIEEE([]byte(data)))
- }
- func CalCheckXor(command16 string) string {
- var a int64 = 0
- for i := 0; i < len(command16); i++ {
- a1, _ := strconv.ParseInt(command16[i:i+2], 16, 16)
- a2, _ := strconv.ParseInt(command16[i+2:i+4], 16, 16)
- a = a ^ (a1 ^ a2)
- i = i + 3
- }
- h := fmt.Sprintf("%X", a)
- return h
- }
- func CalCheckCrc16_Modbus(data []byte) string {
- var crc16 uint16
- crc16 = 0xffff
- for _, v := range data {
- n := uint8(uint16(v) ^ crc16)
- crc16 >>= 8
- crc16 ^= MbTable[n]
- }
- int16buf := new(bytes.Buffer)
- binary.Write(int16buf, binary.LittleEndian, crc16)
- r := fmt.Sprintf("%+X", int16buf.Bytes())
- return r
- }
- func CalCheckCRC(commd string) (r string) {
- var i = 0
- var sum = 0
- for {
- h1 := commd[i : i+2]
- n, _ := strconv.ParseUint(h1, 16, 32)
- //fmt.Println(h1 + "=====" + strconv.Itoa(int(n)))
- sum = sum + int(n)
- i = i + 2
- if i >= len(commd) {
- break
- }
- }
- if sum > 256 {
- sum = sum - 256
- }
- var hexStr = strconv.FormatInt(int64(sum), 16)
- return hexStr
- }
- func GetNetPicBase64(picurl string) string {
- resp, err := http.Get(picurl)
- if err != nil {
- fmt.Println(err)
- return ""
- }
- defer resp.Body.Close()
- /*body, er := ioutil.ReadAll(resp.Body)
- if er != nil {
- fmt.Println(er)
- return ""
- }*/
- dir, _ := filepath.Abs(filepath.Dir(os.Args[0]))
- reader := bufio.NewReaderSize(resp.Body, 500*1024)
- fileName := path.Base(picurl)
- file, _ := os.Create(dir + "/tests/" + fileName)
- writer := bufio.NewWriter(file)
- io.Copy(writer, reader)
- defer file.Close()
- ff, _ := os.Open(dir + "/tests/" + fileName)
- defer ff.Close()
- sourcebuffer := make([]byte, 500*1024)
- n, _ := ff.Read(sourcebuffer)
- //base64压缩
- sourcestring := base64.StdEncoding.EncodeToString(sourcebuffer[:n])
- return sourcestring
- }
- func GetPicBase64(picpath string, saveas string) string {
- ff, _ := os.Open(picpath)
- defer ff.Close()
- sourcebuffer := make([]byte, 5*1024*1024)
- n, _ := ff.Read(sourcebuffer)
- //base64压缩
- sourcestring := base64.StdEncoding.EncodeToString(sourcebuffer[:n])
- if saveas != "" {
- os.WriteFile(saveas, []byte(sourcestring), fs.ModePerm)
- }
- return sourcestring
- }
- func Base64MakePic(base64str string, picPath string) error {
- //fmt.Println(base64str)
- ddd, _ := base64.StdEncoding.DecodeString(base64str[len("data:image/jpeg;base64,"):])
- return ioutil.WriteFile(picPath, ddd, 0666)
- }
- //解析datatable参数为Map对象
- func ParseDatatableParamByFormdata(reqStr url.Values) map[string]interface{} {
- var result = map[string]interface{}{"datatable": map[string]interface{}{}}
- datatable := map[string]interface{}{"sort": map[string]interface{}{"sort": reqStr.Get("datatable[sort][sort]"), "field": reqStr.Get("datatable[sort][field]")}, "pagination": map[string]interface{}{"page": reqStr.Get("datatable[pagination][page]"), "perpage": reqStr.Get("datatable[pagination][perpage]")}}
- for k, v := range reqStr {
- if strings.HasPrefix(k, "datatable") {
- if !strings.HasPrefix(k, "datatable[pagination]") && !strings.HasPrefix(k, "datatable[sort]") {
- newKey := strings.Replace(k, "datatable", "", 1)
- newKey = strings.Replace(newKey, "[", "", 1)
- newKey = strings.Replace(newKey, "]", "", 1)
- datatable[newKey], _ = url.QueryUnescape(v[0])
- }
- } else {
- datatable[k], _ = url.QueryUnescape(v[0])
- }
- }
- result["datatable"] = datatable
- return result
- }
- //解析datatable参数为Map对象
- func ParseFormParam(reqStr url.Values) map[string]interface{} {
- var result = map[string]interface{}{"datatable": map[string]interface{}{}}
- datatable := map[string]interface{}{"sort": map[string]interface{}{"sort": reqStr.Get("datatable[sort][sort]"), "field": reqStr.Get("datatable[sort][field]")}, "pagination": map[string]interface{}{"page": reqStr.Get("datatable[pagination][page]"), "perpage": reqStr.Get("datatable[pagination][perpage]")}}
- result["datatable"] = datatable
- for k, v := range reqStr {
- if strings.Index(k, "datatable") == -1 {
- result[k] = v[0]
- } else if strings.Index(k, "datatable[sort]") == -1 && strings.Index(k, "datatable[pagination]") == -1 {
- k1 := strings.ReplaceAll(k, "datatable[", "")
- k1 = strings.ReplaceAll(k1, "]", "")
- result[k1] = v[0]
- }
- }
- return result
- }
- func OriginalCode(password interface{}) string {
- var result string
- pwd := strings.Split(password.(string), "0000")
- if len(pwd) == 2 {
- password2 := pwd[1]
- password2 = password2[len(password2)-(len(password2)/3*3):] //去除前面多余的0
- for i := 0; i < len(password2)/3; i++ {
- number := password2[i*3 : (i+1)*3]
- p, _ := strconv.ParseInt(number, 16, 32)
- result += string(p)
- }
- }
- return result
- }
- //解析datatable参数为Map对象
- func ParseDatatableParam(reqStr string) map[string]interface{} {
- var result = map[string]interface{}{"datatable": map[string]interface{}{}}
- tmp, _ := url.QueryUnescape(reqStr)
- datatable := map[string]interface{}{"sort": map[string]interface{}{"sort": "", "field": ""}, "pagination": map[string]interface{}{"page": 1, "perpage": 20}}
- if strings.HasPrefix(tmp, "datatable") {
- pl := strings.Split(tmp, "&")
- for _, item := range pl {
- vs := strings.Split(item, "=")
- re, _ := regexp.Compile(`\[.*?\]`)
- all := re.FindAll([]byte(vs[0]), -1)
- if len(all) == 1 {
- key := strings.Replace(strings.Replace(string(all[0]), "[", "", 1), "]", "", 1)
- datatable[key] = vs[1]
- } else if len(all) == 2 {
- ele := strings.Replace(strings.Replace(string(all[0]), "[", "", 1), "]", "", 1)
- key := strings.Replace(strings.Replace(string(all[1]), "[", "", 1), "]", "", 1)
- if _, ok := datatable[ele]; ok {
- tmp := datatable[ele].(map[string]interface{})
- tmp[key] = vs[1]
- } else {
- datatable[ele] = map[string]interface{}{key: vs[1]}
- }
- }
- }
- result["datatable"] = datatable
- }
- return result
- }
- //从请求参数中解析分页数据,组装成SQL格式
- func ParsePaginationToSqlPart(param map[string]interface{}, alias ...string) (string, map[string]interface{}) {
- var limit = ""
- var hasfield bool
- var t_alias = ""
- var t_id_alias = "id"
- if alias != nil {
- t_alias = alias[0] + "."
- }
- if alias != nil && len(alias) > 1 {
- t_id_alias = alias[1]
- }
- var meta = map[string]interface{}{"field": "", "sort": "", "page": "1", "pages": "0", "perpage": "1000000", "total": "0"}
- if datatable, ok := param["datatable"]; ok {
- dt := datatable.(map[string]interface{})
- var field, sorttype interface{}
- if sort, hassort := dt["sort"]; hassort {
- sortmap := sort.(map[string]interface{})
- if field, hasfield = sortmap["field"]; !hasfield || IsEmpty(field) == "" {
- field = t_id_alias
- }
- if sorttype, hasfield = sortmap["sort"]; !hasfield || IsEmpty(field) == "" {
- sorttype = "desc"
- }
- meta["field"] = field
- meta["sort"] = sorttype
- limit = limit + " order by " + t_alias + field.(string) + " " + sorttype.(string)
- }
- if pagination, has := dt["pagination"]; has {
- paginationmap := pagination.(map[string]interface{})
- if limit == "" {
- if field, hasfield = paginationmap["field"]; hasfield && IsEmpty(field) != "" {
- if sorttype, hasfield = paginationmap["sort"]; !hasfield || IsEmpty(sorttype) != "" {
- meta["field"] = field
- meta["sort"] = sorttype
- limit = limit + " order by " + t_alias + field.(string) + " " + sorttype.(string)
- }
- }
- }
- var page, perpage interface{}
- if page, hasfield = paginationmap["page"]; !hasfield || IsEmpty(page) == "" {
- page = "1"
- }
- if perpage, hasfield = paginationmap["perpage"]; !hasfield || IsEmpty(perpage) == "" {
- perpage = "50"
- }
- var pageno, pagesize int
- if Typeof(page) == "int" {
- pageno = page.(int)
- } else {
- pageno, _ = strconv.Atoi(page.(string))
- }
- if Typeof(perpage) == "int" {
- pagesize = perpage.(int)
- } else {
- pagesize, _ = strconv.Atoi(perpage.(string))
- }
- meta["page"] = pageno
- meta["perpage"] = pagesize
- //limit = limit + " limit " + strconv.Itoa((pageno-1)*pagesize) + "," + strconv.Itoa(pageno*pagesize)
- limit = limit + " limit " + strconv.Itoa((pageno-1)*pagesize) + "," + strconv.Itoa(pagesize)
- } else {
- limit = limit + " limit 0,1000000"
- }
- }
- return limit, meta
- }
- //从请求参数中解析分页数据,组装成不包含排序的SQL格式
- func ParsePaginationToSqlPartNoOrderBY(param map[string]interface{}, alias ...string) (string, map[string]interface{}) {
- var limit = ""
- var hasfield bool
- var t_alias = ""
- var t_id_alias = ""
- if alias != nil {
- t_alias = alias[0] + "."
- }
- if alias != nil && len(alias) > 1 {
- t_id_alias = alias[1]
- }
- var meta = map[string]interface{}{"field": "", "sort": "", "page": "1", "pages": "0", "perpage": "1000000", "total": "0"}
- if datatable, ok := param["datatable"]; ok {
- dt := datatable.(map[string]interface{})
- var field, sorttype interface{}
- if sort, hassort := dt["sort"]; hassort {
- sortmap := sort.(map[string]interface{})
- if field, hasfield = sortmap["field"]; !hasfield || IsEmpty(field) == "" {
- field = t_id_alias
- }
- if sorttype, hasfield = sortmap["sort"]; !hasfield || IsEmpty(field) == "" {
- sorttype = "desc"
- }
- meta["field"] = field
- meta["sort"] = sorttype
- if field != "" {
- limit = limit + " order by " + t_alias + field.(string) + " " + sorttype.(string)
- }
- }
- if pagination, has := dt["pagination"]; has {
- paginationmap := pagination.(map[string]interface{})
- if limit == "" {
- if field, hasfield = paginationmap["field"]; hasfield && IsEmpty(field) != "" {
- if sorttype, hasfield = paginationmap["sort"]; !hasfield || IsEmpty(sorttype) != "" {
- meta["field"] = field
- meta["sort"] = sorttype
- if field != "" {
- limit = limit + " order by " + t_alias + field.(string) + " " + sorttype.(string)
- }
- }
- }
- }
- var page, perpage interface{}
- if page, hasfield = paginationmap["page"]; !hasfield || IsEmpty(page) == "" {
- page = "1"
- }
- if perpage, hasfield = paginationmap["perpage"]; !hasfield || IsEmpty(perpage) == "" {
- perpage = "50"
- }
- var pageno, pagesize int
- if Typeof(page) == "int" {
- pageno = page.(int)
- } else {
- pageno, _ = strconv.Atoi(page.(string))
- }
- if Typeof(perpage) == "int" {
- pagesize = perpage.(int)
- } else {
- pagesize, _ = strconv.Atoi(perpage.(string))
- }
- meta["page"] = pageno
- meta["perpage"] = pagesize
- //limit = limit + " limit " + strconv.Itoa((pageno-1)*pagesize) + "," + strconv.Itoa(pageno*pagesize)
- limit = " limit " + strconv.Itoa((pageno-1)*pagesize) + "," + strconv.Itoa(pagesize)
- } else {
- limit = limit + " limit 0,1000000"
- }
- }
- return limit, meta
- }
- func IsUTF8(data []byte) bool {
- var preNUm = func(data byte) int {
- str := fmt.Sprintf("%b", data)
- var i int = 0
- for i < len(str) {
- if str[i] != '1' {
- break
- }
- i++
- }
- return i
- }
- for i := 0; i < len(data); {
- if data[i]&0x80 == 0x00 {
- // 0XXX_XXXX
- i++
- continue
- } else if num := preNUm(data[i]); num > 2 {
- // 110X_XXXX 10XX_XXXX
- // 1110_XXXX 10XX_XXXX 10XX_XXXX
- // 1111_0XXX 10XX_XXXX 10XX_XXXX 10XX_XXXX
- // 1111_10XX 10XX_XXXX 10XX_XXXX 10XX_XXXX 10XX_XXXX
- // 1111_110X 10XX_XXXX 10XX_XXXX 10XX_XXXX 10XX_XXXX 10XX_XXXX
- // preNUm() 返回首个字节的8个bits中首个0bit前面1bit的个数,该数量也是该字符所使用的字节数
- i++
- for j := 0; j < num-1; j++ {
- //判断后面的 num - 1 个字节是不是都是10开头
- if data[i]&0xc0 != 0x80 {
- return false
- }
- i++
- }
- } else {
- //其他情况说明不是utf-8
- return false
- }
- }
- return true
- }
- func IsGBK(data []byte) bool {
- length := len(data)
- var i int = 0
- for i < length {
- //fmt.Printf("for %x\n", data[i])
- if data[i] <= 0xff {
- //编码小于等于127,只有一个字节的编码,兼容ASCII吗
- i++
- continue
- } else {
- //大于127的使用双字节编码
- if data[i] >= 0x81 &&
- data[i] <= 0xfe &&
- data[i+1] >= 0x40 &&
- data[i+1] <= 0xfe &&
- data[i+1] != 0xf7 {
- i += 2
- continue
- } else {
- return false
- }
- }
- }
- return true
- }
- func HasChineseChar(str string) bool {
- for _, r := range str {
- if unicode.Is(unicode.Han, r) || (regexp.MustCompile("[\u3002\uff1b\uff0c\uff1a\u201c\u201d\uff08\uff09\u3001\uff1f\u300a\u300b]").MatchString(string(r))) {
- return true
- }
- }
- return false
- }
- //字节转换成字符,支持中文转换。
- //charset:GB18030|UTF-8
- func ConvertByte2String(byte []byte, charset string) string {
- var str string
- switch charset {
- case "GB18030":
- var decodeBytes, _ = simplifiedchinese.GB18030.NewDecoder().Bytes(byte)
- str = string(decodeBytes)
- case "UTF-8":
- fallthrough
- default:
- str = string(byte)
- }
- return str
- }
- //IP verify.支持通配符*
- func VerifyIPFormat(ip string) bool {
- if ip == "" {
- return false
- }
- ipParts := strings.Split(ip, ".")
- if len(ipParts) != 3 {
- return false
- }
- for _, v := range ipParts {
- if v != "*" {
- v1, e := strconv.Atoi(v)
- if e != nil {
- return false
- }
- if v1 < 0 || v1 > 255 {
- return false
- }
- }
- }
- return true
- }
- //email verify
- func VerifyEmailFormat(email string) bool {
- pattern := `\w+([-+.]\w+)*@\w+([-.]\w+)*\.\w+([-.]\w+)*` //匹配电子邮箱
- //pattern := `^[0-9a-z][_.0-9a-z-]{0,31}@([0-9a-z][0-9a-z-]{0,30}[0-9a-z]\.){1,4}[a-z]{2,4}$`
- reg := regexp.MustCompile(pattern)
- return reg.MatchString(email)
- }
- //mobile verify
- func VerifyMobileFormat(mobileNum string) bool {
- regular := "^((13[0-9])|(14[5,7])|(15[0-3,5-9])|(17[0,3,5-8])|(18[0-9])|166|198|199|(147))\\d{8}$"
- reg := regexp.MustCompile(regular)
- return reg.MatchString(mobileNum)
- }
- func Unzip(zipFile string, destDir string) error {
- zipReader, err := zip.OpenReader(zipFile)
- if err != nil {
- return err
- }
- defer zipReader.Close()
- decodeName := ""
- for _, f := range zipReader.File {
- if f.Flags == 1 {
- //如果标志为是 1 << 11也就是 2048 则是utf-8编码
- decodeName = f.Name
- } else {
- //如果标致位是0或其他 则是默认的本地编码 默认为gbk
- i := bytes.NewReader([]byte(f.Name))
- decoder := transform.NewReader(i, simplifiedchinese.GBK.NewDecoder())
- content, _ := ioutil.ReadAll(decoder)
- decodeName = string(content)
- }
- fpath := filepath.Join(destDir, decodeName)
- if f.FileInfo().IsDir() {
- os.MkdirAll(fpath, os.ModePerm)
- } else {
- if err = os.MkdirAll(filepath.Dir(fpath), os.ModePerm); err != nil {
- return err
- }
- inFile, err := f.Open()
- if err != nil {
- return err
- }
- defer inFile.Close()
- outFile, err := os.OpenFile(fpath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
- if err != nil {
- return err
- }
- defer outFile.Close()
- _, err = io.Copy(outFile, inFile)
- if err != nil {
- return err
- }
- }
- }
- return nil
- }
- func Zip(srcFile string, destZip string) error {
- zipfile, err := os.Create(destZip)
- if err != nil {
- return err
- }
- defer zipfile.Close()
- archive := zip.NewWriter(zipfile)
- defer archive.Close()
- filepath.Walk(srcFile, func(path string, info os.FileInfo, err error) error {
- if err != nil {
- return err
- }
- header, err := zip.FileInfoHeader(info)
- if err != nil {
- return err
- }
- header.Name = strings.TrimPrefix(path, filepath.Dir(srcFile)+string(os.PathSeparator))
- // header.Name = path
- if info.IsDir() {
- header.Name += string(os.PathSeparator)
- } else {
- header.Method = zip.Deflate
- }
- writer, err := archive.CreateHeader(header)
- if err != nil {
- return err
- }
- if !info.IsDir() {
- file, err := os.Open(path)
- if err != nil {
- return err
- }
- defer file.Close()
- _, err = io.Copy(writer, file)
- }
- return err
- })
- return err
- }
- func ZipMutile(srcFileList []string, destZip string) error {
- zipfile, err := os.OpenFile(destZip, os.O_CREATE|os.O_RDWR|os.O_APPEND, fs.ModePerm)
- if err != nil {
- return err
- }
- defer zipfile.Close()
- //zipfile.Seek(0, os.SEEK_END)
- archive := zip.NewWriter(zipfile)
- defer archive.Close()
- for _, srcFile := range srcFileList {
- filepath.Walk(srcFile, func(path string, info os.FileInfo, err error) error {
- if err != nil {
- return err
- }
- header, err := zip.FileInfoHeader(info)
- if err != nil {
- return err
- }
- header.Name = strings.TrimPrefix(path, filepath.Dir(srcFile)+string(os.PathSeparator))
- // header.Name = path
- if info.IsDir() {
- header.Name += string(os.PathSeparator)
- } else {
- header.Method = zip.Deflate
- }
- writer, err := archive.CreateHeader(header)
- if err != nil {
- return err
- }
- if !info.IsDir() {
- file, err := os.Open(path)
- if err != nil {
- return err
- }
- defer file.Close()
- _, err = io.Copy(writer, file)
- }
- return err
- })
- }
- return err
- }
- func Utf8ToGBK(text string) (string, error) {
- dst := make([]byte, len(text)*2)
- tr := simplifiedchinese.GB18030.NewEncoder()
- nDst, _, err := tr.Transform(dst, []byte(text), true)
- if err != nil {
- return text, err
- }
- return string(dst[:nDst]), nil
- }
- //统一删除数据表记录
- func DeleteTableReorce(tableName, primaryField, Id string) error {
- o := orm.NewOrm()
- var sqlParameter []interface{}
- var sqlCommandText = "delete from " + tableName + " where " + primaryField + "=?;"
- sqlParameter = append(sqlParameter, Id)
- _, err := o.Raw(sqlCommandText, sqlParameter).Exec()
- return err
- }
- //获得单条数据记录
- func GetSingleRecord(tableName, primaryField, Id string) ([]orm.Params, error) {
- o := orm.NewOrm()
- var sqlParameter []interface{}
- var sqlCommandText string
- if tableName == "t_data_monitor_task" {
- sqlCommandText = "select a.*,tr.path checkreport,u.name adminusername,s.name stationname,sc1.name switchgear_one_name,sc2.name switchgear_two_name,sc3.name switchgear_comp_name,(select tpl_name from t_data_config_tpl where id=a.tpl) tplname " +
- "from t_data_monitor_task a left join t_data_user u on a.adminuser=u.id inner join t_data_station s on a.stationid=s.stationid left join t_data_switchgear sc1 on a.switchgear_one=sc1.id left join t_data_switchgear sc2 on a.switchgear_two=sc2.id left join t_data_switchgear sc3 on a.switchgear_comp=sc3.id left join t_data_monitor_report tr on a.id=tr.taskid where a.id=?"
- sqlParameter = append(sqlParameter, Id)
- } else {
- sqlCommandText = "select * from " + tableName + " where " + primaryField + "=?;"
- sqlParameter = append(sqlParameter, Id)
- }
- var tableData []orm.Params
- _, err := o.Raw(sqlCommandText, sqlParameter).Values(&tableData)
- return tableData, err
- }
- func LocalIPv4() (string, error) {
- ifaces, err := net.Interfaces()
- if err != nil {
- return "", err
- }
- for _, iface := range ifaces {
- if iface.Flags&net.FlagUp == 0 {
- continue
- }
- if iface.Flags&net.FlagLoopback != 0 {
- continue
- }
- addrs, err := iface.Addrs()
- if err != nil {
- return "", err
- }
- for _, addr := range addrs {
- var ip net.IP
- switch v := addr.(type) {
- case *net.IPNet:
- ip = v.IP
- case *net.IPAddr:
- ip = v.IP
- }
- if ip == nil || ip.IsLoopback() {
- continue
- }
- ip = ip.To4()
- if ip == nil {
- continue
- }
- return ip.String(), nil
- }
- }
- return "", errors.New("未连接网络")
- }
- //获取当前运行程序版本号
- func GetVersion() string {
- exename := conf.GlobalConfig["appid"]
- if exename == "" {
- logger.Logger.Println("系统还未配置appid属性,请正确设置后再试!")
- os.Exit(0)
- return ""
- }
- if string(runtime.GOOS) == "windows" {
- exename = exename + ".exe"
- }
- f, err1 := os.Open(exename)
- if err1 != nil {
- logger.Logger.Println("获取版本号失败,请检查appid是否与程序名称匹配!")
- logger.Logger.Error(err1)
- os.Exit(0)
- return ""
- }
- fi, err2 := f.Stat()
- if err2 != nil {
- logger.Logger.Info("获取版本号失败:")
- logger.Logger.Error(err2)
- return ""
- }
- return fi.ModTime().Format("20060102-15")
- }
- func GetAppCurPath() string {
- file, _ := exec.LookPath(os.Args[0])
- execpath, _ := filepath.Abs(file)
- execpath = filepath.Dir(execpath)
- if execpath[len(execpath)-1:] != string(os.PathSeparator) {
- execpath = execpath + string(os.PathSeparator)
- }
- return execpath
- }
|