[我怀疑我正试图以一种OOP的方式来刺耳行为,但我不知道该去成语做我想做的事。
我有一个Message结构,可用于在客户端服务器应用程序中传递数据:
type Message struct {
ID string `json:"id,omitempty"`
Type string `json:"type"`
Data interface{} `json:"data"`
}
这里的数据可以是不同的东西,例如许多命令:
type Command struct {
User *types.UserInfo `json:"user"`
}
type CommandA struct {
Command
A *AData `json:"a_data"`
}
type CommandB struct {
CommandB
B *BData `json:"b_data"`
}
[我要做的是检查消息数据类型是否为命令,并执行所有命令共有的操作,例如授权,所有操作都集中在一个地方,而不必键入断言哪种命令类型,调用适当的命令。处理函数,然后执行身份验证,因为这将导致大量的代码重复。
下面的代码反映了我想要发生的事情。
for {
select {
case m := <-in:
// what I would like to do, obviously not working as
// m.Data is not of type Command but the actual command type
if c, ok := m.Data.(msg.Command); ok {
// do auth and other common stuff
}
switch t := m.Data.(type) {
case *msg.CommandA:
go srv.handleCommandA(m.ID, t)
case *msg.CommandB:
go srv.handleCommandB(m.ID, t)
// etc etc
default:
// do something
}
}
}
我该如何习惯性地解决此问题?
您可以在界面中定义常用命令内容
type Commander interface{
DoCommonStuff()
}
为命令结构实现
func (c Command) DoCommonStuff(){
//do stuff
}
然后断言
if c, ok := m.Data.(Commander); ok {
c.DoCommonStuff()
}
您的其他代码应保持不变
一种方法是使用反射从Data
中提取公共字段值。在您的示例中,由于所有Command
都具有User
字段,因此我们可以使用它来识别Message.Data
是否是命令。如果Command
未嵌入到数据中,则只需返回nil
。示例代码:
func GetUserInfo(v interface{}) *types.UserInfo {
vt := reflect.ValueOf(v)
if vt.Kind() == reflect.Ptr {
vt = vt.Elem()
}
if vt.Kind() != reflect.Struct {
return nil
}
u := vt.FieldByName("User")
if !u.IsValid() {
return nil
}
user, ok := u.Interface().(*types.UserInfo)
if !ok {
return nil
}
return user
}
//Call GetUserInfo then perform common operation
user := GetUserInfo(m.Data)
if user != nil {
//Do auth and other common stuff
}