【golang】发送自定义数量的UDP报文

最新推荐文章于 2024-07-26 04:00:00 发布

作业还没写完

最新推荐文章于 2024-07-26 04:00:00 发布

阅读量617

点赞数 1

分类专栏： go

本文链接：https://blog.csdn.net/qq_38858546/article/details/117424959

版权

go 专栏收录该内容

5 篇文章 1 订阅

订阅专栏

package main

import (
	"net"
	"fmt"
	"flag"
	"context"
	"time"
)

const (
	IP = "192.168.0.245"
	PORT = "514"
)

var (
	number int
	count int
	timer bool
	mesg = "<10>Apr 29 17:59:08 localhost t_syslog[29121]: This is a syslog test message generated by program './t_syslog'"
)

func SendMesg(number, cnt int, timer bool) (err error) {
	Addr := IP + ":" + PORT
	conn, err := net.Dial("udp", Addr)
	if err != nil {
		fmt.Println("Dial() err:", err)
		return err
	}
	defer conn.Close()

	ch := make(chan bool)
	ctx, cancel := context.WithCancel(context.Background())

	var sendNum int = 0
	go func() {
		select {
		case <-time.After(1 * time.Second):
			fmt.Printf("In a second, %d messages send\n", sendNum)
		case <-ctx.Done():
			fmt.Printf("Less than a second, %d messages send!\n", sendNum)
		}
		ch <- true
	}()

	sendData := []byte(mesg)
	for i := 0; i < number; i++ {
		_, err = conn.Write(sendData)
		if err != nil {
			fmt.Println("Write() err:", err)
			break
		}
		sendNum++
	}
	if !timer {
		cancel()
	}
	<-ch 
	return err
}

func main() {
	flag.IntVar(&number, "n", 100, "number of message to send")
	flag.IntVar(&count, "c", 1, "frequency")
	flag.BoolVar(&timer, "t", false, "send in a second")
	flag.Parse()

	t1 := time.Now()
	for i := 0; i < count; i++ {
		err := SendMesg(number, i, timer)
		if err != nil {
			break
		}
	}
	fmt.Println("Total sending time:", time.Since(t1))
}