How to use init method of run Package

Best Venom code snippet using run.init

run.go

Source:run.go

...44	"max-rate", 0, "Maximum frequency of operations (reads/writes). If 0, no limit.")45var maxOps = runFlags.Uint64("max-ops", 0, "Maximum number of operations to run")46var duration = runFlags.Duration("duration", 0,47	"The duration to run (in addition to --ramp). If 0, run forever.")48var doInit = runFlags.Bool("init", false, "Automatically run init. DEPRECATED: Use workload init instead.")49var ramp = runFlags.Duration("ramp", 0*time.Second, "The duration over which to ramp up load.")50var initFlags = pflag.NewFlagSet(`init`, pflag.ContinueOnError)51var drop = initFlags.Bool("drop", false, "Drop the existing database, if it exists")52var sharedFlags = pflag.NewFlagSet(`shared`, pflag.ContinueOnError)53var pprofport = sharedFlags.Int("pprofport", 33333, "Port for pprof endpoint.")54var dataLoader = sharedFlags.String("data-loader", `INSERT`,55	"How to load initial table data. Options are INSERT and IMPORT")56var initConns = sharedFlags.Int("init-conns", 16,57	"The number of connections to use during INSERT init")58var displayEvery = runFlags.Duration("display-every", time.Second, "How much time between every one-line activity reports.")59var displayFormat = runFlags.String("display-format", "simple", "Output display format (simple, incremental-json)")60var prometheusPort = sharedFlags.Int(61	"prometheus-port",62	2112,63	"Port to expose prometheus metrics if the workload has a prometheus gatherer set.",64)65var histograms = runFlags.String(66	"histograms", "",67	"File to write per-op incremental and cumulative histogram data.")68var histogramsMaxLatency = runFlags.Duration(69	"histograms-max-latency", 100*time.Second,70	"Expected maximum latency of running a query")71var securityFlags = pflag.NewFlagSet(`security`, pflag.ContinueOnError)72var secure = securityFlags.Bool("secure", false,73	"Run in secure mode (sslmode=require). "+74		"Running in secure mode expects the relevant certs to have been created for the user in the certs/ directory."+75		"For example when using root, certs/client.root.crt certs/client.root.key should exist.")76var user = securityFlags.String("user", "root", "Specify a user to run the workload as")77var password = securityFlags.String("password", "", "Optionally specify a password for the user")78func init() {79	_ = sharedFlags.MarkHidden("pprofport")80	AddSubCmd(func(userFacing bool) *cobra.Command {81		var initCmd = SetCmdDefaults(&cobra.Command{82			Use:   `init`,83			Short: `set up tables for a workload`,84		})85		for _, meta := range workload.Registered() {86			gen := meta.New()87			var genFlags *pflag.FlagSet88			if f, ok := gen.(workload.Flagser); ok {89				genFlags = f.Flags().FlagSet90			}91			genInitCmd := SetCmdDefaults(&cobra.Command{92				Use:   meta.Name + " [pgurl...]",93				Short: meta.Description,94				Long:  meta.Description + meta.Details,95				Args:  cobra.ArbitraryArgs,96			})97			genInitCmd.Flags().AddFlagSet(initFlags)98			genInitCmd.Flags().AddFlagSet(sharedFlags)99			genInitCmd.Flags().AddFlagSet(genFlags)100			genInitCmd.Flags().AddFlagSet(securityFlags)101			genInitCmd.Run = CmdHelper(gen, runInit)102			if userFacing && !meta.PublicFacing {103				genInitCmd.Hidden = true104			}105			initCmd.AddCommand(genInitCmd)106		}107		return initCmd108	})109	AddSubCmd(func(userFacing bool) *cobra.Command {110		var runCmd = SetCmdDefaults(&cobra.Command{111			Use:   `run`,112			Short: `run a workload's operations against a cluster`,113		})114		for _, meta := range workload.Registered() {115			gen := meta.New()116			if _, ok := gen.(workload.Opser); !ok {117				// If Opser is not implemented, this would just fail at runtime,118				// so omit it.119				continue120			}121			var genFlags *pflag.FlagSet122			if f, ok := gen.(workload.Flagser); ok {123				genFlags = f.Flags().FlagSet124			}125			genRunCmd := SetCmdDefaults(&cobra.Command{126				Use:   meta.Name + " [pgurl...]",127				Short: meta.Description,128				Long:  meta.Description + meta.Details,129				Args:  cobra.ArbitraryArgs,130			})131			genRunCmd.Flags().AddFlagSet(runFlags)132			genRunCmd.Flags().AddFlagSet(sharedFlags)133			genRunCmd.Flags().AddFlagSet(genFlags)134			genRunCmd.Flags().AddFlagSet(securityFlags)135			initFlags.VisitAll(func(initFlag *pflag.Flag) {136				// Every init flag is a valid run flag that implies the --init option.137				f := *initFlag138				f.Usage += ` (implies --init)`139				genRunCmd.Flags().AddFlag(&f)140			})141			genRunCmd.Run = CmdHelper(gen, runRun)142			if userFacing && !meta.PublicFacing {143				genRunCmd.Hidden = true144			}145			runCmd.AddCommand(genRunCmd)146		}147		return runCmd148	})149}150// CmdHelper handles common workload command logic, such as error handling and151// ensuring the database name in the connection string (if provided) matches the152// expected one.153func CmdHelper(154	gen workload.Generator, fn func(gen workload.Generator, urls []string, dbName string) error,155) func(*cobra.Command, []string) {156	return HandleErrs(func(cmd *cobra.Command, args []string) error {157		// Apply the logging configuration if none was set already.158		if active, _ := log.IsActive(); !active {159			cfg := logconfig.DefaultStderrConfig()160			if err := cfg.Validate(nil /* no default log directory */); err != nil {161				return err162			}163			if _, err := log.ApplyConfig(cfg); err != nil {164				return err165			}166		}167		if h, ok := gen.(workload.Hookser); ok {168			if h.Hooks().Validate != nil {169				if err := h.Hooks().Validate(); err != nil {170					return errors.Wrapf(err, "could not validate")171				}172			}173		}174		// HACK: Steal the dbOverride out of flags. This should go away175		// once more of run.go moves inside workload.176		var dbOverride string177		if dbFlag := cmd.Flag(`db`); dbFlag != nil {178			dbOverride = dbFlag.Value.String()179		}180		urls := args181		if len(urls) == 0 {182			crdbDefaultURL := fmt.Sprintf(`postgres://%s@localhost:26257?sslmode=disable`, *user)183			if *secure {184				if *password != "" {185					crdbDefaultURL = fmt.Sprintf(186						`postgres://%s:%s@localhost:26257?sslmode=require&sslrootcert=certs/ca.crt`,187						*user, *password)188				} else {189					crdbDefaultURL = fmt.Sprintf(190						// This URL expects the certs to have been created by the user.191						`postgres://%s@localhost:26257?sslcert=certs/client.%s.crt&sslkey=certs/client.%s.key&sslrootcert=certs/ca.crt&sslmode=require`,192						*user, *user, *user)193				}194			}195			urls = []string{crdbDefaultURL}196		}197		dbName, err := workload.SanitizeUrls(gen, dbOverride, urls)198		if err != nil {199			return err200		}201		return fn(gen, urls, dbName)202	})203}204// SetCmdDefaults ensures that the provided Cobra command will properly report205// an error if the user specifies an invalid subcommand. It is safe to call on206// any Cobra command.207//208// This is a wontfix bug in Cobra: https://github.com/spf13/cobra/pull/329209func SetCmdDefaults(cmd *cobra.Command) *cobra.Command {210	if cmd.Run == nil && cmd.RunE == nil {211		cmd.Run = func(cmd *cobra.Command, args []string) {212			_ = cmd.Usage()213		}214	}215	if cmd.Args == nil {216		cmd.Args = cobra.NoArgs217	}218	return cmd219}220// numOps keeps a global count of successful operations.221var numOps uint64222// workerRun is an infinite loop in which the worker continuously attempts to223// read / write blocks of random data into a table in cockroach DB. The function224// returns only when the provided context is canceled.225func workerRun(226	ctx context.Context,227	errCh chan<- error,228	wg *sync.WaitGroup,229	limiter *rate.Limiter,230	workFn func(context.Context) error,231) {232	if wg != nil {233		defer wg.Done()234	}235	for {236		if ctx.Err() != nil {237			return238		}239		// Limit how quickly the load generator sends requests based on --max-rate.240		if limiter != nil {241			if err := limiter.Wait(ctx); err != nil {242				return243			}244		}245		if err := workFn(ctx); err != nil {246			if ctx.Err() != nil && (errors.Is(err, ctx.Err()) || errors.Is(err, driver.ErrBadConn)) {247				// lib/pq may return either the `context canceled` error or a248				// `bad connection` error when performing an operation with a context249				// that has been canceled. See https://github.com/lib/pq/pull/1000250				return251			}252			errCh <- err253			continue254		}255		v := atomic.AddUint64(&numOps, 1)256		if *maxOps > 0 && v >= *maxOps {257			return258		}259	}260}261func runInit(gen workload.Generator, urls []string, dbName string) error {262	ctx := context.Background()263	initDB, err := gosql.Open(`cockroach`, strings.Join(urls, ` `))264	if err != nil {265		return err266	}267	startPProfEndPoint(ctx)268	return runInitImpl(ctx, gen, initDB, dbName)269}270func runInitImpl(271	ctx context.Context, gen workload.Generator, initDB *gosql.DB, dbName string,272) error {273	if *drop {274		if _, err := initDB.ExecContext(ctx, `DROP DATABASE IF EXISTS `+dbName); err != nil {275			return err276		}277	}278	if _, err := initDB.ExecContext(ctx, `CREATE DATABASE IF NOT EXISTS `+dbName); err != nil {279		return err280	}281	var l workload.InitialDataLoader282	switch strings.ToLower(*dataLoader) {283	case `insert`, `inserts`:284		l = workloadsql.InsertsDataLoader{285			Concurrency: *initConns,286		}287	case `import`, `imports`:288		l = workload.ImportDataLoader289	default:290		return errors.Errorf(`unknown data loader: %s`, *dataLoader)291	}292	_, err := workloadsql.Setup(ctx, initDB, gen, l)293	return err294}295func startPProfEndPoint(ctx context.Context) {296	b := envutil.EnvOrDefaultInt64("COCKROACH_BLOCK_PROFILE_RATE",297		10000000 /* 1 sample per 10 milliseconds spent blocking */)298	m := envutil.EnvOrDefaultInt("COCKROACH_MUTEX_PROFILE_RATE",299		1000 /* 1 sample per 1000 mutex contention events */)300	runtime.SetBlockProfileRate(int(b))301	runtime.SetMutexProfileFraction(m)302	go func() {303		err := http.ListenAndServe(":"+strconv.Itoa(*pprofport), nil)304		if err != nil {305			log.Errorf(ctx, "%v", err)306		}307	}()308}309func runRun(gen workload.Generator, urls []string, dbName string) error {310	ctx := context.Background()311	var formatter outputFormat312	switch *displayFormat {313	case "simple":314		formatter = &textFormatter{}315	case "incremental-json":316		formatter = &jsonFormatter{w: os.Stdout}317	default:318		return errors.Errorf("unknown display format: %s", *displayFormat)319	}320	startPProfEndPoint(ctx)321	initDB, err := gosql.Open(`cockroach`, strings.Join(urls, ` `))322	if err != nil {323		return err324	}325	if *doInit || *drop {326		log.Info(ctx, `DEPRECATION: `+327			`the --init flag on "workload run" will no longer be supported after 19.2`)328		for {329			err = runInitImpl(ctx, gen, initDB, dbName)330			if err == nil {331				break332			}333			if !*tolerateErrors {334				return err335			}336			log.Infof(ctx, "retrying after error during init: %v", err)337		}338	}339	var limiter *rate.Limiter340	if *maxRate > 0 {341		// Create a limiter using maxRate specified on the command line and342		// with allowed burst of 1 at the maximum allowed rate.343		limiter = rate.NewLimiter(rate.Limit(*maxRate), 1)344	}345	o, ok := gen.(workload.Opser)346	if !ok {347		return errors.Errorf(`no operations defined for %s`, gen.Meta().Name)348	}349	reg := histogram.NewRegistry(350		*histogramsMaxLatency,351		gen.Meta().Name,352	)353	// Expose the prometheus gatherer.354	go func() {355		if err := http.ListenAndServe(356			fmt.Sprintf(":%d", *prometheusPort),357			promhttp.HandlerFor(reg.Gatherer(), promhttp.HandlerOpts{}),358		); err != nil {359			log.Errorf(context.Background(), "error serving prometheus: %v", err)360		}361	}()362	var ops workload.QueryLoad363	prepareStart := timeutil.Now()364	log.Infof(ctx, "creating load generator...")365	const prepareTimeout = 60 * time.Minute366	prepareCtx, cancel := context.WithTimeout(ctx, prepareTimeout)367	defer cancel()368	if prepareErr := func(ctx context.Context) error {369		retry := retry.StartWithCtx(ctx, retry.Options{})370		var err error371		for retry.Next() {372			if err != nil {373				log.Warningf(ctx, "retrying after error while creating load: %v", err)374			}375			ops, err = o.Ops(ctx, urls, reg)376			if err == nil {377				return nil378			}379			err = errors.Wrapf(err, "failed to initialize the load generator")380			if !*tolerateErrors {381				return err382			}383		}384		if ctx.Err() != nil {385			// Don't retry endlessly. Note that this retry loop is not under the386			// control of --duration, so we're avoiding retrying endlessly.387			log.Errorf(ctx, "Attempt to create load generator failed. "+388				"It's been more than %s since we started trying to create the load generator "+389				"so we're giving up. Last failure: %s", prepareTimeout, err)390		}391		return err392	}(prepareCtx); prepareErr != nil {393		return prepareErr...

run_as_non_root_test.go

Source:run_as_non_root_test.go

...75			name:     "pod run as non root false, container security context unset",76			objs:     containerNonRoot(&falseVar, nil),77			expected: diagnostic(),78		},79		// init container tests80		{81			name:     "pod security context and init container security context unset",82			objs:     initContainerSecurityContextNil(),83			expected: diagnostic(),84		},85		{86			name:     "pod security context unset, init container with run as non root set to true",87			objs:     initContainerNonRoot(nil, &trueVar),88			expected: nil,89		},90		{91			name:     "pod security context unset, init container with run as non root set to false",92			objs:     initContainerNonRoot(nil, &falseVar),93			expected: diagnostic(),94		},95		{96			name:     "pod run as non root true, init container run as non root true",97			objs:     initContainerNonRoot(&trueVar, &trueVar),98			expected: nil,99		},100		{101			name:     "pod run as non root true, init container run as non root false",102			objs:     initContainerNonRoot(&trueVar, &falseVar),103			expected: nil,104		},105		{106			name:     "pod run as non root false, init container run as non root true",107			objs:     initContainerNonRoot(&falseVar, &trueVar),108			expected: nil,109		},110		{111			name:     "pod run as non root false, init container run as non root false",112			objs:     initContainerNonRoot(&falseVar, &falseVar),113			expected: diagnostic(),114		},115		{116			name:     "pod run as non root true, init container security context unset",117			objs:     initContainerNonRoot(&trueVar, nil),118			expected: nil,119		},120		{121			name:     "pod run as non root false, init container security context unset",122			objs:     initContainerNonRoot(&falseVar, nil),123			expected: diagnostic(),124		},125	}126	for _, test := range tests {127		t.Run(test.name, func(t *testing.T) {128			d, _, err := nonRootUserCheck.Run(test.objs)129			assert.NoError(t, err)130			assert.ElementsMatch(t, test.expected, d)131		})132	}133}134func diagnostic() []check.Diagnostic {135	pod := initPod().Pods.Items[0]136	d := []check.Diagnostic{137		{138			Severity: check.Warning,139			Message:  "Container `bar` can run as root user. Please ensure that the image is from a trusted source.",140			Kind:     check.Pod,141			Object:   &pod.ObjectMeta,142			Owners:   pod.ObjectMeta.GetOwnerReferences(),143		},144	}145	return d146}...

pipe.go

Source:pipe.go

...44	opt := &RunOpt{}45	for i := range opts {46		opts[i](opt)47	}48	initCtx := opt.InitContext49	if initCtx == nil {50		initCtx = NewContext()51	}52	handlerMw := NewHandlerMiddleware(handler)53	if p.globalOrchestrator != nil {54		p.mws = p.globalOrchestrator(p.mws)55	}56	if p.mwOrchestrator != nil {57		p.mws = p.mwOrchestrator(p.mws)58	}59	for i, mw := range p.mws {60		if i < len(p.mws)-1 {61			mw.SetNext(p.mws[i+1])62			continue63		}64		mw.SetNext(handlerMw)65	}66	if len(p.mws) == 0 {67		err := handlerMw.Handle(initCtx)68		return initCtx, err69	}70	err := p.mws[0].Handle(initCtx)71	return initCtx.Output(), err72}...

init

Using AI Code Generation

1func main() {2    run.Init()3}4func main() {5    run.Init()6}7func main() {8    run.Init()9}10func main() {11    run.Init()12}13func main() {14    run.Init()15}16func main() {17    run.Init()18}19func main() {20    run.Init()21}22func main() {23    run.Init()24}25func main() {26    run.Init()27}28func main() {29    run.Init()30}31func main() {32    run.Init()33}34func main() {35    run.Init()36}37func main() {38    run.Init()39}40func main() {41    run.Init()42}43func main() {44    run.Init()45}46func main() {47    run.Init()48}49func main() {50    run.Init()51}52func main() {53    run.Init()54}55func main() {56    run.Init()57}58func main() {

init

Using AI Code Generation

1import "fmt"2func main() {3    fmt.Println("hello world")4}5import "fmt"6func init() {7    fmt.Println("init 2")8}9import "fmt"10func init() {11    fmt.Println("init 3")12}13import "fmt"14func init() {15    fmt.Println("init 4")16}

init

Using AI Code Generation

1import "fmt"2func init() { 3    fmt.Println("init method of run class") 4}5func main() {6    fmt.Println("main method of run class")7}

init

Using AI Code Generation

1func main() {2    run.Init()3    run.Run()4}5func main() {6    run.Run()7}8func main() {9    run.Run()10}11import (12var (13func init() {14}15func Init() {16    fmt.Println("init")17}18func Run() {19    fmt.Println("run")20}

init

Using AI Code Generation

1func main() {2    run.Init()3    run.Run()4}5import (6func Run() {7    fmt.Println("Running")8}9import (10func Init() {11    fmt.Println("Initializing")12}

init

Using AI Code Generation

1import (2func init() {3    fmt.Println("init method is called")4}5func main() {6    fmt.Println("main method is called")7    if runtime.GOOS == "windows" {8    } else {9    }10    err := cmd.Start()11    if err != nil {12        log.Fatal(err)13    }14}

init

Using AI Code Generation

1func main() {2    run.Run()3}4import (5func Run() {6    if runtime.GOOS == "windows" {7    } else if runtime.GOOS == "linux" {8    } else {9        fmt.Println("Unsupported platform!")10        os.Exit(1)11    }12    cmd.Start()13}

init

Using AI Code Generation

1import (2func main() {3    fmt.Println("Hello, World!")4    fmt.Println(os.Args[1])5}6import (7func main() {8    fmt.Println("Hello, World!")9    fmt.Println(os.Args[1])10}11import (12func main() {13    fmt.Println("Hello, World!")14    fmt.Println(os.Args[1])15}16import (17func main() {18    fmt.Println("Hello, World!")19    fmt.Println(os.Args[1])20}21import (22func main() {23    fmt.Println("Hello, World!")24    fmt.Println(os.Args[1])25}26import (27func main() {28    fmt.Println("Hello, World!")29    fmt.Println(os.Args[1])30}31import (32func main() {33    fmt.Println("Hello, World!")34    fmt.Println(os.Args[1])35}36import (37func main() {38    fmt.Println("Hello, World!")39    fmt.Println(os.Args[1])40}

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.