You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
169 lines
4.8 KiB
169 lines
4.8 KiB
package models
|
|
|
|
import (
|
|
"leit.com/LAPP_GAAS_GFrame/conf"
|
|
"leit.com/LAPP_GAAS_GFrame/db"
|
|
"leit.com/LAPP_GAAS_GFrame/utils"
|
|
"fmt"
|
|
"context"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
"log"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type Buffer struct {
|
|
Orderid string `bson:"orderid" json:"buffer-orderid"` //流水号id
|
|
Eid string `bson:"eid" json:"buffer-eid"` //映射关系
|
|
Data string `bson:"data" json:"buffer-data"` //json串
|
|
TimeStamp string `bson:"timestamp" json:"buffer-timestamp"` //时间戳
|
|
Todb string `bson:"todb" json:"buffer-todb"` //导入数据库
|
|
Todrivername string `bson:"todrivername" json:"buffer-todrivername"` //导入数据库引擎
|
|
Totable string `bson:"totable" json:"buffer-totable"` //导入数据表
|
|
Flag int `bson:"flag" json:"buffer-flag"` //是否导入状态
|
|
Status string `bson:"status" json:"buffer-status"` //ok or error
|
|
Message string `bson:"message" json:"buffer-message"` //错误信息
|
|
}
|
|
|
|
func (t *Buffer) TableName() string {
|
|
return "buffer"
|
|
}
|
|
|
|
func (t *Buffer) InsertRecord() (insertID primitive.ObjectID) {
|
|
//1.初始化链接
|
|
client := db.MgoDb()
|
|
//2.选择数据库，数据表
|
|
collect := client.Database(conf.DbConfig.Mongdbname).Collection("buffer")
|
|
insertRest, err := collect.InsertOne(context.TODO(), t)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
insertID = insertRest.InsertedID.(primitive.ObjectID)
|
|
return insertID
|
|
}
|
|
|
|
func (t *Buffer) FindList(logred SerchData, skip int64, limit int64) ([]*Buffer, int64) {
|
|
res := bson.M{}
|
|
// 创建需要过滤的条件
|
|
if logred.Key == "message" {
|
|
res = bson.M{"message": primitive.Regex{Pattern: logred.Val}}
|
|
}
|
|
if logred.Key == "totable" {
|
|
res = bson.M{"totable": primitive.Regex{Pattern: logred.Val}}
|
|
}
|
|
if logred.Key == "status" {
|
|
res = bson.M{"status": primitive.Regex{Pattern: logred.Val}}
|
|
}
|
|
if logred.Key == "timestamp" {
|
|
times := strings.Split(logred.Val, "-")
|
|
stime := time.Unix(int64(utils.ValueToInt(times[0], 0)), 0)
|
|
etime := time.Unix(int64(utils.ValueToInt(times[1], 0)), 0)
|
|
stimestr := utils.TimeFormat(stime, "yyyyMMddHHmmss")
|
|
etimestr := utils.TimeFormat(etime, "yyyyMMddHHmmss")
|
|
res = bson.M{"timestamp": bson.M{"$lte": etimestr, "$gte": stimestr}}
|
|
}
|
|
//1.初始化链接
|
|
client := db.MgoDb()
|
|
//2.选择数据库，数据表
|
|
collect := client.Database(conf.DbConfig.Mongdbname).Collection("buffer")
|
|
|
|
//var skip int64 = 0//从那个开始
|
|
//var limit int64 = 2//炼制几个输出字段
|
|
cursor, err := collect.Find(context.TODO(), res, &options.FindOptions{
|
|
Skip: &skip,
|
|
Limit: &limit,
|
|
Sort: bson.D{{"timestamp", -1}},
|
|
})
|
|
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return nil, 0
|
|
}
|
|
defer cursor.Close(context.TODO())
|
|
//创建需要反序列化成什么样子的结构体对象
|
|
records := make([]*Buffer, 0)
|
|
for cursor.Next(context.TODO()) {
|
|
record := &Buffer{}
|
|
//反序列化
|
|
err = cursor.Decode(record)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return nil, 0
|
|
}
|
|
//追加
|
|
records = append(records, record)
|
|
}
|
|
// 获取数据总数
|
|
count, err := collect.CountDocuments(context.Background(), bson.D{})
|
|
if err != nil {
|
|
log.Fatal(count)
|
|
}
|
|
return records, count
|
|
}
|
|
|
|
func (t *Buffer) FindData() []*Buffer {
|
|
limit := int64(500)
|
|
skip := int64(0)
|
|
//1.初始化链接
|
|
client := db.MgoDb()
|
|
//2.选择数据库，数据表
|
|
collect := client.Database(conf.DbConfig.Mongdbname).Collection("buffer")
|
|
|
|
//var skip int64 = 0//从那个开始
|
|
//var limit int64 = 2//炼制几个输出字段
|
|
cursor, err := collect.Find(context.TODO(), bson.D{
|
|
{"flag", 0},
|
|
{"status", "ok"},
|
|
}, &options.FindOptions{
|
|
Skip: &skip,
|
|
Limit: &limit,
|
|
Sort: bson.D{{"timestamp", -1}},
|
|
})
|
|
fmt.Println(cursor)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
defer cursor.Close(context.TODO())
|
|
//创建需要反序列化成什么样子的结构体对象
|
|
records := make([]*Buffer, 0)
|
|
for cursor.Next(context.TODO()) {
|
|
record := &Buffer{}
|
|
//反序列化
|
|
err = cursor.Decode(record)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return nil
|
|
}
|
|
//追加
|
|
records = append(records, record)
|
|
}
|
|
return records
|
|
}
|
|
|
|
func (t *Buffer) UpdateData() error {
|
|
//1.初始化链接
|
|
client := db.MgoDb()
|
|
//2.选择数据库，数据表
|
|
collect := client.Database(conf.DbConfig.Mongdbname).Collection("buffer")
|
|
|
|
// 修改一条数据，如果不存在则插入
|
|
new := &Buffer{
|
|
Orderid: t.Orderid,
|
|
Data: t.Data,
|
|
TimeStamp: t.TimeStamp,
|
|
Totable: t.Totable,
|
|
Todb: t.Todb,
|
|
Status: t.Status,
|
|
Message: t.Message,
|
|
Flag: 1,
|
|
}
|
|
update := bson.M{"$set": new}
|
|
_, err := collect.UpdateOne(context.Background(), bson.M{"orderid": new.Orderid}, update)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|