golang连接kafka消费进ES操作

1.首先初始化conf配置把kafka和ES的地址配置好还有一个日志方便查看

配置信息如下 用到的库是

github.com/astaxie/beego/config[logs]log_level = debuglog_path = './logs/log_transfer.log'[kafka]server_addr = 192.168.0.134:9092topic = nginx_log[ES]addr = http://192.168.0.134:9200/

2.读取conf配置存取进结构体

type LogConfig struct { kafkaAddr string ESAddr string LogPath string LogLevel string Topic string}var ( logConfig *LogConfig)

3.读取conf配置代码如下

func initConfig(conftype string,filename string)(err error) { conf, err := config. NewConfig(conftype,filename) if err != nil { fmt. Println( 'new config faild,err:',err) return } logConfig = &LogConfig{} logConfig.LogLevel = conf. String( 'logs::log_level') if len(logConfig.LogLevel) == 0 { logConfig.LogLevel = 'debug' } logConfig.LogPath = conf. String( 'logs::log_path') if len(logConfig.LogPath) == 0 { logConfig.LogPath = './logs' } logConfig.kafkaAddr = conf. String( 'kafka::server_addr') if len(logConfig.kafkaAddr) == 0 { err = fmt. Errorf( 'invalid kafka addr err') return } logConfig.ESAddr = conf. String( 'ES::addr') if len(logConfig.ESAddr) == 0 { err = fmt. Errorf( 'invalid ES addr err') return } logConfig.Topic = conf. String( 'kafka::topic') if len(logConfig.Topic) == 0 { err = fmt. Errorf( 'invalid topic addr err') return } return}

4.完成了initConfig的初始化

5.初始化initLogger

func convertLogLevel(level string) int { switch(level) { case 'debug': return logs.LevelDebug case 'warn': return logs.LevelWarn case 'info': return logs.LevelInfo case 'trace': return logs.LevelTrace } return logs.LevelDebug}func initLogger(logpath string, logLevel string) (err error) { config := make( map[ string] interface{}) config[ 'filename'] = logpath config[ 'level'] = convertLogLevel(logLevel) configStr, err := json. Marshal(config) if err!= nil { fmt. Println( 'marshal failed,err:',err) return } logs. SetLogger(logs.AdapterFile, string(configStr)) return}

6.初始化kafka

type KafkaClient struct { client sarama.Consumer addr string topic string wg sync.WaitGroup}var ( kafkaClient *KafkaClient)func initKafKa(addr string,topic string)(err error) { kafkaClient = &KafkaClient{} consumer, err := sarama. NewConsumer(strings. Split(addr, ','), nil) if err != nil { logs. Error( 'Failed to strat consumer :',err) return } kafkaClient.client = consumer kafkaClient.addr = addr kafkaClient.topic = topic return}

7.初始化ES

gopkg.in/olivere/elastic.v2 // 这个是操作ES的库

type LogMessage struct { App string Topic string Message string}var ( esClient *elastic.Client)func initES(addr string)(err error) { client, err := elastic. NewClient(elastic. SetSniff( false),elastic. SetURL(addr)) if err != nil { fmt. Println( 'connect es error',err) return } esClient = client return}

8.干活把kafka的数据写入ES

github.com/Shopify/sarama 这个是操作kafka的驱动库

func run()(err error) { fmt. Println( 'run') partitionList, err := kafkaClient.client. Partitions(kafkaClient.topic) if err != nil { logs. Error( 'ini failed ,err:%v',err) fmt. Printf( 'ini failed ,err:%v',err) return } for partition := range partitionList { fmt. Println( 'for进入') pc, errRet := kafkaClient.client. ConsumePartition(kafkaClient.topic, int32(partition),sarama.OffsetNewest) if errRet != nil { err = errRet logs. Error( 'Failed to start consumer for partition %d: %s n ',partition,err) fmt. Printf( 'Failed to start consumer for partition %d: %s n ',partition,err) return } defer pc. AsyncClose() fmt. Println( '马上进入协程') kafkaClient.wg. Add( 1) go func(pc sarama.PartitionConsumer){ fmt. Println( '进来了') for msg := range pc. Messages() { fmt. Println( 'func执行') logs. Debug( 'Partition:%d,Offset:%d,key:%s,value:%s',msg.Partition,msg.Offset, string(msg.Key), string(msg.Value)) //fmt.Println() err = sendToES(kafkaClient.topic,msg.Value) if err != nil { logs. Warn( 'send to es failed,err:%v',err) } } kafkaClient.wg. Done() }(pc) } kafkaClient.wg. Wait() fmt. Println( '协程执行完毕') return}

上面代码是读kafka消费数据通过sendToES这个函数发送至ES里面

sendToES代码如下

func sendToES(topic string,data [] byte) (err error) { msg := &LogMessage{} msg.Topic = topic msg.Message = string(data) _, err = esClient. Index(). Index(topic). Type(topic). BodyJson(msg). Do() if err != nil { return } return}

Index就是索引名称

index().type().bodyjson().do()这样的写法是链式执行操作

9.写完了基本操作后 再写一个模拟写入数据进kafka的数据 代码如下

func main() {config := sarama. NewConfig()config.Producer.RequiredAcks = sarama.WaitForAllconfig.Producer.Partitioner = sarama.NewRandomPartitionerconfig.Producer.Return.Successes = trueclient, err := sarama. NewSyncProducer([] string{ '127.0.0.1:9092'}, config)if err != nil {fmt. Println( 'producer close,err:', err)return}defer client. Close()var n int= 0for {n++msg := &sarama.ProducerMessage{}msg.Topic = 'nginx_log'msg.Value = sarama. StringEncoder( 'this is a good test,hello maomaoChong!!,' + strconv. Itoa(n))pid, offset, err := client. SendMessage(msg)if err != nil {fmt. Println( 'send message failed,', err)return}fmt. Printf( 'pid:%v offset:%v n ', pid, offset)time. Sleep(time.Second * 2)}}

这个就是生产者往kafka里面写入数据进去消费

10.我们启动我们的kafka 注意kafka依赖于zookeeper 先启动ZK然后启动kafka

我这里用的是zookeeper-3.4.12网上有下载

启动ZK

ZK已经成功启动

11.启动kafka 我这里是kafka_2.11-1.1.0

.binwindowskafka-server-start.bat .configserver.properties

kafka已经跑起来了

12.把kafka消费测试端也启动

.binwindowskafka-console-consumer.bat --topic nginx_log --zookeeper 127.0.0.1 2181

消费端启动成功 一直等待数据进来消费

13.然后我们把ES 和Kib 都启动了

这是我们的ES版本是5.5.1的 已经跑起来了 接着启动我们的kib

kib里面有个配置config下面的叫kibana.yml里面设置好ES的地址和端口就处于监听ES状态

启动kib有点慢 稍微等一下就好

此时启动好了kib

14.测试kib是否启动

默认地址是http://localhost:5601

进入成功 确定没问题

15.编译我们的代码 写数据进kafka

从上面看我们知道一个再写 一个再消费

16.编译运行我们把kafka写入进ES里面的代码

运行了 这里就把kafka消费的数据 写入进ES里面

17.我们看一下数据是否有 进入kib

我们看到有数据了 已经成功了~

以上为个人经验，希望能给大家一个参考，也希望大家多多支持优爱好网。如有错误或未考虑完全的地方，望不吝赐教。