GAAS_APS_LEO/web/models/buffer_model.go

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
}