77 lines
2.4 KiB
C#
77 lines
2.4 KiB
C#
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;
|
|
}
|
|
}
|
|
}
|