解决 golang 中接口和实现类型之间的具体返回类型不兼容问题的方法

编程并不是一个机械性的工作，而是需要有思考，有创新的工作，语法是固定的，但解决问题的思路则是依靠人的思维，这就需要我们坚持学习和更新自己的知识。今天golang学习网就整理分享《解决 golang 中接口和实现类型之间的具体返回类型不兼容问题的方法》，文章讲解的知识点主要包括，如果你对Golang方面的知识点感兴趣，就不要错过golang学习网，在这可以对大家的知识积累有所帮助，助力开发能力的提升。

我仍在学习 go，我决定使用领域驱动设计和界面。我想通过在运行时根据执行环境选择引擎来支持 mysql、postgresql 和 sqlite。由于无论引擎如何，所有数据库操作都是相同的，因此我决定使用 sqlc 来生成数据库代码。

现在我遇到了一些问题，我试图创建一个边界接口 storer 来位于消费者和引擎之间并向消费者公开数据库操作。它不起作用，因为 storer 公开的返回类型与引擎为每个操作（方法）返回的返回类型不兼容。

下面是 finduserbyid 操作的 sqlite 实现示例，该操作应该返回 user 对象：

package sqlite

/* begin sqlc generated code */
type user struct {
    id        string
    firstname string
    lastname  sql.nullstring
    createdat sql.nulltime
}

// finduserbyid is an sql statement
func (q *queries) finduserbyid(ctx context.context, id string) (user, error) {
    row := q.db.queryrowcontext(ctx, finduserbyid, id)
    var i user
    err := row.scan(
        &i.id,
        &i.firstname,
        &i.lastname,
        &i.createdat,
    )
    return i, err
}

/* end sqlc generated code */

func (u *user) getfirstname() string {
    return u.firstname
}


func newuserstorer() *queries { ... }

这是具有 userstorer 接口的 storage 包：

package storage
import "time.time"

type user struct {
    id        string
    firstname string
    lastname  string
    createdat time.time
}
func (u *user) getfirstname() string {
    return u.firstname
}
type userstorer interface {
    finduserbyid(ctx context.context, id string) (user, error)
}

最后，这是 auserstorer 消费者的主包：

package main

import "myproject/storage"
import "myproject/storage/sqlite"
import "fmt"

func showfirstname(s storage.userstorer, id string) {
    u, _ := s.finduserbyid(id) //assuming the user exists
    fmt.println(u.getfirstname())
}

func main(){
    store := sqlute.newuserstorer()
    id := "abc123"
    showfirstname(store, id)
}

现在，当我尝试调用 showfirstname 时，出现此错误：

cannot use store (variable of type *sqlite.Queries) as storage.UserStorer value in argument to ShowFirstName: *sqlite.Queries does not implement storage.UserStorer(wrong type for method FindUserById)
        have FindUserById(context.Context, string) (sqlite.User, error)
        want FindUserById(context.Context, sql.NullString) (storage.User, error)compilerInvalidIfaceAssign

我尝试将 storage.user 类型更改为仅包含 getfirstname() 方法的接口，但这不起作用。

如何确保返回类型相同，并且可以在不更改使用者的情况下将 sqlite 替换为 psql？我无法更改 sqlc 生成的代码和文件来满足类型要求，并且出于可维护性原因并避免与 sql 相关的错误和 bug，我不想手动编写它生成的代码。

正确答案

正如评论中提到的，sqlite 包中 user 定义的 user 类型与 storage 包中定义的 user 类型不同。这是因为 sqlc 生成自己的 user 类型，其字段与数据库表中的列匹配，而存储包中的 user 类型具有与您的域模型匹配的字段。

为了解决这个问题，我们可以在应用程序的域模型中定义一个单独的用户类型，该类型映射到 sqlc 生成的用户类型。然后，您可以在 userstorer 界面和应用程序代码中使用此域模型用户类型。在这里，我们将一个接口适配为另一个接口，以便两个不兼容的接口可以一起工作 (Adapter Pattern)。

package main

// Write imports here

// Define a domain model User type (storage pkg) that maps to the User type generated by sqlc
type User struct {
    ID        string
    FirstName string
    LastName  sql.NullString
    CreatedAt sql.NullTime
}

// Define a UserMapper interface that maps between sqlc User type and domain model User type
type UserMapper interface {
    FromDB(u *sqlite.User) *storage.User
    ToDB(u *storage.User) *sqlite.User
}

// Implement the UserMapper interface
type sqliteUserMapper struct{}

func (m *sqliteUserMapper) FromDB(u *sqlite.User) *storage.User {
    return &storage.User{
        ID:        u.ID,
        FirstName: u.FirstName,
        LastName:  u.LastName.String,
        CreatedAt: u.CreatedAt.Time,
    }
}
func (m *sqliteUserMapper) ToDB(u *storage.User) *sqlite.User {
    return &sqlite.User{
        ID:        u.ID,
        FirstName: u.FirstName,
        LastName:  sql.NullString{String: u.LastName, Valid: u.LastName != ""},
        CreatedAt: sql.NullTime{Time: u.CreatedAt, Valid: !u.CreatedAt.IsZero()},
    }
}

// Modify your UserStorer interface to use the domain model User type
type UserStorer interface {
    FindUserById(ctx context.Context, id string) (*User, error)
}

// Update your sqlite implementation of FindUserById to use the domain model User type and the UserMapper
func (q *Queries) FindUserById(ctx context.Context, id string) (*storage.User, error) {
    row := q.db.QueryRowContext(ctx, findUserById, id)
    var u sqlite.User
    err := row.Scan(&u.ID, &u.FirstName, &u.LastName, &u.CreatedAt)
    if err != nil {
        return nil, err
    }
    return q.mapper.FromDB(&u), nil
}

// Update your ShowFirstName function to use the domain model User type
func ShowFirstName(s storage.UserStorer, id string) {
    u, _ := s.FindUserById(context.Background(), id) //assuming the user exists
    fmt.Println(u.GetFirstName())
}

// Instantiate your sqlite UserStorer with a UserMapper
func NewUserStorer(mapper storage.UserMapper) *Queries {
    return &Queries{
        db:     db,
        mapper: mapper,
    }
}

func main() {
    // all functions in pkg main to understand the logical order of implementation
    mapper := &sqliteUserMapper{}
    store := sqlite.NewUserStorer(mapper)
    id := "abc123"
    ShowFirstName(store, id)
}

以上就是《解决 golang 中接口和实现类型之间的具体返回类型不兼容问题的方法》的详细内容，更多关于的资料请关注golang学习网公众号！