jy-data-extract/fullproject/udp/src/main.go

package main

import (
	"encoding/json"
	"flag"
	"log"
	mu "mfw/util"
	"net"
	"os"
)

var ip, sid, eid, stype, infoid, pid, udpataField string
var p int

func main() {

	flag.StringVar(&sid, "sid", "", "开始id")
	flag.StringVar(&eid, "eid", "", "结束id")
	flag.StringVar(&infoid, "infoid", "", "修改的公告id")
	flag.StringVar(&pid, "pid", "", "修改的项目id")
	flag.StringVar(&udpataField, "field", "", "修改的字段信息")
	flag.StringVar(&ip, "ip", "127.0.0.1", "ip")
	flag.IntVar(&p, "p", 1482, "端口")
	flag.StringVar(&stype, "stype", "", "stype")
	flag.Parse()

	if stype != ""{
		addr := &net.UDPAddr{
			IP:   net.ParseIP(ip),
			Port: p,
		}
		udp := mu.UdpClient{Local: ":50010", BufSize: 1024}
		udp.Listen(func(b byte, data []byte, add *net.UDPAddr) {
			switch b {
			case mu.OP_NOOP:
				log.Println(string(data))
				os.Exit(0)
			}
		})
		if stype == "updatePro" {
			udpatamap := make(map[string]interface{})
			err := json.Unmarshal([]byte(udpataField), &udpatamap)
			if err == nil {
				m1 := map[string]interface{}{
					"pid":  pid,
					"updateField": udpatamap,
					"stype": stype,
				}
				by, _ := json.Marshal(m1)
				log.Println(string(by))
				udp.WriteUdp(by, mu.OP_TYPE_DATA, addr)
			}else {
				log.Println(err)
			}
		}else if stype == "deleteInfo" || stype == "updateInfo" {
			m1 := map[string]interface{}{
				"infoid":  infoid,
				"stype": stype,
			}
			by, _ := json.Marshal(m1)
			log.Println(string(by))
			udp.WriteUdp(by, mu.OP_TYPE_DATA, addr)
		}else if sid != "" && eid != "" {
			m1 := map[string]interface{}{
				"gtid":  sid,
				"lteid": eid,
				"stype": stype,
			}
			by, _ := json.Marshal(m1)
			log.Println(string(by))
			udp.WriteUdp(by, mu.OP_TYPE_DATA, addr)
		}
	} else {
		flag.PrintDefaults()
		log.Println("参数错误.")
	}
}