Compare commits
10 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| c2468a7389 | |||
| b6ba3486d6 | |||
| f59e909d30 | |||
| 3bb47de45a | |||
| c71973850e | |||
| 65f062b860 | |||
| 7b502fed36 | |||
| fd6b15bdaf | |||
| f31bf24ec2 | |||
| 30edb4d9fa |
+11
-7
@@ -357,14 +357,18 @@ func (that *Application) handler(w http.ResponseWriter, req *http.Request) {
|
|||||||
defer func() {
|
defer func() {
|
||||||
//是否展示日志
|
//是否展示日志
|
||||||
if that.WebConnectLog != nil {
|
if that.WebConnectLog != nil {
|
||||||
ipStr := Substr(context.Req.RemoteAddr, 0, strings.Index(context.Req.RemoteAddr, ":"))
|
|
||||||
//负载均衡优化
|
//负载均衡优化
|
||||||
if ipStr == "127.0.0.1" {
|
ipStr := ""
|
||||||
if req.Header.Get("X-Forwarded-For") != "" {
|
if req.Header.Get("X-Forwarded-For") != "" {
|
||||||
ipStr = req.Header.Get("X-Forwarded-For")
|
ipStr = req.Header.Get("X-Forwarded-For")
|
||||||
} else if req.Header.Get("X-Real-IP") != "" {
|
} else if req.Header.Get("X-Real-IP") != "" {
|
||||||
ipStr = req.Header.Get("X-Real-IP")
|
ipStr = req.Header.Get("X-Real-IP")
|
||||||
}
|
}
|
||||||
|
//负载均衡优化
|
||||||
|
if ipStr == "" {
|
||||||
|
//RemoteAddr := that.Req.RemoteAddr
|
||||||
|
ipStr = Substr(context.Req.RemoteAddr, 0, strings.Index(context.Req.RemoteAddr, ":"))
|
||||||
}
|
}
|
||||||
|
|
||||||
that.WebConnectLog.Infoln(ipStr, context.Req.Method,
|
that.WebConnectLog.Infoln(ipStr, context.Req.Method,
|
||||||
|
|||||||
+27
-23
@@ -1016,7 +1016,7 @@ func (that *MakeCode) Search(table string, userData Map, req *http.Request, db *
|
|||||||
|
|
||||||
if keywordTableStr != "" {
|
if keywordTableStr != "" {
|
||||||
if keywordTableStr == v.GetString("name") {
|
if keywordTableStr == v.GetString("name") {
|
||||||
keyword[table+"."+keywordTableStr+"[~]"] = keywordStr
|
data[table+"."+keywordTableStr+"[~]"] = keywordStr
|
||||||
}
|
}
|
||||||
|
|
||||||
if keywordTableStr == v.GetString("value") {
|
if keywordTableStr == v.GetString("value") {
|
||||||
@@ -1026,7 +1026,7 @@ func (that *MakeCode) Search(table string, userData Map, req *http.Request, db *
|
|||||||
childIds = append(childIds, cv.GetString("id"))
|
childIds = append(childIds, cv.GetString("id"))
|
||||||
}
|
}
|
||||||
if len(childIds) != 0 {
|
if len(childIds) != 0 {
|
||||||
keyword[v.GetString("link")+".id"] = childIds
|
data[v.GetString("link")+".id"] = childIds
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
continue
|
continue
|
||||||
@@ -1069,9 +1069,10 @@ func (that *MakeCode) Search(table string, userData Map, req *http.Request, db *
|
|||||||
parentID = userData.GetCeilInt(table + "_id")
|
parentID = userData.GetCeilInt(table + "_id")
|
||||||
data["OR"] = Map{table + ".id": parentID, table + ".parent_id": nil}
|
data["OR"] = Map{table + ".id": parentID, table + ".parent_id": nil}
|
||||||
} else {
|
} else {
|
||||||
|
//是否展示全部子级
|
||||||
|
where := Map{}
|
||||||
|
if req.FormValue("showall") == "1" {
|
||||||
|
|
||||||
if req.FormValue("showself") != "" {
|
|
||||||
where := Map{}
|
|
||||||
for _, v := range reqValue {
|
for _, v := range reqValue {
|
||||||
if len(where) == 0 {
|
if len(where) == 0 {
|
||||||
where[table+"."+parent_idsStr] = "," + v + ","
|
where[table+"."+parent_idsStr] = "," + v + ","
|
||||||
@@ -1079,9 +1080,21 @@ func (that *MakeCode) Search(table string, userData Map, req *http.Request, db *
|
|||||||
}
|
}
|
||||||
where = Map{"OR": where, table + "." + parent_idsStr: "," + v + ","}
|
where = Map{"OR": where, table + "." + parent_idsStr: "," + v + ","}
|
||||||
}
|
}
|
||||||
data["OR"] = Map{"OR": where, table + ".id": reqValue}
|
|
||||||
} else {
|
} else {
|
||||||
data[table+".parent_id"] = reqValue
|
where[table+".parent_id"] = reqValue
|
||||||
|
}
|
||||||
|
//是否展示自己
|
||||||
|
if req.FormValue("showself") == "1" {
|
||||||
|
if len(where) == 0 {
|
||||||
|
data["OR"] = Map{table + ".id": reqValue}
|
||||||
|
} else {
|
||||||
|
data["OR"] = Map{"OR": where, table + ".id": reqValue}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if len(where) != 0 {
|
||||||
|
data["OR"] = where
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
@@ -1093,14 +1106,9 @@ func (that *MakeCode) Search(table string, userData Map, req *http.Request, db *
|
|||||||
if parent_idsStr != "" {
|
if parent_idsStr != "" {
|
||||||
|
|
||||||
where := Map{}
|
where := Map{}
|
||||||
if len(reqValue) == 1 {
|
//是否全部展示
|
||||||
if len(where) == 0 {
|
if req.FormValue("showall") != "1" {
|
||||||
where[table+"."+searchItem.GetString("name")] = reqValue[0]
|
where[table+"."+searchItem.GetString("name")] = reqValue
|
||||||
|
|
||||||
} else {
|
|
||||||
where = Map{"OR": where, table + "." + searchItem.GetString("name"): reqValue[0]}
|
|
||||||
}
|
|
||||||
|
|
||||||
} else {
|
} else {
|
||||||
for _, v := range reqValue {
|
for _, v := range reqValue {
|
||||||
if len(where) == 0 {
|
if len(where) == 0 {
|
||||||
@@ -1110,6 +1118,7 @@ func (that *MakeCode) Search(table string, userData Map, req *http.Request, db *
|
|||||||
where = Map{"OR": where, searchItem.GetString("link") + "." + parent_idsStr: "," + v + ","}
|
where = Map{"OR": where, searchItem.GetString("link") + "." + parent_idsStr: "," + v + ","}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
//用户
|
//用户
|
||||||
if userData[searchItem.GetString("name")] != nil {
|
if userData[searchItem.GetString("name")] != nil {
|
||||||
|
|
||||||
@@ -1143,17 +1152,12 @@ func (that *MakeCode) Search(table string, userData Map, req *http.Request, db *
|
|||||||
data[k] = v
|
data[k] = v
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(keyword) > 1 {
|
if len(keyword) > 1 {
|
||||||
|
if data["OR"] != nil {
|
||||||
if data["AND"] != nil {
|
data = Map{"AND": data, "OR": keyword}
|
||||||
and := data.GetMap("AND")
|
|
||||||
for k, v := range keyword {
|
|
||||||
and[k] = v
|
|
||||||
}
|
|
||||||
|
|
||||||
//data = Map{"AND":Map{"OR": keyword} data, }
|
|
||||||
} else {
|
} else {
|
||||||
data["AND"] = keyword
|
data["OR"] = keyword
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -5,6 +5,7 @@ import (
|
|||||||
"encoding/hex"
|
"encoding/hex"
|
||||||
"math"
|
"math"
|
||||||
"strings"
|
"strings"
|
||||||
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
//安全锁
|
//安全锁
|
||||||
@@ -36,6 +37,42 @@ func StrFirstToUpper(str string) string {
|
|||||||
return strings.ToUpper(first) + other
|
return strings.ToUpper(first) + other
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 时间转字符串,第二个参数支持1-5对应显示年月日时分秒
|
||||||
|
func Time2Str(t *time.Time, qu ...interface{}) string {
|
||||||
|
if t == nil || t.Unix() < 0 {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
tp := 5
|
||||||
|
if len(qu) != 0 {
|
||||||
|
tp = (qu[0]).(int)
|
||||||
|
}
|
||||||
|
|
||||||
|
switch tp {
|
||||||
|
case 1:
|
||||||
|
return t.Format("2006-01")
|
||||||
|
case 2:
|
||||||
|
return t.Format("2006-01-02")
|
||||||
|
case 3:
|
||||||
|
return t.Format("2006-01-02 15")
|
||||||
|
case 4:
|
||||||
|
return t.Format("2006-01-02 15:04")
|
||||||
|
case 5:
|
||||||
|
return t.Format("2006-01-02 15:04:05")
|
||||||
|
case 12:
|
||||||
|
return t.Format("01-02")
|
||||||
|
case 14:
|
||||||
|
return t.Format("01-02 15:04")
|
||||||
|
case 15:
|
||||||
|
return t.Format("01-02 15:04:05")
|
||||||
|
case 34:
|
||||||
|
return t.Format("15:04")
|
||||||
|
case 35:
|
||||||
|
return t.Format("15:04:05")
|
||||||
|
}
|
||||||
|
return t.Format("2006-01-02 15:04:05")
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
// StrLd 相似度计算 ld compares two strings and returns the levenshtein distance between them.
|
// StrLd 相似度计算 ld compares two strings and returns the levenshtein distance between them.
|
||||||
func StrLd(s, t string, ignoreCase bool) int {
|
func StrLd(s, t string, ignoreCase bool) int {
|
||||||
if ignoreCase {
|
if ignoreCase {
|
||||||
|
|||||||
+10
-7
@@ -75,14 +75,17 @@ func (that *Context) View() {
|
|||||||
if that.Session("user_id").Data != nil {
|
if that.Session("user_id").Data != nil {
|
||||||
that.Log["user_id"] = that.Session("user_id").ToCeilInt()
|
that.Log["user_id"] = that.Session("user_id").ToCeilInt()
|
||||||
}
|
}
|
||||||
ipStr := Substr(that.Req.RemoteAddr, 0, strings.Index(that.Req.RemoteAddr, ":"))
|
|
||||||
//负载均衡优化
|
//负载均衡优化
|
||||||
if ipStr == "127.0.0.1" {
|
ipStr := ""
|
||||||
if that.Req.Header.Get("X-Forwarded-For") != "" {
|
if that.Req.Header.Get("X-Forwarded-For") != "" {
|
||||||
ipStr = that.Req.Header.Get("X-Forwarded-For")
|
ipStr = that.Req.Header.Get("X-Forwarded-For")
|
||||||
} else if that.Req.Header.Get("X-Real-IP") != "" {
|
} else if that.Req.Header.Get("X-Real-IP") != "" {
|
||||||
ipStr = that.Req.Header.Get("X-Real-IP")
|
ipStr = that.Req.Header.Get("X-Real-IP")
|
||||||
}
|
}
|
||||||
|
//负载均衡优化
|
||||||
|
if ipStr == "" {
|
||||||
|
//RemoteAddr := that.Req.RemoteAddr
|
||||||
|
ipStr = Substr(that.Req.RemoteAddr, 0, strings.Index(that.Req.RemoteAddr, ":"))
|
||||||
}
|
}
|
||||||
that.Log["ip"] = ipStr
|
that.Log["ip"] = ipStr
|
||||||
that.Db.Insert("logs", that.Log)
|
that.Db.Insert("logs", that.Log)
|
||||||
|
|||||||
@@ -18,13 +18,13 @@ func Down(url, path, name string, e ...*Error) bool {
|
|||||||
}
|
}
|
||||||
out, err := os.Create(path + name)
|
out, err := os.Create(path + name)
|
||||||
|
|
||||||
if err != nil && e[0] != nil {
|
if err != nil && len(e) != 0 {
|
||||||
e[0].SetError(err)
|
e[0].SetError(err)
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
defer out.Close()
|
defer out.Close()
|
||||||
resp, err := http.Get(url)
|
resp, err := http.Get(url)
|
||||||
if err != nil && e[0] != nil {
|
if err != nil && len(e) != 0 {
|
||||||
e[0].SetError(err)
|
e[0].SetError(err)
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
@@ -32,7 +32,7 @@ func Down(url, path, name string, e ...*Error) bool {
|
|||||||
|
|
||||||
pix, err := ioutil.ReadAll(resp.Body)
|
pix, err := ioutil.ReadAll(resp.Body)
|
||||||
_, err = io.Copy(out, bytes.NewReader(pix))
|
_, err = io.Copy(out, bytes.NewReader(pix))
|
||||||
if err != nil && e[0] != nil {
|
if err != nil && len(e) != 0 {
|
||||||
e[0].SetError(err)
|
e[0].SetError(err)
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|||||||
+40
-10
@@ -8,7 +8,7 @@ import (
|
|||||||
"go.mongodb.org/mongo-driver/mongo/options"
|
"go.mongodb.org/mongo-driver/mongo/options"
|
||||||
)
|
)
|
||||||
|
|
||||||
type mongoDb struct {
|
type MongoDb struct {
|
||||||
Client *mongo.Client
|
Client *mongo.Client
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
DataBase *mongo.Database
|
DataBase *mongo.Database
|
||||||
@@ -16,12 +16,12 @@ type mongoDb struct {
|
|||||||
LastErr error
|
LastErr error
|
||||||
}
|
}
|
||||||
|
|
||||||
func Init(database, url string) (*mongoDb, error) {
|
func GetMongoDb(database, url string) (*MongoDb, error) {
|
||||||
db := mongoDb{}
|
db := MongoDb{}
|
||||||
clientOptions := options.Client().ApplyURI(url)
|
clientOptions := options.Client().ApplyURI(url)
|
||||||
|
|
||||||
db.Ctx = context.TODO()
|
db.Ctx = context.TODO()
|
||||||
// Connect to MongoDB
|
// Connect to MongoDb
|
||||||
var err error
|
var err error
|
||||||
db.Client, err = mongo.Connect(db.Ctx, clientOptions)
|
db.Client, err = mongo.Connect(db.Ctx, clientOptions)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -32,7 +32,7 @@ func Init(database, url string) (*mongoDb, error) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
fmt.Println("Connected to MongoDB!")
|
fmt.Println("Connected to MongoDb!")
|
||||||
//databases, err := db.Client.ListDatabaseNames(db.Ctx, bson.M{})
|
//databases, err := db.Client.ListDatabaseNames(db.Ctx, bson.M{})
|
||||||
//if err != nil {
|
//if err != nil {
|
||||||
// return nil, err
|
// return nil, err
|
||||||
@@ -43,7 +43,7 @@ func Init(database, url string) (*mongoDb, error) {
|
|||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (that *mongoDb) Insert(table string, data interface{}) string {
|
func (that *MongoDb) Insert(table string, data interface{}) string {
|
||||||
collection := that.DataBase.Collection(table)
|
collection := that.DataBase.Collection(table)
|
||||||
re, err := collection.InsertOne(that.Ctx, data)
|
re, err := collection.InsertOne(that.Ctx, data)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -53,7 +53,7 @@ func (that *mongoDb) Insert(table string, data interface{}) string {
|
|||||||
return ObjToStr(re.InsertedID)
|
return ObjToStr(re.InsertedID)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (that *mongoDb) InsertMany(table string, data ...interface{}) Slice {
|
func (that *MongoDb) InsertMany(table string, data ...interface{}) Slice {
|
||||||
collection := that.DataBase.Collection(table)
|
collection := that.DataBase.Collection(table)
|
||||||
re, err := collection.InsertMany(that.Ctx, data)
|
re, err := collection.InsertMany(that.Ctx, data)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -65,7 +65,33 @@ func (that *mongoDb) InsertMany(table string, data ...interface{}) Slice {
|
|||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (that *mongoDb) Get(table string, where Map) Map {
|
func (that *MongoDb) Update(table string, data Map, where Map) int64 {
|
||||||
|
collection := that.DataBase.Collection(table)
|
||||||
|
re, err := collection.UpdateMany(that.Ctx, where, data)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
that.LastErr = err
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
return re.ModifiedCount
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func (that *MongoDb) Delete(table string, where Map) int64 {
|
||||||
|
collection := that.DataBase.Collection(table)
|
||||||
|
re, err := collection.DeleteMany(that.Ctx, where)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
that.LastErr = err
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
return re.DeletedCount
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func (that *MongoDb) Get(table string, where Map) Map {
|
||||||
results := []Map{}
|
results := []Map{}
|
||||||
var cursor *mongo.Cursor
|
var cursor *mongo.Cursor
|
||||||
var err error
|
var err error
|
||||||
@@ -86,10 +112,14 @@ func (that *mongoDb) Get(table string, where Map) Map {
|
|||||||
that.LastErr = err
|
that.LastErr = err
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
return results[0]
|
if len(results) > 0 {
|
||||||
|
|
||||||
|
return results[0]
|
||||||
|
}
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (that mongoDb) Select(table string, where Map, page, pageRow int64) []Map {
|
func (that MongoDb) Select(table string, where Map, page, pageRow int64) []Map {
|
||||||
page = (page - 1) * pageRow
|
page = (page - 1) * pageRow
|
||||||
if page < 0 {
|
if page < 0 {
|
||||||
page = 0
|
page = 0
|
||||||
|
|||||||
Reference in New Issue
Block a user