增加图片数量询问接口

api/v1/levelMonitor/image_record.go

@@ -14,7 +14,7 @@ type ImageRecordApi struct {
 }
 
 //创建图片记录
-func (a *ImageRecordApi) CreateImageRecord (c *gin.Context)  {
+func (a *ImageRecordApi) CreateImageRecord(c *gin.Context) {
 	var record levelMonitor.ImageRecord
 	_ = c.ShouldBindJSON(&record)
 	if err := utils.Verify(record, utils.ImageRecordVerify); err != nil {
@@ -29,12 +29,24 @@ func (a *ImageRecordApi) CreateImageRecord (c *gin.Context)  {
 	}
 }
 
+func (a *ImageRecordApi) UploadOrNot(c *gin.Context) {
+	var record levelMonitor.ImageRecord
+	_ = c.ShouldBindJSON(&record)
+	if record.TaskId == 0 {
+		response.FailWithMessage("任务id不能为空", c)
+	}
+	if err := imageRecordService.UploadOrNot(record); err != nil {
+		response.FailWithMessage("不用上传", c)
+	} else {
+		response.OkWithMessage("可以上传", c)
+	}
+}
 
 //删除三天前的图片记录
 //func (a *ImageRecordApi) DeleteExpireImageRecord (c *gin.Context)  {}
 
 //查询图片记录列表
-func (a *ImageRecordApi) GetImageRecordList (c *gin.Context)  {
+func (a *ImageRecordApi) GetImageRecordList(c *gin.Context) {
 	var paramsInfo request.SearchImageRecordParams
 	_ = c.ShouldBindJSON(&paramsInfo)
 	if err := utils.Verify(paramsInfo.PageInfo, utils.PageInfoVerify); err != nil {
@@ -54,4 +66,3 @@ func (a *ImageRecordApi) GetImageRecordList (c *gin.Context)  {
 		}, "获取成功", c)
 	}
 }
-

router/levelMonitor/image_record.go

@@ -14,6 +14,7 @@ func (r *ImageRecordRouter) InitImageRecordRouter(Router *gin.RouterGroup) {
 	{
 		imageRecordRouter.POST("getImageRecordList", imageRecordApi.GetImageRecordList) //获取图片记录列表
 		imageRecordRouter.POST("createImageRecord", imageRecordApi.CreateImageRecord)   //创建图片记录
+		imageRecordRouter.POST("uploadOrNot", imageRecordApi.UploadOrNot)               //询问是否上传图片
 	}
 	//imageRecordRouter1 := Router.Group("levelMonitor").Use(middleware.OperationRecord())
 	//{

service/cache/cache.go

@@ -9,6 +9,7 @@ import (
 
 const (
 	TemporaryTaskPcCode = "%s:temporaryTaskPcCode:%s"
+	UploadImageNum      = "%s:UploadImageNum:%d"
 )
 
 type Cache struct {

service/levelMonitor/image_record.go

@@ -12,6 +12,7 @@ import (
 	"log-server/model/common/request"
 	"log-server/model/levelMonitor"
 	levelRequest "log-server/model/levelMonitor/request"
+	"log-server/service/cache"
 	"net/http"
 	"net/url"
 	"strconv"
@@ -20,12 +21,19 @@ import (
 )
 
 type ImageRecordService struct {
+	cache cache.Cache
 }
 
 //创建图片记录
 func (s *ImageRecordService) CreateImageRecord(record levelMonitor.ImageRecord) (err error) {
 	//获取今日日期
 	date := time.Now().Format("2006-01-02")
+	key := fmt.Sprintf(cache.UploadImageNum, date, record.TaskId)
+	ctx := context.Background()
+	num, _ := s.cache.GetCacheNum(ctx, key)
+	if num > 50 {
+		return errors.New("数据上传已到上限")
+	}
 	var entity levelMonitor.ImageRecord
 	//创建日期赋值
 	record.CreateDate = date
@@ -39,10 +47,22 @@ func (s *ImageRecordService) CreateImageRecord(record levelMonitor.ImageRecord)
 	if err != nil {
 		return err
 	}
+	s.cache.SetCacheNum(ctx, key)
 	go s.ImageIdentify(record)
 	return
 }
 
+func (s *ImageRecordService) UploadOrNot(record levelMonitor.ImageRecord) (err error) {
+	date := time.Now().Format("2006-01-02")
+	key := fmt.Sprintf(cache.UploadImageNum, date, record.TaskId)
+	ctx := context.Background()
+	num, _ := s.cache.GetCacheNum(ctx, key)
+	if num > 50 {
+		return errors.New("数据上传已到上限")
+	}
+	return nil
+}
+
 //删除三日前的图片记录
 func (s *ImageRecordService) DeleteExpireImageRecord() {
 	markTime := time.Now().Add(-time.Hour * 48).Format("2006-01-02")