Kurangkan jumlah kod pendua apabila memetakan DTO dalam Go

Kandungan soalan

Saya sedang belajar Go dan akan menghargai pandangan orang tentang cara terbaik untuk mengurangkan jumlah kod pendua.

Struktur folder bahagian yang berkaitan adalah seperti berikut:

.
├── http
│   ├── handlers
│   └── routes
├── models
│   └── dto
├── specifications
└── store
    └── postgres

Dalam folder specations saya, saya mempunyai 2 antara muka "storan":

type TaskStore interface {
    CreateTask(ctx context.Context, input dto.TaskCreate) error
    UpdateTask(ctx context.Context, id int, input dto.TaskUpdate) error
    GetTask(ctx context.Context, id int) (dto.TaskResult, error)
    ListTasks(ctx context.Context) ([]dto.TaskResult, error)
    DeleteTask(ctx context.Context, id int) error
}

type TagStore interface {
    CreateTag(ctx context.Context, input dto.TagCreate) error
    RenameTag(ctx context.Context, id int, input dto.TagUpdate) error
    ListTags(ctx context.Context) ([]dto.TagResult, error)
    GetTag(ctx context.Context, id int) (dto.TagResult, error)
    DeleteTag(ctx context.Context, id int) error
}

store/postgres folder mengandungi pelaksanaan tugas dan tag (corak repositori).

Soalan yang saya nampak:

Dalam folder handlers saya, saya mempunyai struct yang mengambil input daripada salah satu antara muka storan:

type TaskHandler struct {
    store specifications.TaskStore
}

func NewTaskHandler(store specifications.TaskStore) TaskHandler {
    return TaskHandler{
        store: store,
    }
}

type TagHandler struct {
    store specifications.TagStore
}

func NewTagHandler(store specifications.TagStore) TagHandler {
    return TagHandler{
        store: store,
    }
}

Pengendali ini mengandungi kaedah yang akan dipetakan ke laluan api:

func (h TaskHandler) List() http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        tasks, err := h.store.ListTasks(r.Context())
        if err != nil {
            log.Err(err).Msg("failed to retrieve tasks")
            w.WriteHeader(http.StatusInternalServerError)
            return
        }
        render.JSON(w, r, tasks)
    }
}

func (h TagHandler) List() http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        tags, err := h.store.ListTags(r.Context())
        if err != nil {
            log.Err(err).Msg("failed to retrieve tags")
            w.WriteHeader(http.StatusInternalServerError)
            return
        }
        render.JSON(w, r, tags)
    }
}

Anda akan perasan bahawa List kaedah pada setiap pengendali pada asasnya adalah sama, kecuali antara muka yang digunakan oleh setiap kedai.

Bagaimana untuk menukar tetapan ini untuk mengurangkan kod pendua?

Saya pada mulanya fikir saya boleh menggunakan generik untuk menyelesaikan masalah ini, contohnya:

type EntityStore[CreateDto any, UpdateDto any, ResultDto any] interface {
    Create(ctx context.Context, input CreateDto) error
    Update(ctx context.Context, id int, input UpdateDto) error
    List(ctx context.Context) ([]ResultDto, error)
    Get(ctx context.Context, id int) (ResultDto, error)
    Delete(ctx context.Context, id int) error
}

Tetapi ini bermakna memetakan setiap jenis ke dalam pengendali, yang saya rasa bukan penyelesaian praktikal.

Ada sebarang cadangan tentang cara memetakan DTO dan antara muka saya dengan lebih baik?

Jawapan betul

Anda boleh mempunyai fungsi pembantu

func ListHandler[T any](name string, lister func(ctx context.Context) ([]T, error)) func(w http.ResponseWriter, r *http.Request) {
    return func(w http.ResponseWriter, r *http.Request) {
        list, err := lister(r.Context())
        if err != nil {
            log.Err(err).Msg("failed to retrieve " + name)
            w.WriteHeader(http.StatusInternalServerError)
            return
        }
        render.JSON(w, r, list)
    }
}

Maka anda akan mempunyai

func (h TaskHandler) List() http.HandlerFunc {
    return ListHandler("tasks", h.store.ListTasks)
}

func (h TagHandler) List() http.HandlerFunc {
    return ListHandler("tags", h.store.ListTags)
}

Atas ialah kandungan terperinci Kurangkan jumlah kod pendua apabila memetakan DTO dalam Go. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!