Пример #1
0
func NewTask(ui common.UI) (workers.WorkerTask, error) {
	c := ui.ConfigGet()
	conf := c.GetSwiftConfig()
	if conf == nil {
		panic("Invalid Configuration object for swift worker")
	}

	conn := swift.Connection{
		UserName: conf.Username,
		ApiKey:   conf.ApiKey,
		AuthUrl:  conf.AuthUrl,
		// https://auth.api.rackspacecloud.com/v1.0
		Region:    conf.Region,
		UserAgent: fmt.Sprintf("hurl/1 http load tester; https://github.com/pquerna/hurl;  username=%s", conf.Username),
	}

	if authTokenCache != "" && storageUrlCache != "" {
		conn.AuthToken = authTokenCache
		conn.StorageUrl = storageUrlCache
	} else {
		err := conn.Authenticate()
		if err != nil {
			return nil, err
		}
		authTokenCache = conn.AuthToken
		storageUrlCache = conn.StorageUrl
	}

	return &Task{
		conf: conf,
		conn: &conn}, nil
}
Пример #2
0
func NewTask(ui common.UI) (workers.WorkerTask, error) {
	c := ui.ConfigGet()
	conf := c.GetHttpConfig()
	if conf == nil {
		panic("Invalid Configuration object for http/1 worker")
	}

	// TODO(pquerna): TLS transport configuration
	trans := &nhttp.Transport{
		/* TODO: conf.Compression */
		DisableCompression:  true,
		DisableKeepAlives:   !conf.Keepalive,
		MaxIdleConnsPerHost: conf.Concurrency,
	}
	//	client := &nhttp.Client{
	//		Transport: trans,
	//		CheckRedirect: func(req *nhttp.Request, via []*nhttp.Request) error {
	//			return errors.New("do not follow redirects")
	//		},
	//	}

	url, err := url.Parse(conf.Url)
	if err != nil {
		panic("Broken URL")
	}

	return &Task{conf: conf, transport: trans, URL: url}, nil
}
Пример #3
0
func NewTask(ui common.UI) (workers.WorkerTask, error) {
	c := ui.ConfigGet()
	conf := c.GetEtcdConfig()
	if conf == nil {
		panic("Invalid Configuration object for etcd worker")
	}

	client := etcd.NewClient([]string{conf.Url})

	return &Task{conf: conf, client: client}, nil
}
Пример #4
0
func ConsoleRun(ui common.UI, cmd *cobra.Command, args []string) {
	if len(args) != 1 {
		common.ConsoleErr(cmd, fmt.Sprintf("Error: Expected 1 URL, got: %s", args))
		return
	}

	g_config.Url = args[0]

	ui.ConfigSet(&g_config)

	common.ConsoleRun(workers.Run, "http", ui, cmd, args)
}
Пример #5
0
func resultHanlder(ui common.UI, wgres *sync.WaitGroup, resChan chan *common.Result, raw *common.ResultArchiveWriter) {
	defer func() { wgres.Done() }()
	var i int64 = 0
	for {
		rv, ok := <-resChan
		if !ok {
			return
		}
		i++
		ui.WorkStatus(i)
		raw.Write(rv)
	}
}
Пример #6
0
func ConsoleRun(ui common.UI, cmd *cobra.Command, args []string) {
	if len(args) != 1 {
		common.ConsoleErr(cmd, fmt.Sprintf("Error: Missing mode: %s", args))
		return
	}

	// https://auth.api.rackspacecloud.com/v1.0

	g_config.Mode = args[0]

	ui.ConfigSet(&g_config)

	common.ConsoleRun(workers.Run, "swift", ui, cmd, args)
}
Пример #7
0
func Run(ui common.UI, taskType string) error {
	//	if clusterConf != "" {
	//		// TODO: add ClusterWorker
	//		return nil, fmt.Errorf("TODO: Cluster support")
	//	}
	wt, ok := g_workers_tasks[taskType]
	if !ok {
		return fmt.Errorf("unknown worker type: %s", taskType)
	}

	conf := ui.ConfigGet()
	bconf := conf.GetBasicConfig()
	workers := make([]Worker, bconf.Concurrency)
	for index, _ := range workers {
		w, err := wt(ui)
		if err != nil {
			return err
		}
		workers[index] = &LocalWorker{task: w}
	}

	ui.WorkStart(bconf.NumRequests)
	defer func() {
		for _, worker := range workers {
			worker.Halt()
		}
	}()

	var wg sync.WaitGroup
	var wgres sync.WaitGroup
	reqchan := make(chan int64, 1024*1024)
	// TODO: how big should this be?
	reschan := make(chan *common.Result)
	for _, worker := range workers {
		err := worker.Start(taskType, &wg, reqchan, reschan)
		if err != nil {
			return err
		}
	}

	var i int64

	for i = 0; i < bconf.NumRequests; i++ {
		reqchan <- i
		// TOOD: ui.WorkStatus(numDone int64)
	}
	wgres.Add(1)

	rw := common.NewResultArchiveWriter()
	defer rw.Remove()
	go resultHanlder(ui, &wgres, reschan, rw)

	close(reqchan)
	wg.Wait()
	close(reschan)
	wgres.Wait()
	ui.WorkEnd()

	rw.Close()

	rr := common.NewResultArchiveReader(rw.Path)
	return reports.Run(ui, taskType, conf, rr)
}