feat:支持初始化管理员帐户

Author: chuntaojun

admin/maintain.go

@@ -47,6 +47,9 @@ func (s *Server) HasMainUser(ctx context.Context) (*apisecurity.User, error) {
 		log.Error("check hash main user", zap.Error(err), utils.RequestID(ctx))
 		return nil, err
 	}
+	if mainUser == nil {
+		return nil, nil
+	}
 	ret := mainUser.ToSpec()
 	ret.AuthToken = wrapperspb.String("")
 	return ret, nil

apiserver/httpserver/admin_access.go

@@ -70,7 +70,7 @@ func (h *HTTPServer) GetAdminAccessServer() *restful.WebService {
 	ws.Route(docs.EnrichEnablePprofApiDocs(ws.POST("/pprof/enable").To(h.EnablePprof)))
 	ws.Route(docs.EnrichGetServerFunctionsApiDocs(ws.GET("/server/functions").To(h.GetServerFunctions)))
 	ws.Route(ws.GET("/mainuser/exist").To(h.HasMainUser))
-	ws.Route(ws.GET("/mainuser/create").To(h.InitMainUser))
+	ws.Route(ws.POST("/mainuser/create").To(h.InitMainUser))
 	return ws
 }
 

config/config_file.go

@@ -37,6 +37,11 @@ import (
 
 // CreateConfigFile 创建配置文件
 func (s *Server) CreateConfigFile(ctx context.Context, req *apiconfig.ConfigFile) *apiconfig.ConfigResponse {
+	savaData := model.ToConfigFileStore(req)
+	if errResp := s.chains.BeforeCreateFile(ctx, savaData); errResp != nil {
+		return errResp
+	}
+
 	if rsp := s.prepareCreateConfigFile(ctx, req); rsp.Code.Value != api.ExecuteSuccess {
 		return rsp
 	}
@@ -50,7 +55,7 @@ func (s *Server) CreateConfigFile(ctx context.Context, req *apiconfig.ConfigFile
 		_ = tx.Rollback()
 	}()
 
-	resp := s.handleCreateConfigFile(ctx, tx, req)
+	resp := s._handleCreateConfigFile(ctx, tx, savaData)
 	if resp.GetCode().GetValue() != uint32(apimodel.Code_ExecuteSuccess) {
 		return resp
 	}
@@ -63,30 +68,37 @@ func (s *Server) CreateConfigFile(ctx context.Context, req *apiconfig.ConfigFile
 	return resp
 }
 
+// handleCreateConfigFile .
 func (s *Server) handleCreateConfigFile(ctx context.Context, tx store.Tx,
 	req *apiconfig.ConfigFile) *apiconfig.ConfigResponse {
 
-	data, err := s.storage.GetConfigFileTx(tx, req.GetNamespace().GetValue(), req.GetGroup().GetValue(),
-		req.GetName().GetValue())
+	savaData := model.ToConfigFileStore(req)
+	if errResp := s.chains.BeforeCreateFile(ctx, savaData); errResp != nil {
+		return errResp
+	}
+
+	return s._handleCreateConfigFile(ctx, tx, savaData)
+}
+
+func (s *Server) _handleCreateConfigFile(ctx context.Context, tx store.Tx,
+	saveData *model.ConfigFile) *apiconfig.ConfigResponse {
+
+	data, err := s.storage.GetConfigFileTx(tx, saveData.Namespace, saveData.Group,
+		saveData.Name)
 	if err != nil {
 		log.Error("[Config][File] create config file when get save data.", utils.RequestID(ctx),
-			utils.ZapNamespace(req.GetNamespace().GetValue()), utils.ZapGroup(req.GetGroup().GetValue()),
-			utils.ZapFileName(req.GetName().GetValue()), zap.Error(err))
+			utils.ZapNamespace(saveData.Namespace), utils.ZapGroup(saveData.Group),
+			utils.ZapFileName(saveData.Name), zap.Error(err))
 		return api.NewConfigResponse(commonstore.StoreCode2APICode(err))
 	}
 	if data != nil {
 		return api.NewConfigResponse(apimodel.Code_ExistedResource)
 	}
-
-	savaData := model.ToConfigFileStore(req)
-	if errResp := s.chains.BeforeCreateFile(ctx, savaData); errResp != nil {
-		return errResp
-	}
 	// 创建配置文件
-	if err := s.storage.CreateConfigFileTx(tx, savaData); err != nil {
+	if err := s.storage.CreateConfigFileTx(tx, saveData); err != nil {
 		log.Error("[Config][File] create config file error.", utils.RequestID(ctx),
-			utils.ZapNamespace(req.GetNamespace().GetValue()), utils.ZapGroup(req.GetGroup().GetValue()),
-			utils.ZapFileName(req.GetName().GetValue()), zap.Error(err))
+			utils.ZapNamespace(saveData.Namespace), utils.ZapGroup(saveData.Group),
+			utils.ZapFileName(saveData.Name), zap.Error(err))
 		return api.NewConfigResponse(commonstore.StoreCode2APICode(err))
 	}
 	return api.NewConfigResponse(apimodel.Code_ExecuteSuccess)

store/boltdb/load.go

@@ -44,6 +44,10 @@ func (m *boltStore) loadByFile(loadFile string) error {
 	}
 	cf, err := os.Open(loadFile)
 	if err != nil {
+		// 如果文件不存在，则需要用户首次调用接口进行管理员帐户的初始化操作
+		if errors.Is(err, os.ErrNotExist) {
+			return nil
+		}
 		return err
 	}
 	defer cf.Close()