wms-be/modules/maintenance_group/query/maintenance_group_query.go

85 lines
2.4 KiB
Go

package query
import (
"github.com/Caknoooo/go-pagination"
"gorm.io/gorm"
)
type M_MaintenanceGroup struct {
ID string `json:"id"`
Name string `json:"name"`
Code string `json:"code"`
Description string `json:"description"`
TotalMaintenanceGroupRole int `json:"total_maintenance_group_role"`
}
type MaintenanceGroupFilter struct {
pagination.BaseFilter
Name string `form:"name"` // tambahkan ini
Code string `form:"code"` // tambahkan ini
Description string `form:"description"` // tambahkan ini
ClientID string `form:"client_id"` // tambahkan ini
Includes []string `form:"includes"` // tambahkan ini
}
func (f *MaintenanceGroupFilter) ApplyFilters(query *gorm.DB) *gorm.DB {
// Apply your filters here
if f.Name != "" {
query = query.Where("name ILIKE ?", "%"+f.Name+"%")
}
if f.ClientID != "" {
query = query.Where("client_id = ?", f.ClientID)
}
// Manual preload untuk roles dengan field terbatas
// for _, include := range f.Includes {
// if include == "Roles" {
// query = query.Preload("Roles", func(db *gorm.DB) *gorm.DB {
// return db.Select("id", "name") // Hanya ambil id dan name
// })
// }
// }
// Hitung total maintenance group role per group
query = query.Select("m_maintenance_groups.*, " +
"(SELECT COUNT(*) FROM m_maintenance_group_roles mgr WHERE mgr.maintenance_group_id = m_maintenance_groups.id AND mgr.deleted_at IS NULL) as total_maintenance_group_role")
return query
}
func (f *MaintenanceGroupFilter) GetTableName() string {
return "m_maintenance_groups"
}
func (f *MaintenanceGroupFilter) GetSearchFields() []string {
return []string{"name"}
}
func (f *MaintenanceGroupFilter) GetDefaultSort() string {
return "id asc"
}
func (f *MaintenanceGroupFilter) GetIncludes() []string {
return f.Includes
}
func (f *MaintenanceGroupFilter) GetPagination() pagination.PaginationRequest {
return f.Pagination
}
func (f *MaintenanceGroupFilter) Validate() {
var validIncludes []string
allowedIncludes := f.GetAllowedIncludes()
for _, include := range f.Includes {
if allowedIncludes[include] {
validIncludes = append(validIncludes, include)
}
}
f.Includes = validIncludes
}
func (f *MaintenanceGroupFilter) GetAllowedIncludes() map[string]bool {
return map[string]bool{
"Client": true,
}
}