52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package mapper
|
|
|
|
import (
|
|
"github.com/jinzhu/copier"
|
|
)
|
|
|
|
type CopierMapper[DTO any, MODEL any] struct {
|
|
copierOption copier.Option
|
|
}
|
|
|
|
func NewCopierMapper[DTO any, MODEL any]() *CopierMapper[DTO, MODEL] {
|
|
return &CopierMapper[DTO, MODEL]{
|
|
copierOption: copier.Option{
|
|
Converters: []copier.TypeConverter{},
|
|
},
|
|
}
|
|
}
|
|
|
|
func (m *CopierMapper[DTO, MODEL]) AppendConverter(converter copier.TypeConverter) {
|
|
m.copierOption.Converters = append(m.copierOption.Converters, converter)
|
|
}
|
|
|
|
func (m *CopierMapper[DTO, MODEL]) AppendConverters(converters []copier.TypeConverter) {
|
|
m.copierOption.Converters = append(m.copierOption.Converters, converters...)
|
|
}
|
|
|
|
func (m *CopierMapper[DTO, MODEL]) ToModel(dto *DTO) *MODEL {
|
|
if dto == nil {
|
|
return nil
|
|
}
|
|
|
|
var model MODEL
|
|
if err := copier.CopyWithOption(&model, dto, m.copierOption); err != nil {
|
|
panic(err) // Handle error appropriately in production code
|
|
}
|
|
|
|
return &model
|
|
}
|
|
|
|
func (m *CopierMapper[DTO, MODEL]) ToDto(model *MODEL) *DTO {
|
|
if model == nil {
|
|
return nil
|
|
}
|
|
|
|
var dto DTO
|
|
if err := copier.CopyWithOption(&dto, model, m.copierOption); err != nil {
|
|
panic(err) // Handle error appropriately in production code
|
|
}
|
|
|
|
return &dto
|
|
}
|