「连载」边缘计算（十四）02-02：边缘部分源码（源码分析篇）

KubeEdge/beehive/pkg/core/core.go

import (

...

"GitHub.com/KubeEdge/beehive/pkg/core/context"

)

// StartModules starts modules that are registered

func StartModules() {

coreContext := context.GetContext(context.MsgCtxTypeChannel)

modules := GetModules()

for name, module := range modules {

//Init the module

coreContext.AddModule(name)

//Assemble typeChannels for sendToGroup

coreContext.AddModuleGroup(name, module.Group())

go module.Start(coreContext)

klog.Infof("Starting module %v", name)

}

}

KubeEdge/beehive/pkg/core/context/contex_factory.go

// GetContext gets global context instance

func GetContext(contextType string) *Context {

once.Do(func() {

context = &Context{}

switch contextType {

case MsgCtxTypeChannel:

channelContext := NewChannelContext()

context.messageContext = channelContext

context.moduleContext = channelContext

default:

klog.Warningf("Do not support context type:%s", contextType)

}

})

return context

}

KubeEdge/beehive/pkg/core/context/context.go

// Context is global context object

type Context struct {

moduleContext  ModuleContext

messageContext MessageContext

}

//ModuleContext is interface for context module management

type ModuleContext interface {

AddModule(module string)

AddModuleGroup(module, group string)

Cleanup(module string)

}

//MessageContext is interface for message syncing

type MessageContext interface {

// async mode

Send(module string, message model.Message)

Receive(module string) (model.Message, error)

// sync mode

SendSync(module string, message model.Message, timeout time.Duration) ( model.Message, error)

SendResp(message model.Message)

// group broadcast

SendToGroup(moduleType string, message model.Message)

SendToGroupSync(moduleType string, message model.Message, timeout time.Duration) error

}