JCS-pub/common/pkgs/mq/agent/client.go

package agent

import (
	"gitlink.org.cn/cloudream/common/pkgs/mq"
	stgmq "gitlink.org.cn/cloudream/storage-common/pkgs/mq"
)

type Client struct {
	rabbitCli *mq.RabbitMQClient
	id        int64
}

func NewClient(id int64, cfg *stgmq.Config) (*Client, error) {
	rabbitCli, err := mq.NewRabbitMQClient(cfg.MakeConnectingURL(), stgmq.MakeAgentQueueName(id), "")
	if err != nil {
		return nil, err
	}

	return &Client{
		rabbitCli: rabbitCli,
		id:        id,
	}, nil
}

func (c *Client) Close() {
	c.rabbitCli.Close()
}

type PoolClient struct {
	*Client
	owner *Pool
}

func (c *PoolClient) Close() {
	c.owner.Release(c)
}

type Pool struct {
	mqcfg *stgmq.Config
}

func NewPool(mqcfg *stgmq.Config) *Pool {
	return &Pool{
		mqcfg: mqcfg,
	}
}
func (p *Pool) Acquire(id int64) (*PoolClient, error) {
	cli, err := NewClient(id, p.mqcfg)
	if err != nil {
		return nil, err
	}

	return &PoolClient{
		Client: cli,
		owner:  p,
	}, nil
}

func (p *Pool) Release(cli *PoolClient) {
	cli.Client.Close()
}