add patch to factory (#3187)

Co-authored-by: Haaai <55118568+Liuhaai@users.noreply.github.com>

chainservice/chainservice.go

@@ -133,7 +133,12 @@ func New(
 			}
 			sf, err = factory.NewStateDB(cfg, opts...)
 		} else {
-			sf, err = factory.NewFactory(cfg, factory.DefaultTrieOption(), factory.RegistryOption(registry))
+			sf, err = factory.NewFactory(
+				cfg,
+				factory.DefaultTrieOption(),
+				factory.RegistryOption(registry),
+				factory.DefaultTriePatchOption(),
+			)
 		}
 		if err != nil {
 			return nil, errors.Wrapf(err, "Failed to create state factory")

state/factory/factory.go

@@ -157,6 +157,14 @@ func SkipBlockValidationOption() Option {
 	}
 }
 
+// DefaultTriePatchOption loads patchs
+func DefaultTriePatchOption() Option {
+	return func(sf *factory, cfg config.Config) (err error) {
+		sf.ps, err = newPatchStore(cfg.Chain.TrieDBPatchFile)
+		return
+	}
+}
+
 // NewFactory creates a new state factory
 func NewFactory(cfg config.Config, opts ...Option) (Factory, error) {
 	sf := &factory{
@@ -278,6 +286,17 @@ func (sf *factory) newWorkingSet(ctx context.Context, height uint64) (*workingSe
 	if err := store.Start(ctx); err != nil {
 		return nil, err
 	}
+	for _, p := range sf.ps.Get(height) {
+		if p.Type == _Delete {
+			if err := store.Delete(p.Namespace, p.Key); err != nil {
+				return nil, err
+			}
+		} else {
+			if err := store.Put(p.Namespace, p.Key, p.Value); err != nil {
+				return nil, err
+			}
+		}
+	}
 
 	return newWorkingSet(height, store), nil
 }