Add Filter Partner

internal/handlers/http/sites/sites.go

@@ -150,7 +150,7 @@ func (h *Handler) GetAll(c *gin.Context) {
 	}
 
 	ctx := request.GetMyContext(c)
-	Sites, total, err := h.service.GetAll(c.Request.Context(), req.ToEntity(ctx))
+	Sites, total, err := h.service.GetAll(c.Request.Context(), req.ToEntity(ctx, req.PartnerID))
 	if err != nil {
 		response.ErrorWrapper(c, err)
 		return
@@ -252,7 +252,7 @@ func (h *Handler) Count(c *gin.Context) {
 	}
 
 	ctx := request.GetMyContext(c)
-	res, err := h.service.Count(ctx, req.ToEntity(ctx))
+	res, err := h.service.Count(ctx, req.ToEntity(ctx, ctx.GetPartnerID()))
 
 	if err != nil {
 		response.ErrorWrapper(c, err)
@@ -290,13 +290,13 @@ func (h *Handler) toSiteResponse(resp *entity.Site) response.Site {
 }
 
 func (h *Handler) toSiteResponseList(resp []*entity.Site, total int64, req request.SiteParam) response.SiteList {
-	var Sites []response.Site
+	sites := []response.Site{}
 	for _, b := range resp {
-		Sites = append(Sites, h.toSiteResponse(b))
+		sites = append(sites, h.toSiteResponse(b))
 	}
 
 	return response.SiteList{
-		Sites:  Sites,
+		Sites:  sites,
 		Total:  total,
 		Limit:  req.Limit,
 		Offset: req.Offset,

internal/handlers/request/site.go

@@ -60,19 +60,20 @@ func (r *Site) ToEntity(createdBy int64) *entity.Site {
 }
 
 type SiteParam struct {
-	Search string `form:"search"`
-	Name   string `form:"name"`
-	Limit  int    `form:"limit,default=10"`
-	Offset int    `form:"offset,default=0"`
+	Search    string `form:"search"`
+	PartnerID *int64 `form:"partner_id"`
+	Name      string `form:"name"`
+	Limit     int    `form:"limit,default=10"`
+	Offset    int    `form:"offset,default=0"`
 }
 
-func (r *SiteParam) ToEntity(ctx mycontext.Context) entity.SiteSearch {
+func (r *SiteParam) ToEntity(ctx mycontext.Context, partnerID *int64) entity.SiteSearch {
 	return entity.SiteSearch{
-		PartnerID: ctx.GetPartnerID(),
-		IsAdmin: ctx.IsAdmin(),
-		Search: r.Search,
-		Name:   r.Name,
-		Limit:  r.Limit,
-		Offset: r.Offset,
+		PartnerID: partnerID,
+		IsAdmin:   ctx.IsAdmin(),
+		Search:    r.Search,
+		Name:      r.Name,
+		Limit:     r.Limit,
+		Offset:    r.Offset,
 	}
 }