如何在Go中避免长的switch-case语句

时间:2017-06-28 20:50:31

标签: go

我正在Go中编写一个聊天机器人,并想知道如何避免类似于此的长切换案例声明:

switch {

// @bot search me HMAC
case strings.Contains(message, "search me"):
    query := strings.Split(message, "search me ")[1]
    return webSearch(query), "html"

// @bot thesaurus me challenge
case strings.Contains(message, "thesaurus me"):
    query := strings.Split(message, "thesaurus me ")[1]
    return synonyms(query), "html"

我应该在一个单独的包中定义这些处理程序,还是应该只使用结构和接口?哪种方法可以让我拥有一个好的结构,避免使用switch-case并让外部开发人员更容易创建处理程序?

我认为软件包将是一个更好的选择,但我不确定如何使用主机器人注册处理程序。希望得到一个例子。

1 个答案:

答案 0 :(得分:9)

您可以使用map[string]command类似于net/http包注册处理程序的方式。类似于此的东西:

https://play.golang.org/p/9YzHyLodAQ

package main

import (
    "fmt"
    "errors"
)

type BotFunc func(string) (string, error)

type BotMap map[string]BotFunc

var Bot = BotMap{}

func (b BotMap) RegisterCommand(command string, f BotFunc) error {
    if _, exists := b[command]; exists {
        return errors.New("command already exists")
    }
    b[command] = f
    return nil
}

func (b BotMap) Execute(statement string) (string, error) {
    // parse out command and query however you choose (not this way obviously)
    command := statement[:9]
    query := statement[10:]

    return b.ExecuteQuery(command, query)
}

func (b BotMap) ExecuteQuery(command, query string) (string, error) {
    if com, exists := b[command]; exists {
        return com(query)
    }
    return "", errors.New("command doesn't exist")

}

func main() {
    err := Bot.RegisterCommand("search me", func(query string) (string, error) {
        fmt.Println("search", query)
        return "searched", nil
    })
    if err != nil {
        fmt.Println(err)
        return
    }
    err = Bot.RegisterCommand("thesaurus me", func(query string) (string, error) {
        fmt.Println("thesaurus", query)
        return "thesaurused", nil
    })
    if err != nil {
        fmt.Println(err)
        return
    }

    result, err := Bot.Execute("search me please")
    if err != nil {
        fmt.Println(err)
        return
    }
    fmt.Println(result)
}

显然这里缺少很多支票,但这是基本的想法。