JiShe.CollectBus/modules/JiShe.CollectBus.Kafka/Attributes/KafkaSubscribeAttribute.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace JiShe.CollectBus.Kafka.Attributes
{
    [AttributeUsage(AttributeTargets.Method)]
    public class KafkaSubscribeAttribute : Attribute
    {
        /// <summary>
        /// 订阅的主题
        /// </summary>
        public string Topic { get; set; }

        /// <summary>
        /// 分区
        /// </summary>
        public int Partition { get; set; } = -1;

        /// <summary>
        /// 消费者组
        /// </summary>
        public string GroupId { get; set; }

        /// <summary>
        /// 任务数(默认是多少个分区多少个任务)
        /// 如设置订阅指定Partition则任务数始终为1
        /// </summary>
        public int TaskCount { get; set; } = -1;

        /// <summary>
        /// 批量处理数量
        /// </summary>
        public int BatchSize { get; set; } = 100;

        /// <summary>
        /// 是否启用批量处理
        /// </summary>
        public bool EnableBatch { get; set; } = false;

        /// <summary>
        /// 批次超时时间
        /// </summary>
        public TimeSpan? BatchTimeout { get; set; }=null;

        /// <summary>
        /// 订阅主题
        /// </summary>
        /// <param name="batchTimeout">batchTimeout格式:("00:05:00")</param>
        public KafkaSubscribeAttribute(string topic, string groupId = "default", bool enableBatch = false, int batchSize = 100, string? batchTimeout = null)
        {
            this.Topic = topic;
            this.GroupId = groupId;
            this.EnableBatch = enableBatch;
            this.BatchSize = batchSize;
            this.BatchTimeout = batchTimeout != null? TimeSpan.Parse(batchTimeout): null;
        }

        /// <summary>
        /// 订阅主题
        /// </summary>
        /// <param name="batchTimeout">batchTimeout格式:("00:05:00")</param>
        public KafkaSubscribeAttribute(string topic, int partition, string groupId = "default", bool enableBatch = false, int batchSize = 100, string? batchTimeout = null)
        {
            this.Topic = topic;
            this.Partition = partition;
            this.GroupId = groupId;
            this.TaskCount = 1;
            this.EnableBatch = enableBatch;
            this.BatchSize = batchSize;
            this.BatchTimeout = batchTimeout != null ? TimeSpan.Parse(batchTimeout) : null;
        }
    }
}