[frontend][nvme][ctrl] UpdateNvmeController: use AllowMissing

This patch use AllowMissing parameter in UpdateNvmeController
to actually update one

Signed-off-by: minhee-han <minhee.han@mangoboost.io>

Author: han-minhee

pkg/frontend/nvme_controller.go

@@ -114,19 +114,29 @@ func (s *Server) DeleteNvmeController(ctx context.Context, in *pb.DeleteNvmeCont
 }
 
 // UpdateNvmeController updates an Nvme controller
-func (s *Server) UpdateNvmeController(_ context.Context, in *pb.UpdateNvmeControllerRequest) (*pb.NvmeController, error) {
+func (s *Server) UpdateNvmeController(ctx context.Context, in *pb.UpdateNvmeControllerRequest) (*pb.NvmeController, error) {
 	// check input correctness
 	if err := s.validateUpdateNvmeControllerRequest(in); err != nil {
 		return nil, err
 	}
 	// fetch object from the database
 	ctrlr, ok := s.Nvme.Controllers[in.NvmeController.Name]
 	if !ok {
-		if in.AllowMissing {
-			log.Printf("TODO: in case of AllowMissing, create a new resource, don;t return error")
+		if !in.AllowMissing {
+			err := status.Errorf(codes.NotFound, "unable to find key %s", in.NvmeController.Name)
+			return nil, err
+		}
+
+		// If AllowMissing is true and controller doesn't exist, create it using CreateNvmeController
+		createRequest := &pb.CreateNvmeControllerRequest{
+			NvmeController: in.NvmeController,
+			Parent:         utils.GetSubsystemIDFromNvmeName(in.NvmeController.Name),
+		}
+		var err error
+		ctrlr, err = s.CreateNvmeController(ctx, createRequest)
+		if err != nil {
+			return nil, err
 		}
-		err := status.Errorf(codes.NotFound, "unable to find key %s", in.NvmeController.Name)
-		return nil, err
 	}
 	resourceID := path.Base(ctrlr.Name)
 	// update_mask = 2