Skip to content

Commit

Permalink
mapper framework fix
Browse files Browse the repository at this point in the history
Signed-off-by: cl2017 <[email protected]>
  • Loading branch information
cl2017 authored and wbc6080 committed Nov 27, 2023
1 parent ed4634c commit e3018fa
Show file tree
Hide file tree
Showing 7 changed files with 8 additions and 11 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -9,9 +9,9 @@ import (
"github.com/kubeedge/Template/device"
"github.com/kubeedge/Template/pkg/common"
"github.com/kubeedge/Template/pkg/config"
"github.com/kubeedge/Template/pkg/grpcclient"
"github.com/kubeedge/Template/pkg/grpcserver"
"github.com/kubeedge/Template/pkg/httpserver"
"github.com/kubeedge/Template/pkg/util/grpcclient"
"github.com/kubeedge/Template/pkg/util/parse"
)

Expand Down Expand Up @@ -50,7 +50,7 @@ func main() {
// if dev init mode is register, mapper's dev will init when registry to edgecore
if c.DevInit.Mode != common.DevInitModeRegister {
klog.Infoln("======dev init mode is not register, will register to edgecore")
if _, _, err = grpcclient.RegisterMapper(&c, false); err != nil {
if _, _, err = grpcclient.RegisterMapper(false); err != nil {
klog.Fatal(err)
}
klog.Infoln("registerMapper finished")
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -277,7 +277,7 @@ func (d *DevPanel) DevInit(cfg *config.Config) error {
// return err
// }
case common.DevInitModeRegister:
if err := parse.ParseByUsingRegister(cfg, devs, d.models, d.protocols); err != nil {
if err := parse.ParseByUsingRegister(devs, d.models, d.protocols); err != nil {
return err
}
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ import (
"github.com/kubeedge/Template/driver"
"github.com/kubeedge/Template/pkg/common"
dmiapi "github.com/kubeedge/Template/pkg/dmi-api"
"github.com/kubeedge/Template/pkg/util/grpcclient"
"github.com/kubeedge/Template/pkg/grpcclient"
"github.com/kubeedge/Template/pkg/util/parse"
)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -9,12 +9,11 @@ import (
"google.golang.org/grpc"

"github.com/kubeedge/Template/pkg/common"
"github.com/kubeedge/Template/pkg/config"
dmiapi "github.com/kubeedge/Template/pkg/dmi-api"
)

// RegisterMapper if withData is true, edgecore will send device and model list.
func RegisterMapper(cfg *config.Config, withData bool) ([]*dmiapi.Device, []*dmiapi.DeviceModel, error) {
func RegisterMapper(withData bool) ([]*dmiapi.Device, []*dmiapi.DeviceModel, error) {
// connect grpc server
conn, err := grpc.Dial(cfg.Common.EdgeCoreSock,
grpc.WithInsecure(),
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,17 +22,15 @@ import (
"k8s.io/klog/v2"

"github.com/kubeedge/Template/pkg/common"
"github.com/kubeedge/Template/pkg/config"
"github.com/kubeedge/Template/pkg/util/grpcclient"
"github.com/kubeedge/Template/pkg/grpcclient"
)

var ErrEmptyData error = errors.New("device or device model list is empty")

func ParseByUsingRegister(cfg *config.Config,
devices map[string]*common.DeviceInstance,
func ParseByUsingRegister(devices map[string]*common.DeviceInstance,
dms map[string]common.DeviceModel,
protocols map[string]common.ProtocolConfig) error {
deviceList, deviceModelList, err := grpcclient.RegisterMapper(cfg, true)
deviceList, deviceModelList, err := grpcclient.RegisterMapper(true)
if err != nil {
return err
}
Expand Down

0 comments on commit e3018fa

Please sign in to comment.