add: custom kpi feature

features/pm/kpi_c_title/controller.go

@@ -1,6 +1,7 @@
 package kpi_c_title
 
 import (
+	"fmt"
 	"net/http"
 	"strings"
 
@@ -23,29 +24,54 @@ func (k *KpiCTitle) GetToalList(c *gin.Context) {
 	var conditions []string
 	var params []any
 
+	var querys KpiCTitleQuery
+	if err := c.ShouldBindQuery(&querys); err != nil {
+		c.JSON(http.StatusBadRequest, services.ErrResp(err.Error()))
+		return
+	}
+
+	dborm := datasource.DefaultDB().Table(k.TableName())
 	// construct condition to get
-	if neType := c.Query("neType"); neType != "" {
+	if neType := querys.NeType; neType != "" {
 		conditions = append(conditions, "ne_type = ?")
 		params = append(params, strings.ToUpper(neType))
 	}
-	if status := c.Query("status"); status != "" {
+	if status := querys.Status; status != "" {
 		conditions = append(conditions, "status = ?")
 		params = append(params, status)
 	}
 	whereSql := ""
 	if len(conditions) > 0 {
 		whereSql += strings.Join(conditions, " and ")
+		dborm = dborm.Where(whereSql, params...)
 	}
-	if err := datasource.DefaultDB().Where(whereSql, params...).Find(&titles).Error; err != nil {
-		c.JSON(http.StatusInternalServerError, services.ErrResp(err.Error()))
-		return
-	}
+
+	// Get total number
 	var total int64 = 0
-	if err := datasource.DefaultDB().Table(k.TableName()).Where(whereSql, params...).Count(&total).Error; err != nil {
+	if err := dborm.Count(&total).Error; err != nil {
 		c.JSON(http.StatusInternalServerError, services.ErrResp(err.Error()))
 		return
 	}
-	c.JSON(http.StatusOK, services.TotalDataResp(titles, total)
+
+	// page number and size
+	if pageSize := querys.PageSize; pageSize > 0 {
+		dborm = dborm.Limit(pageSize)
+		if pageNum := querys.PageNum; pageNum > 0 {
+			dborm = dborm.Offset((pageNum - 1) * pageSize)
+		}
+	}
+
+	// order by
+	if sortField, sortOrder := querys.SortField, querys.SortOrder; sortField != "" && sortOrder != "" {
+		orderBy := fmt.Sprintf("%s %s", sortField, sortOrder)
+		dborm = dborm.Order(orderBy)
+	}
+	if err := dborm.Find(&titles).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, services.ErrResp(err.Error()))
+		return
+	}
+
+	c.JSON(http.StatusOK, services.TotalDataResp(titles, total))
 	//c.JSON(http.StatusOK, titles)
 }
 
@@ -122,7 +148,7 @@ func (k *KpiCTitle) Put(c *gin.Context) {
 	id := c.Param("id")
 
 	if err := datasource.DefaultDB().First(&title, id).Error; err != nil {
-		c.JSON(http.StatusNotFound, gin.H{CodeKeyName: CodeError, MsgKeyName: "Title not found"})
+		c.JSON(http.StatusNotFound, services.ErrResp("Title not found"))
 		return
 	}
 
@@ -138,7 +164,7 @@ 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{CodeKeyName: CodeError, MsgKeyName: "Title not found"})
+		c.JSON(http.StatusNotFound, services.ErrResp("Title not found"))
 		return
 	}