package kpi_c_title import ( "net/http" "strings" "be.ems/src/framework/datasource" "be.ems/src/framework/vo/result" "github.com/gin-gonic/gin" ) func (k *KpiCTitle) Get(c *gin.Context) { var titles []KpiCTitle var conditions []string var params []any var total int64 // construct condition to get if neType := c.Query("neType"); neType != "" { conditions = append(conditions, "ne_type = ?") params = append(params, strings.ToUpper(neType)) } if status := c.Query("status"); status != "" { conditions = append(conditions, "status = ?") params = append(params, status) } whereSql := "" totalSql := "" if len(conditions) > 0 { whereSql += strings.Join(conditions, " and ") totalSql += strings.Join(conditions, " and ") } data := map[string]any{ "total": 0, "rows": []KpiCTitle{}, } if err := datasource.DefaultDB().Where(whereSql, params...).Find(&titles).Error; err != nil { c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) return } data["rows"] = titles if err := datasource.DefaultDB().Table(k.TableName()).Where(totalSql, params...).Count(&total).Error; err != nil { c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) return } data["total"] = total //c.JSON(http.StatusOK, map[string]any{"data": titles}) c.JSON(http.StatusOK, result.Ok(data)) } func (k *KpiCTitle) Total(c *gin.Context) { var conditions []string var params []any // construct condition to get if neType := c.Query("neType"); neType != "" { conditions = append(conditions, "ne_type = ?") params = append(params, strings.ToUpper(neType)) } if status := c.Query("status"); status != "" { conditions = append(conditions, "status = ?") params = append(params, status) } whereSql := "" if len(conditions) > 0 { whereSql += strings.Join(conditions, " and ") } var total int64 = 0 if err := datasource.DefaultDB().Table(k.TableName()).Where(whereSql, params...).Count(&total).Error; err != nil { c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) return } c.JSON(http.StatusOK, map[string]any{"total": total}) } func (k *KpiCTitle) Post(c *gin.Context) { var title KpiCTitle if err := c.ShouldBindJSON(&title); err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } if err := datasource.DefaultDB().Create(&title).Error; err != nil { c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) return } c.JSON(http.StatusCreated, title) } func (k *KpiCTitle) Put(c *gin.Context) { var title KpiCTitle id := c.Param("id") if err := datasource.DefaultDB().First(&title, id).Error; err != nil { c.JSON(http.StatusNotFound, gin.H{"error": "Title not found"}) return } if err := c.ShouldBindJSON(&title); err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } datasource.DefaultDB().Save(&title) c.JSON(http.StatusOK, title) } func (k *KpiCTitle) Delete(c *gin.Context) { id := c.Param("id") if err := datasource.DefaultDB().Delete(&KpiCTitle{}, id).Error; err != nil { c.JSON(http.StatusNotFound, gin.H{"error": "Title not found"}) return } c.JSON(http.StatusNoContent, nil) // 204 No Content } func GetActiveKPICList(neType string) []KpiCTitle { k := new([]KpiCTitle) err := datasource.DefaultDB().Where("`ne_type` = ? and `status` = 'Active'", neType).Find(&k).Error if err != nil { return nil } return *k }