使用普遍适用的代码替换 switch 语句中的逻辑

你在学习Golang相关的知识吗？本文《使用普遍适用的代码替换 switch 语句中的逻辑》，主要介绍的内容就涉及到，如果你想提升自己的开发能力，就不要错过这篇文章，大家要知道编程理论基础和实战操作都是不可或缺的哦！

我在代码中有一个执行数据库查询的地方。

代码不是我的，我可能不会就其架构做出决定。

最近，我们接到一项执行几个额外查询的任务。

我的上级决定将额外的查询放在同一段数据库代码中。

现在他们希望我以某种方式优化这部分代码。

所有这些 ifs 和 switches 都“刺穿他们的眼睛”。

我们使用sqlx进行数据库查询。

我决定制作 2 个 maps，查询类型为 key，具体为 struct/sql 作为数据。

这显着减少了视觉噪音，但我来这里是想问我是否还能做点什么？

下面是示例代码，我相信它是不言自明的，但如果需要，请要求澄清：

/*
     I will use UI windows as an example.
     The point is that all window types have some common properties,
         and some properties specific to each window type.
*/

type ProgressBar struct {

    X                   int
    Y                   int
    Width               int
    Height              int

    LowestValue         int
    Highestvalue        int
    CurrentPosition     int
}

type TextBox struct {

    X                   int
    Y                   int
    Width               int
    Height              int

    TextLength          int
    TextContent         string

}

type Calendar struct {

    X                  int
    Y                  int
    Width              int
    Height             int

    SelectedDate       time.Time
}

var (
    // holds specific SQL query for each window type
    queries             map[int] string
    // each window type has it's own function that returns struct for storing query results
    scanStructs         map[int]func() interface{}

    // since Go has no enum type, we have to simulate them like below
    progressbar     =   1
    textbox         =   2
    calendar        =   3

)

func init(){

    queries                     =   make(map[int]string)
    scanStructs                 =   make(map[int]func() interface{})

    scanStructs[progressbar]    =   func() interface{} { return ProgressBar{} }
    scanStructs[textbox]        =   func() interface{} { return TextBox{} }
    scanStructs[calendar]       =   func() interface{} { return Calendar{} }

    queries[progressbar]        =   "SELECT * FROM progress_bars"
    queries[textbox]            =   "SELECT * FROM textboxes"
    queries[calendar]           =   "SELECT * FROM calendars"

}
 
// I am providing non-refactored version for comparison
func OriginalPerformSqlQuery(type int) ([]interface{}, error) {

    query := string

    switch type {
        case progressbar:
            query =     "SELECT * FROM progress_bars"
        case textbox:
            query =     "SELECT * FROM textboxes"
        case calendar:
            query =     "SELECT * FROM calendars"
        default:  
            return interface{}, 
                   fmt.Errorf("unknown window type")
    }

    rows, err := db.Queryx(query)
    if err != nil {
        return interface{}, 
               fmt.Errorf("db.Queryx() failed: %v", err)
    }

    defer func() { _ = rows.Close() }()

    result := make([]interface{}, 0)

    for rows.Next() {

        switch type {
            case progressbar:
                r     :=     ProgressBar{}

                if err := rows.StructScan(&r); err != nil {
                      return interface{}, 
                             fmt.Errorf("rows.StructScan() failed: %v", err)
                }

                result = append(result, r)

            case textbox:
                r     :=     TextBox{}

                if err := rows.StructScan(&r); err != nil {
                      return interface{}, 
                             fmt.Errorf("rows.StructScan() failed: %v", err)
                }

                result = append(result, r)

            case calendar:
                r     :=     Calendar{}

                if err := rows.StructScan(&r); err != nil {
                      return interface{}, 
                             fmt.Errorf("rows.StructScan() failed: %v", err)
                }

                result = append(result, r)
        }
    }

    return result, nil

}

func PerformSqlQuery(type int) ([]interface{}, error) {

    query, exists = queries[type]
    if !exists {
        return interface{}, 
               fmt.Errorf("unknown window type")
    }

    rows, err := db.Queryx(query)
    if err != nil {
        return interface{}, 
               fmt.Errorf("db.Queryx() failed: %v", err)
    }

    defer func() { _ = rows.Close() }()

    result := make([]interface{}, 0)

    for rows.Next() {
        // get required destination struct
        r := scanStructs[type]()

        if err := rows.StructScan(&r); err != nil {
            return interface{}, 
                   fmt.Errorf("rows.StructScan() failed: %v", err)
        }

        result = append(result, r)
    }

    return result, nil

}

正确答案

• 您可以将窗口类型更改为 const 并输入：

type windowtype int

const (
    progressbar windowtype = iota + 1
    textbox
    calendar
)

• 您可以将地图初始化更改为复合文字

type helperData struct {
    query       string
    scanStruct  func() interface{}
}

var helperMap = map[windowType]helperData{
    progressbar: {
        query:       "SELECT * FROM progress_bars",
        scanStruct:  func() interface{} { return ProgressBar{} },
    },
    textbox: {
        query:       "SELECT * FROM textboxes",
        scanStruct:  func() interface{} { return TextBox{} },
    },
    calendar: {
        query:       "SELECT * FROM calendars",
        scanStruct:  func() interface{} { return Calendars{} },
    }
}

理论要掌握，实操不能落！以上关于《使用普遍适用的代码替换 switch 语句中的逻辑》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！