hoteas/iedc-go
1
0
Fork 0
forked from golang/hotime
Code Issues Pull Requests Projects Releases Wiki Activity

修复下载插件bug

Browse Source
This commit is contained in:
hoteas 2022-08-31 19:10:08 +08:00
parent 65f062b860
commit c71973850e
2 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dri/download/download.go
View File

@ -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
} }

24
dri/mongodb/mongodb.go
View File

@ -1,4 +1,4 @@
package mongodb package libs
import ( import (
. "code.hoteas.com/golang/hotime/common" . "code.hoteas.com/golang/hotime/common"
@ -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,7 @@ func (that *mongoDb) InsertMany(table string, data ...interface{}) Slice {
} }
func (that *mongoDb) Update(table string, data Map, where Map) int64 { func (that *MongoDb) Update(table string, data Map, where Map) int64 {
collection := that.DataBase.Collection(table) collection := that.DataBase.Collection(table)
re, err := collection.UpdateMany(that.Ctx, where, data) re, err := collection.UpdateMany(that.Ctx, where, data)
@ -78,7 +78,7 @@ func (that *mongoDb) Update(table string, data Map, where Map) int64 {
} }
func (that *mongoDb) Delete(table string, where Map) int64 { func (that *MongoDb) Delete(table string, where Map) int64 {
collection := that.DataBase.Collection(table) collection := that.DataBase.Collection(table)
re, err := collection.DeleteMany(that.Ctx, where) re, err := collection.DeleteMany(that.Ctx, where)
@ -91,7 +91,7 @@ func (that *mongoDb) Delete(table string, where Map) int64 {
} }
func (that *mongoDb) Get(table string, where Map) Map { 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
@ -119,7 +119,7 @@ func (that *mongoDb) Get(table string, where Map) Map {
return nil 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