Skip to content

Commit

Permalink
Compile corejs sources once
Browse files Browse the repository at this point in the history
  • Loading branch information
@mstoykov
mstoykov committed Nov 20, 2020
1 parent ff2e0c8 commit ddf9ecd
Showing 1 changed file with 169 additions and 57 deletions.
226 changes: 169 additions & 57 deletions js/lib/lib.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -24,97 +24,209 @@ package lib

import (
"path"
"sync"
"time"

rice "github.com/GeertJohan/go.rice"
"github.com/dop251/goja"
"github.com/loadimpact/k6/js/compiler"
"github.com/loadimpact/k6/lib"
)

//nolint:gochecknoglobals
var (
once sync.Once
coreJs *goja.Program
compiled map[string]*goja.Program
)

func GetCoreJS() *goja.Program {
once.Do(func() {
coreJs = goja.MustCompile(
"core-js/shim.min.js",
rice.MustFindBox("core-js").MustString("shim.min.js"),
true,
)
})

return coreJs
}
func init() {
var list = []string{
"_add-to-unscopables.js",
"_a-function.js",
"_an-instance.js",
"_an-object.js",
"_array-from-iterable.js",
"_array-includes.js",
"_array-methods.js",
"_array-species-constructor.js",
"_array-species-create.js",
"_classof.js",
"_cof.js",
"_collection.js",
"_collection-strong.js",
"_collection-to-json.js",
"_collection-weak.js",
"core.get-iterator-method.js",
"_core.js",
"_create-property.js",
"_ctx.js",
"_defined.js",
"_descriptors.js",
"_dom-create.js",
"_enum-bug-keys.js",
"es6.map.js",
"es6.promise.js",
"es6.set.js",
"es6.weak-map.js",
"es7.array.flat-map.js",
"es7.array.flatten.js",
"es7.asap.js",
"es7.error.is-error.js",
"es7.global.js",
"es7.map.from.js",
"es7.map.of.js",
"es7.map.to-json.js",
"es7.object.define-getter.js",
"es7.object.define-setter.js",
"es7.object.entries.js",
"es7.object.get-own-property-descriptors.js",
"es7.object.lookup-getter.js",
"es7.object.lookup-setter.js",
"es7.object.values.js",
"es7.observable.js",
"es7.promise.finally.js",
"es7.promise.try.js",
"es7.reflect.define-metadata.js",
"es7.reflect.delete-metadata.js",
"es7.reflect.get-metadata.js",
"es7.reflect.get-metadata-keys.js",
"es7.reflect.get-own-metadata.js",
"es7.reflect.get-own-metadata-keys.js",
"es7.reflect.has-metadata.js",
"es7.reflect.has-own-metadata.js",
"es7.reflect.metadata.js",
"es7.set.from.js",
"es7.set.of.js",
"es7.set.to-json.js",
"es7.string.at.js",
"es7.string.match-all.js",
"es7.string.pad-end.js",
"es7.string.pad-start.js",
"es7.string.trim-left.js",
"es7.string.trim-right.js",
"es7.symbol.async-iterator.js",
"es7.symbol.observable.js",
"es7.system.global.js",
"es7.weak-map.from.js",
"es7.weak-map.of.js",
"es7.weak-set.from.js",
"es7.weak-set.of.js",
"_export.js",
"_fails.js",
"_flags.js",
"_flatten-into-array.js",
"_for-of.js",
"_global.js",
"_has.js",
"_hide.js",
"_html.js",
"_ie8-dom-define.js",
"_inherit-if-required.js",
"_invoke.js",
"_iobject.js",
"_is-array-iter.js",
"_is-array.js",
"_is-object.js",
"_is-regexp.js",
"_iterators.js",
"_iter-call.js",
"_iter-create.js",
"_iter-define.js",
"_iter-detect.js",
"_iter-step.js",
"_library.js",
"_metadata.js",
"_meta.js",
"_microtask.js",
"_new-promise-capability.js",
"_object-assign.js",
"_object-create.js",
"_object-dp.js",
"_object-dps.js",
"_object-forced-pam.js",
"_object-gopd.js",
"_object-gopn.js",
"_object-gops.js",
"_object-gpo.js",
"_object-keys-internal.js",
"_object-keys.js",
"_object-pie.js",
"_object-to-array.js",
"_own-keys.js",
"_perform.js",
"_promise-resolve.js",
"_property-desc.js",
"_redefine-all.js",
"_redefine.js",
"_set-collection-from.js",
"_set-collection-of.js",
"_set-proto.js",
"_set-species.js",
"_set-to-string-tag.js",
"_shared.js",
"_shared-key.js",
"_species-constructor.js",
"_string-at.js",
"_string-pad.js",
"_string-repeat.js",
"_string-trim.js",
"_string-ws.js",
"_task.js",
"_to-absolute-index.js",
"_to-integer.js",
"_to-iobject.js",
"_to-length.js",
"_to-object.js",
"_to-primitive.js",
"_uid.js",
"_validate-collection.js",
"_wks-define.js",
"_wks-ext.js",
"_wks.js",
}
compiled = make(map[string]*goja.Program, len(list))
for _, name := range list {
compiled[name] = goja.MustCompile(
name,

type programWithSource struct {
pgm *goja.Program
src string
module *goja.Object
"(function(module, exports){\n"+
rice.MustFindBox("core-js").MustString(name)+"\n})",
true)
}
}

// AddPolyfills adds the polyfils to the provided runtime
func AddPolyfills(rt *goja.Runtime) error {
// TODO refactor this ... maybe merge with the init context code for this ...
var mainPwd string
comp := compiler.New(nil)
var pwd string // TODO this might not be needed as they are in the same folder

// Cache of loaded programs and files.
programs := make(map[string]programWithSource)
modules := make(map[string]*goja.Object)
rt.Set("require", func(str string) (goja.Value, error) {
// fmt.Println(str)
pwd := mainPwd
fileURL := path.Join(pwd, str) + ".js"
filename := path.Join(pwd, str) + ".js"

// First, check if we have a cached program already.
pgm, ok := programs[fileURL]
if !ok || pgm.module == nil {
mainPwd = path.Dir(fileURL)
defer func() { mainPwd = pwd }()
module, ok := modules[filename]
if !ok {
// TODO this is technically not needed as they are all in the same folder currently
defer func(backPwd string) { pwd = backPwd }(pwd)
pwd = path.Dir(filename)
exports := rt.NewObject()
pgm.module = rt.NewObject()
_ = pgm.module.Set("exports", exports)
var err error

if pgm.pgm == nil {
// Load the sources; the loader takes care of remote loading, etc.
// TODO: don't use the Global logger
data, err := rice.MustFindBox("core-js").String(fileURL)
if err != nil {
fileURL := fileURL
go func() { panic(fileURL) }()
time.Sleep(time.Second)
}

pgm.src = string(data)

pgm.pgm, _, err = comp.Compile(pgm.src, fileURL,
"(function(module, exports){\n", "\n})\n", true, lib.CompatibilityModeBase)
if err != nil {
return goja.Undefined(), err
}
}
module = rt.NewObject()
_ = module.Set("exports", exports)

programs[fileURL] = pgm
modules[filename] = module

// Run the program.
f, err := rt.RunProgram(pgm.pgm)
f, err := rt.RunProgram(compiled[filename])
if err != nil {
delete(programs, fileURL)
delete(modules, filename)
return goja.Undefined(), err
}
if call, ok := goja.AssertFunction(f); ok {
if _, err = call(exports, pgm.module, exports); err != nil {
if _, err = call(exports, module, exports); err != nil {
return nil, err
}
}
}

return pgm.module.Get("exports"), nil
return module.Get("exports"), nil
})

defer func() {
Expand Down

0 comments on commit ddf9ecd

Please sign in to comment.