Method handleInitRequest

provisionersdk/session.go:210–274

(init *proto.InitRequest, requests <-chan *proto.Request)

}

func (s *Session) handleInitRequest(init *proto.InitRequest, requests <-chan *proto.Request) (*proto.InitComplete, error) {
	req := &InitRequest{
		InitRequest:   init,
		ModuleArchive: nil,
	}
	if len(init.GetInitialModuleTarHash()) > 0 {
		file, err := HandleReceivingDataUpload(&fromChannel{requests: requests})
		if err != nil {
			return nil, err
		}

		data, err := file.Complete()
		if err != nil {
			return nil, err
		}
		req.ModuleArchive = data
	}

	r := &request[*InitRequest, *proto.InitComplete]{
		req:      req,
		session:  s,
		serverFn: s.server.Init,
		cancels:  requests,
	}
	complete, err := r.do()
	if err != nil {
		return nil, err
	}
	if complete.Error != "" {
		return complete, nil
	}

	// If the size of the complete message is too large, we need to stream the module files separately.
	if protobuf.Size(&proto.Response{Type: &proto.Response_Init{Init: complete}}) > drpcsdk.MaxMessageSize {
		// It is likely the modules that is pushing the message size over the limit.
		// Send the modules over a stream of messages instead.
		s.Logger.Info(s.Context(), "plan response too large, sending modules as stream",
			slog.F("size_bytes", len(complete.ModuleFiles)),
		)
		dataUp, chunks, err := proto.BytesToDataUpload(proto.DataUploadType_UPLOAD_TYPE_MODULE_FILES, complete.ModuleFiles)
		if err != nil {
			complete.Error = fmt.Sprintf("prepare module files upload: %s", err.Error())
		} else {
			complete.ModuleFiles = nil // sent over the stream
			complete.ModuleFilesHash = dataUp.DataHash

			err := s.stream.Send(&proto.Response{Type: &proto.Response_DataUpload{DataUpload: dataUp}})
			if err != nil {
				complete.Error = fmt.Sprintf("send data upload: %s", err.Error())
			} else {
				for i, chunk := range chunks {
					err := s.stream.Send(&proto.Response{Type: &proto.Response_ChunkPiece{ChunkPiece: chunk}})
					if err != nil {
						complete.Error = fmt.Sprintf("send data piece upload %d/%d: %s", i, dataUp.Chunks, err.Error())
						break
					}
				}
			}