11f69b8c38
- Replaced the jaevor/go-nanoid library with the updated matoous/go-nanoid/v2 for generating unique IDs. - Implemented user retrieval methods in the user provider, including GetUsers, PaginateUsers, and CountUsers, enhancing user management capabilities. - Improved error handling and ensured default select fields are set for user queries, streamlining the user data access process.
81 lines
2.1 KiB
Go
81 lines
2.1 KiB
Go
package user
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/yaoapp/gou/model"
|
|
"github.com/yaoapp/kun/maps"
|
|
)
|
|
|
|
// User List and Search
|
|
|
|
// GetUsers retrieves users by query parameters (compatible with Model.Get)
|
|
func (u *DefaultUser) GetUsers(ctx context.Context, param model.QueryParam) ([]maps.MapStr, error) {
|
|
// Set default select fields if not provided
|
|
if param.Select == nil {
|
|
param.Select = u.basicUserFields
|
|
}
|
|
|
|
m := model.Select(u.model)
|
|
users, err := m.Get(param)
|
|
if err != nil {
|
|
return nil, fmt.Errorf(ErrFailedToGetUser, err)
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
// PaginateUsers retrieves paginated list of users (compatible with Model.Paginate)
|
|
func (u *DefaultUser) PaginateUsers(ctx context.Context, param model.QueryParam, page int, pagesize int) (maps.MapStr, error) {
|
|
// Set default select fields if not provided
|
|
if param.Select == nil {
|
|
param.Select = u.basicUserFields
|
|
}
|
|
|
|
m := model.Select(u.model)
|
|
result, err := m.Paginate(param, page, pagesize)
|
|
if err != nil {
|
|
return nil, fmt.Errorf(ErrFailedToGetUser, err)
|
|
}
|
|
|
|
return result, nil
|
|
}
|
|
|
|
// CountUsers returns total count of users with optional filters
|
|
func (u *DefaultUser) CountUsers(ctx context.Context, param model.QueryParam) (int64, error) {
|
|
// Use Paginate with a small page size to get the total count
|
|
// This is more reliable than manual COUNT(*) queries
|
|
m := model.Select(u.model)
|
|
result, err := m.Paginate(param, 1, 1) // Get first page with 1 item to get total
|
|
if err != nil {
|
|
return 0, fmt.Errorf(ErrFailedToGetUser, err)
|
|
}
|
|
|
|
// Extract total from pagination result
|
|
if total, ok := result["total"].(int64); ok {
|
|
return total, nil
|
|
}
|
|
|
|
// Handle different total types returned by Paginate
|
|
if totalInterface, ok := result["total"]; ok {
|
|
switch v := totalInterface.(type) {
|
|
case int:
|
|
return int64(v), nil
|
|
case int32:
|
|
return int64(v), nil
|
|
case int64:
|
|
return v, nil
|
|
case uint:
|
|
return int64(v), nil
|
|
case uint32:
|
|
return int64(v), nil
|
|
case uint64:
|
|
return int64(v), nil
|
|
default:
|
|
return 0, fmt.Errorf("unexpected total type: %T", totalInterface)
|
|
}
|
|
}
|
|
|
|
return 0, fmt.Errorf("total not found in pagination result")
|
|
}
|