ui: delay sql module registration until after potential reset

Fixes loading SQL modules when resetting the state of trace
processor (e.g. loading trace from local storage on an already
created instance of TP)

Change-Id: I3f106f7d164e13c75edff8153c3cdb2965b17a48

ui/src/controller/trace_controller.ts

@@ -430,9 +430,6 @@ export class TraceController extends Controller<States> {
         ftraceDropUntilAllCpusValid: FTRACE_DROP_UNTIL_FLAG.get(),
       });
     }
-    for (const p of globals.extraSqlPackages) {
-      await engine.registerSqlModules(p);
-    }
     this.engine = engine;
 
     if (isMetatracingEnabled()) {
@@ -491,6 +488,9 @@ export class TraceController extends Controller<States> {
       assertTrue(this.engine instanceof HttpRpcEngine);
       await this.engine.restoreInitialTables();
     }
+    for (const p of globals.extraSqlPackages) {
+      await this.engine.registerSqlModules(p);
+    }
 
     // traceUuid will be '' if the trace is not cacheable (URL or RPC).
     const traceUuid = await this.cacheCurrentTrace();