public:

explicit ReaderOpKernel(OpKernelConstruction* context);

~ReaderOpKernel() override;

void Compute(OpKernelContext* context) override;

// Must be called by descendants before the first call to Compute()

// (typically called during construction). factory must return a

// ReaderInterface descendant allocated with new that ReaderOpKernel

// will take ownership of.

void SetReaderFactory(std::function<ReaderInterface*()> factory) {

mutex_lock l(mu_);

DCHECK(!have_handle_);

factory_ = factory;

}

private:

mutex mu_;

bool have_handle_ GUARDED_BY(mu_);

PersistentTensor handle_ GUARDED_BY(mu_);

ContainerInfo cinfo_;

std::function<ReaderInterface*()> factory_;