init
This commit is contained in:
life
269
app/lea/Util.go
Normal file
269
app/lea/Util.go
Normal file
@ -0,0 +1,269 @@
|
||||
package lea
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"regexp"
|
||||
"crypto/md5"
|
||||
"crypto/rand"
|
||||
"encoding/base64"
|
||||
"encoding/hex"
|
||||
"io"
|
||||
"labix.org/v2/mgo/bson"
|
||||
"time"
|
||||
math_rand "math/rand"
|
||||
)
|
||||
|
||||
// 字符串
|
||||
|
||||
// md5
|
||||
func Md5(s string) string {
|
||||
h := md5.New()
|
||||
h.Write([]byte(s))
|
||||
return hex.EncodeToString(h.Sum(nil))
|
||||
}
|
||||
|
||||
// Guid
|
||||
func NewGuid() string {
|
||||
b := make([]byte, 48)
|
||||
|
||||
if _, err := io.ReadFull(rand.Reader, b); err != nil {
|
||||
return ""
|
||||
}
|
||||
return Md5(base64.URLEncoding.EncodeToString(b))
|
||||
}
|
||||
|
||||
// 后面加个str生成之, 更有保障, 确保唯一
|
||||
func NewGuidWith(str string) string {
|
||||
b := make([]byte, 48)
|
||||
|
||||
if _, err := io.ReadFull(rand.Reader, b); err != nil {
|
||||
return ""
|
||||
}
|
||||
return Md5(base64.URLEncoding.EncodeToString([]byte(string(b) + str)))
|
||||
}
|
||||
|
||||
// 以byte来截取
|
||||
func SubstringByte(str string, start int) string {
|
||||
return substr(str, start, len(str)-start, false)
|
||||
}
|
||||
|
||||
func Substring(str string, start int) string {
|
||||
return substr(str, start, len(str)-start, true)
|
||||
}
|
||||
|
||||
func Substr(str string, start, length int) string {
|
||||
return substr(str, start, length, true)
|
||||
}
|
||||
|
||||
func substr(str string, start, length int, isRune bool) string {
|
||||
rs := []rune(str)
|
||||
rs2 := []byte(str)
|
||||
|
||||
rl := len(rs)
|
||||
if !isRune {
|
||||
rl = len(rs2)
|
||||
}
|
||||
end := 0
|
||||
|
||||
if start < 0 {
|
||||
start = rl - 1 + start
|
||||
}
|
||||
end = start + length
|
||||
|
||||
if start > end {
|
||||
start, end = end, start
|
||||
}
|
||||
|
||||
if start < 0 {
|
||||
start = 0
|
||||
}
|
||||
if start > rl {
|
||||
start = rl
|
||||
}
|
||||
if end < 0 {
|
||||
end = 0
|
||||
}
|
||||
if end > rl {
|
||||
end = rl
|
||||
}
|
||||
|
||||
if isRune {
|
||||
return string(rs[start:end])
|
||||
}
|
||||
return string(rs2[start:end])
|
||||
}
|
||||
|
||||
// 截取含html的字符串
|
||||
/*
|
||||
<div>
|
||||
abcdef
|
||||
<p>xxxxxxxxxxx</p>
|
||||
<img src="" />
|
||||
<div>
|
||||
abcdfef
|
||||
<p>aksdfjsld
|
||||
dfsadf
|
||||
--------------- 这里截断
|
||||
</p>
|
||||
<p>xxxxxxxxxxxxxxx</p>
|
||||
abcdef
|
||||
</div>
|
||||
</div>
|
||||
那么只要在最后的html中添加</p></div></div>补全结束即可
|
||||
思路:
|
||||
1. 找到截断的位置
|
||||
2. 去掉成对的标签
|
||||
3. 去掉不要结束的标签
|
||||
4. 补全标签
|
||||
*/
|
||||
func ReplaceAll(oldStr, pattern, newStr string) string {
|
||||
p, _ := regexp.Compile(pattern)
|
||||
s := p.ReplaceAll([]byte(oldStr), []byte(newStr))
|
||||
return string(s)
|
||||
}
|
||||
|
||||
func SubStringHTML(param string, length int, end string) string {
|
||||
// 先取出<pre></pre>占位..
|
||||
|
||||
result := ""
|
||||
|
||||
// 1
|
||||
n := 0
|
||||
var temp rune // 中文问题, 用rune来解决
|
||||
isCode := false //是不是HTML代码
|
||||
isHTML := false //是不是HTML特殊字符,如
|
||||
rStr := []rune(param)
|
||||
for i := 0; i < len(rStr); i++ {
|
||||
temp = rStr[i]
|
||||
if temp == '<' {
|
||||
isCode = true
|
||||
} else if temp == '&' {
|
||||
isHTML = true
|
||||
} else if temp == '>' && isCode {
|
||||
n = n - 1
|
||||
isCode = false
|
||||
} else if temp == ';' && isHTML {
|
||||
isHTML = false
|
||||
}
|
||||
if !isCode && !isHTML {
|
||||
n = n + 1
|
||||
}
|
||||
result += string(temp)
|
||||
if n >= length {
|
||||
break
|
||||
}
|
||||
}
|
||||
result += end
|
||||
|
||||
// 取出所有标签
|
||||
tempResult := ReplaceAll(result, "(>)[^<>]*(<?)", "$1$2") // 把标签中间的所有内容都去掉了
|
||||
|
||||
// 2. 去掉不需要结素标记的HTML标记
|
||||
tempResult = ReplaceAll(tempResult, "</?(embed|AREA|BASE|BASEFONT|BR|COL|COLGROUP|DD|DT|FRAME|HEAD|HR|IMG|INPUT|ISINDEX|LI|LINK|META|OPTION|PARAM|area|wbr|br|col|colgroup|dd|dt|frame|hr|img|input|isindex|link|meta|param)[^<>]*/?>", "")
|
||||
|
||||
// 把<div class=xxx的class=xxx去掉
|
||||
tempResult = ReplaceAll(tempResult, "<(/?[a-zA-Z]+)[^<>]*>", "<$1>")
|
||||
|
||||
// 3 只能用正则,+stack来去有结束的
|
||||
// golang的正则暂不支持back reference, 以后可以用它来去掉重复的标签
|
||||
p, _ := regexp.Compile("<(/?[a-zA-Z]+)[^<>]*>") // 得到所有的<div>, </div>...
|
||||
strs := p.FindAllString(tempResult, -1)
|
||||
|
||||
// fmt.Println(strs)
|
||||
stack := make([]string, len(strs))
|
||||
stackP := -1
|
||||
for _, each := range strs {
|
||||
if stackP >= 0 {
|
||||
// 匹配
|
||||
if stack[stackP][1:] == each[2:] {
|
||||
stackP--
|
||||
continue
|
||||
}
|
||||
}
|
||||
stackP++
|
||||
stack[stackP] = each
|
||||
}
|
||||
// 补全tag
|
||||
if stackP != -1 {
|
||||
fmt.Println(stack[0 : stackP+1])
|
||||
|
||||
for _, each := range stack[0 : stackP+1] {
|
||||
if each[1] != '/' {
|
||||
result += "</" + each[1:]
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
// 是否是合格的密码
|
||||
func IsGoodPwd(pwd string) (bool, string) {
|
||||
if pwd == "" {
|
||||
return false, "密码不能为空"
|
||||
}
|
||||
if len(pwd) < 6 {
|
||||
return false, "密码至少6位"
|
||||
}
|
||||
return true, ""
|
||||
}
|
||||
|
||||
// 是否是email
|
||||
func IsEmail(email string) bool {
|
||||
if email == "" {
|
||||
return false;
|
||||
}
|
||||
ok, _ := regexp.MatchString(`^([a-zA-Z0-9]+[_|\_|\.]?)*[a-zA-Z0-9]+@([a-zA-Z0-9]+[_|\_|\.]?)*[a-zA-Z0-9]+\.[0-9a-zA-Z]{2,3}$`, email)
|
||||
return ok
|
||||
}
|
||||
|
||||
// 是否只包含数字, 字母 -, _
|
||||
func IsUsername(username string) bool {
|
||||
if username == "" {
|
||||
return false;
|
||||
}
|
||||
ok, _ := regexp.MatchString(`[^0-9a-zA-Z_\-]`, username)
|
||||
return !ok
|
||||
}
|
||||
|
||||
// 是否是ObjectId
|
||||
func IsObjectId(id string) (ok bool) {
|
||||
defer func() { // 必须要先声明defer，否则不能捕获到panic异常
|
||||
// 证明有错误发生
|
||||
if err := recover(); err != nil {
|
||||
ok = false
|
||||
} else {
|
||||
ok = true
|
||||
}
|
||||
}()
|
||||
bson.ObjectIdHex(id)
|
||||
return
|
||||
}
|
||||
|
||||
// 随机密码
|
||||
// num 几位
|
||||
func RandomPwd(num int) string {
|
||||
chars := make([]byte, 62)
|
||||
j := 0
|
||||
for i := 48; i <= 57; i++ {
|
||||
chars[j] = byte(i)
|
||||
j++
|
||||
}
|
||||
for i := 65; i <= 90; i++ {
|
||||
chars[j] = byte(i)
|
||||
j++
|
||||
}
|
||||
for i := 97; i <= 122; i++ {
|
||||
chars[j] = byte(i)
|
||||
j++
|
||||
}
|
||||
j--;
|
||||
|
||||
str := ""
|
||||
math_rand.Seed(time.Now().UnixNano())
|
||||
for i := 0; i < num; i++ {
|
||||
x := math_rand.Intn(j)
|
||||
str += string(chars[x])
|
||||
}
|
||||
|
||||
return str
|
||||
}
|
||||
Reference in New Issue
Block a user