blob: 16e76562b996238a05d4ee8a8aa29dcf13cdce4a (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
// Author: arnej
// for starting standalone jdisc containers
package standalone
import (
"os"
"github.com/vespa-engine/vespa/client/go/internal/admin/jvm"
"github.com/vespa-engine/vespa/client/go/internal/admin/trace"
"github.com/vespa-engine/vespa/client/go/internal/osutil"
"github.com/vespa-engine/vespa/client/go/internal/vespa"
)
func commonPreChecks() {
if doTrace := os.Getenv("TRACE_JVM_STARTUP"); doTrace != "" {
trace.AdjustVerbosity(1)
}
if doDebug := os.Getenv("DEBUG_JVM_STARTUP"); doDebug != "" {
trace.AdjustVerbosity(2)
}
veHome := vespa.FindAndVerifyVespaHome()
err := os.Chdir(veHome)
if err != nil {
osutil.ExitErr(err)
}
err = vespa.LoadDefaultEnv()
if err != nil {
osutil.ExitErr(err)
}
}
func StartStandaloneContainer(extraArgs []string) int {
commonPreChecks()
osutil.TuneResourceLimits()
serviceName := os.Getenv("VESPA_SERVICE_NAME")
if serviceName == "" {
osutil.ExitMsg("Missing service name, ensure VESPA_SERVICE_NAME is set in the environment")
}
c := jvm.NewStandaloneContainer(serviceName)
jvmOpts := c.JvmOptions()
jvmOpts.AddOption("-DOnnxBundleActivator.skip=true")
jvmOpts.AddOption("-DLlamaBundleActivator.skip=true")
for _, extra := range extraArgs {
jvmOpts.AddOption(extra)
}
minFallback := jvm.MegaBytesOfMemory(128)
maxFallback := jvm.MegaBytesOfMemory(2048)
jvmOpts.AddDefaultHeapSizeArgs(minFallback, maxFallback)
c.Exec()
// unreachable:
return 1
}
|
