perf: use the Store struct to save states and checkpoint states in memory

perf: use the Store struct to save states and checkpoint states in memory #4123