去掉无用初始化

zhangqian

2023-10-19 cef1a1d6e4aeb7189ff05370139ff367736b7e92

 source/menu.go

@@ -1,52 +1,53 @@
package system

import (
   "aps_crm/model"
   "aps_crm/pkg/logx"
   "aps_crm/service"
   "context"
   "github.com/pkg/errors"
   "gorm.io/gorm"
)

type initMenu struct{}

// auto run
func init() {
   service.RegisterInit(initMenuSequence, &initMenu{})
}

func (i initMenu) InitializerName() string {
   return model.Menu{}.TableName()
}

func (i *initMenu) InitializeData(ctx context.Context) (next context.Context, err error) {
   entities := []*model.Menu{
      {ID: 1, ParentId: 0, Path: "", Name: "", Title: "客户管理"},
      {ID: 2, ParentId: 1, Path: "/client", Name: "", Title: "客户管理"},

      {ID: 3, ParentId: 0, Path: "", Name: "", Title: "销售管理"},
      {ID: 4, ParentId: 3, Path: "/saleChance", Name: "", Title: "销售机会"},

      {ID: 5, ParentId: 0, Path: "", Name: "", Title: "服务管理"},
      {ID: 6, ParentId: 5, Path: "/serviceContract", Name: "", Title: "服务合同"},

      {ID: 7, ParentId: 0, Path: "", Name: "", Title: "后台设置"},
      {ID: 8, ParentId: 7, Path: "/member", Name: "", Title: "成员管理"},
      {ID: 9, ParentId: 7, Path: "/role", Name: "", Title: "角色管理"},
   }
   if err = model.NewMenuSearch(nil).CreateBatch(entities); err != nil {
      return ctx, errors.Wrap(err, i.InitializerName()+"表数据初始化失败!")
   }
   next = context.WithValue(ctx, i.InitializerName(), entities)
   logx.Infof("InitializeData success initName:%v", i.InitializerName())
   return next, nil
}

func (i *initMenu) DataInserted(ctx context.Context) bool {
   _, err := model.NewMenuSearch(nil).SetId(22).First()
   if errors.Is(err, gorm.ErrRecordNotFound) { // 判断是否存在数据
      return false
   }
   return true
}
//
//import (
//   "aps_crm/model"
//   "aps_crm/pkg/logx"
//   "aps_crm/service"
//   "context"
//   "github.com/pkg/errors"
//   "gorm.io/gorm"
//)
//
//type initMenu struct{}
//
//// auto run
//func init() {
//   service.RegisterInit(initMenuSequence, &initMenu{})
//}
//
//func (i initMenu) InitializerName() string {
//   return model.Menu{}.TableName()
//}
//
//func (i *initMenu) InitializeData(ctx context.Context) (next context.Context, err error) {
//   entities := []*model.Menu{
//      {ID: 1, ParentId: 0, Path: "", Name: "", Title: "客户管理"},
//      {ID: 2, ParentId: 1, Path: "/client", Name: "", Title: "客户管理"},
//
//      {ID: 3, ParentId: 0, Path: "", Name: "", Title: "销售管理"},
//      {ID: 4, ParentId: 3, Path: "/saleChance", Name: "", Title: "销售机会"},
//
//      {ID: 5, ParentId: 0, Path: "", Name: "", Title: "服务管理"},
//      {ID: 6, ParentId: 5, Path: "/serviceContract", Name: "", Title: "服务合同"},
//
//      {ID: 7, ParentId: 0, Path: "", Name: "", Title: "后台设置"},
//      {ID: 8, ParentId: 7, Path: "/member", Name: "", Title: "成员管理"},
//      {ID: 9, ParentId: 7, Path: "/role", Name: "", Title: "角色管理"},
//   }
//   if err = model.NewMenuSearch(nil).CreateBatch(entities); err != nil {
//      return ctx, errors.Wrap(err, i.InitializerName()+"表数据初始化失败!")
//   }
//   next = context.WithValue(ctx, i.InitializerName(), entities)
//   logx.Infof("InitializeData success initName:%v", i.InitializerName())
//   return next, nil
//}
//
//func (i *initMenu) DataInserted(ctx context.Context) bool {
//   _, err := model.NewMenuSearch(nil).SetId(22).First()
//   if errors.Is(err, gorm.ErrRecordNotFound) { // 判断是否存在数据
//      return false
//   }
//   return true
//}