331 class ZFixupPartialLoadsClosure : public ZRootsIteratorClosure {

   332 public:

   333   virtual void do_oop(oop* p) {

   334     ZBarrier::mark_barrier_on_root_oop_field(p);

   335   }

   336 

   337   virtual void do_oop(narrowOop* p) {

   338     ShouldNotReachHere();

   339   }

   340 };

   341 

   342 class ZFixupPartialLoadsTask : public ZTask {

   343 private:

   344   ZThreadRootsIterator _thread_roots;

   345 

   346 public:

   347   ZFixupPartialLoadsTask() :

   348       ZTask("ZFixupPartialLoadsTask"),

   349       _thread_roots() {}

   350 

   351   virtual void work() {

   352     ZFixupPartialLoadsClosure cl;

   353     _thread_roots.oops_do(&cl);

   354   }

   355 };

   356 

   357 void ZHeap::fixup_partial_loads() {

   358   ZFixupPartialLoadsTask task;

   359   _workers.run_parallel(&task);

   360 }

   361 

   364 

   365   // C2 can generate code where a safepoint poll is inserted

   366   // between a load and the associated load barrier. To handle

   367   // this case we need to rescan the thread stack here to make

   368   // sure such oops are marked.

   369   fixup_partial_loads();