How to make gqlgen dataloader Loader.Load(key) return a function that returns an explicit data type instead of interface{}?

huangapple
go评论109阅读模式
英文:

How to make gqlgen dataloader Loader.Load(key) return a function that returns an explicit data type instead of interface{}?

问题

我想使用graph-gophers/dataloader来实现一个用于批量获取用户的数据加载器。

我已经按照以下方式实现了它，但是我需要返回一个明确的数据类型，而不是interface{}。

首先，我使用给定的dataloader.Keys通过函数GetUsers创建了UserReader：

  1. type UserReader struct {
  2. 	userCtrl user.Controller // 这是我的用户服务，用于创建、获取和删除用户
  3. }
  5. func newUserReader(userCtrl user.Controller) UserReader {
  6. 	return UserReader{
  7. 		userCtrl: userCtrl,
  8. 	}
  9. }
  11. // GetUsers根据给定的键返回结果列表
  12. func (reader UserReader) GetUsers(ctx context.Context, keys dataloader.Keys) []*dataloader.Result {
  13. 	output := make([]*dataloader.Result, len(keys))
  14. 	for i, key := range keys {
  15. 		userID := key.String()
  16.         
  17.         // 根据字符串userID获取用户列表
  18. 		result, err := reader.userCtrl.GetUsers(ctx, userID)
  19. 		output[i] = &dataloader.Result{Data: result, Error: err}
  20. 	}
  22. 	return output
  23. }

然后，我使用函数GetUser创建了一个Loader，通过userID获取一个用户，它不会在单个请求中多次请求userCtrl的数据。

  1. type Loader struct {
  2. 	userLoader *dataloader.Loader
  3. }
  5. func newLoader(userCtrl user.Controller) Loader {
  6. 	userReader := newUserReader(userCtrl)
  7. 	return Loader{
  8.         // 使用批处理函数创建新的用户加载器
  9. 		userLoader: dataloader.NewBatchedLoader(userReader.GetUsers),
  10. 	}
  11. }
  13. // UserLoader返回用户加载器
  14. func (loader Loader) UserLoader() *dataloader.Loader {
  15. 	return loader.userLoader
  16. }
  18. // GetUser根据给定的参数返回一个用户
  19. func (loader impl) GetUser(ctx context.Context, userID string) (model.User, error) {
  20. 	// FromCtx从上下文中获取加载器（如果存在），否则返回nil。
  21.     ctxLoader := FromCtx(ctx)
  22.     
  23.     // Thunk是带有interface{}类型和错误的函数返回值
  24. 	thunk := ctxLoader.UserLoader().Load(ctx, dataloader.StringKey(userID))
  25. 	result, err := thunk()
  26. 	if err != nil {
  27. 		return nil, err
  28. 	}
  29. 	return result.(model.User), nil // 在这里转换数据类型！！！
  30. }

最后，可以这样使用：

  1. dataloader.FromCtx(ctx).GetUser(ctx, userID)

但是我希望userReader.getUsers返回一个包含model.user的对象，而不是interface{}，因为我不想在Loader.getUser函数中进行数据类型转换。

有没有办法让dataloader.Result返回特定的数据类型，而不是interface{}？

英文:

I want to use graph-gophers/dataloader to implement a dataloader for bulk-fetching users.

I have implemented it like below but I need to return an explicit data type, not an interface{}

Firstly, I create UserReader with function GetUsers by given dataloader.Keys:

  1. type UserReader struct {
  2. 	userCtrl user.Controller // This is my user service for Create, Get, Delete user
  3. }
  5. func newUserReader(userCtrl user.Controller) UserReader {
  6. 	return UserReader{
  7. 		userCtrl: userCtrl,
  8. 	}
  9. }
  11. // GetUsers return a list of result by given keys
  12. func (reader UserReader) GetUsers(ctx context.Context, keys dataloader.Keys) []*dataloader.Result {
  13. 	output := make([]*dataloader.Result, len(keys))
  14. 	for i, key := range keys {
  15. 		userID := key.String()
  16.         
  17.         // Get list of user by string userID
  18. 		result, err := reader.userCtrl.GetUsers(ctx, userID)
  19. 		output[i] = &dataloader.Result{Data: result, Error: err}
  20. 	}
  22. 	return output
  23. }

And then, I create a Loader with function GetUser to get a user by userID, it won't request data from userCtrl more than 1 time in a single request.

  1. type Loader struct {
  2. 	userLoader *dataloader.Loader
  3. }
  5. func newLoader(userCtrl user.Controller) Loader {
  6. 	userReader := newUserReader(userCtrl)
  7. 	return Loader{
  8.         // Create new user loader with batch function
  9. 		userLoader: dataloader.NewBatchedLoader(userReader.GetUsers),
  10. 	}
  11. }
  13. // UserLoader returns the user loader
  14. func (loader Loader) UserLoader() *dataloader.Loader {
  15. 	return loader.userLoader
  16. }
  18. // GetUser return a user by given parameter
  19. func (loader impl) GetUser(ctx context.Context, userID string) (model.User, error) {
  20. 	// FromCtx gets a Loader from context if it exists, else nil.
  21.     ctxLoader := FromCtx(ctx)
  22.     
  23.     // Thunk is function return value with interface{} type and error
  24. 	thunk := ctxLoader.UserLoader().Load(ctx, dataloader.StringKey(userID))
  25. 	result, err := thunk()
  26. 	if err != nil {
  27. 		return nil, err
  28. 	}
  29. 	return result.(model.User), nil // Convert datatype here!!!
  30. }

And finally, It can be use

  1. dataloader.FromCtx(ctx).GetUser(ctx, userID)

But I need userReader.getUsers to return an object containing model.user not interface{} because I don't want to convert the data type in the Loader.getUser function.

Is there a way to have dataloader.Result return a certain data type instead of the interface{}?

答案1

得分: 2

graph-gophers/dataloader的v7版本支持泛型。具体来说，BatchFunc签名已经改变，因此您可以指定结果类型。

英文:

The v7 version of graph-gophers/dataloader supports generics. Specifically, the BatchFunc signature has changed so you can specify the resulting type.

huangapple
  • 本文由 发表于 2022年8月22日 19:50:37
  • 转载请务必保留本文链接：https://go.coder-hub.com/73444645.html
  • go
  • gqlgen
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定