k6 新的扩展参考开发

内容来自官方文档，主要是一个学习

创建项目

go mod init github.com/k6io/xk6-redis

参考代码

需要push github

package redis

​

import (

    "context"

    "time"

​

    "github.com/go-redis/redis/v8"

​

    "github.com/loadimpact/k6/js/common"

    "github.com/loadimpact/k6/js/modules"

)

​

// Register the extension on module initialization, available to

// import from JS as "k6/x/redis".

func init() {

    modules.Register("k6/x/redis", new(Redis))

}

​

// Redis is the k6 extension for a Redis client.

type Redis struct{}

​

// Client is the Redis client wrapper.

type Client struct {

    client *redis.Client

}

​

// XClient represents the Client constructor (i.e. `new redis.Client()`) and

// returns a new Redis client object.

func (r *Redis) Client(ctxPtr *context.Context, opts *redis.Options) interface{} {

    rt := common.GetRuntime(*ctxPtr)

    return common.Bind(rt, &Client{client: redis.NewClient(opts)}, ctxPtr)

}

​

// Set the given key with the given value and expiration time.

func (c *Client) Set(key, value string, exp time.Duration) {

    c.client.Set(c.client.Context(), key, value, exp)

}

​

// Get returns the value for the given key.

func (c *Client) Get(key string) (string, error) {

    res, err := c.client.Get(c.client.Context(), key).Result()

    if err != nil {

        return "", err

    }

    return res, nil

}

构建使用

• 安装xk6

go get -u github.com/k6io/xk6/cmd/xk6

• 构建

xk6 build v0.29.0 --with github.com/k6io/xk6-redis

说明：我们构建的时候也可以指定本地代码位置（类似nginx 模块的构建）

xk6 build v0.29.0 \

  --with github.com/k6io/xk6-redis="/absolute/path/to/xk6-redis"

• 使用
  test.js

 

import redis from 'k6/x/redis';

​

const client = new redis.Client({

  addr: 'localhost:6379',

  password: '',

  db: 0,

});

​

export default function () {

  client.set('mykey', 'myvalue', 0);

  console.log(`mykey => ${client.get('mykey')}`);

}

参考资料

https://k6.io/blog/extending-k6-with-xk6#existing-k6-extensions