优化项目结构
parent
0f2fe50b04
commit
a4004a1d05
|
@ -0,0 +1,49 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/hex"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Command struct {
|
||||||
|
name string
|
||||||
|
description string
|
||||||
|
function func()
|
||||||
|
}
|
||||||
|
|
||||||
|
func cmdhelp() {
|
||||||
|
var page = 0
|
||||||
|
fmt.Printf(">-------Help(%v)-------<\n", page)
|
||||||
|
for i := 0; i < len(commands); i++ {
|
||||||
|
strout(config.outputCode, fmt.Sprintf(" %-10v --%v\n", commands[i].name, commands[i].description))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func cmdexit() {
|
||||||
|
os.Exit(0)
|
||||||
|
}
|
||||||
|
func cmdargs() {
|
||||||
|
fmt.Printf(">-------Args(%v)-------<\n", len(args)-1)
|
||||||
|
fmt.Printf("%q\n", args[1:])
|
||||||
|
}
|
||||||
|
func cmdhex() {
|
||||||
|
fmt.Printf(">-----Hex Send-----<\n")
|
||||||
|
fmt.Printf("%q\n", args[1:])
|
||||||
|
s := strings.Join(args[1:], "")
|
||||||
|
b, err := hex.DecodeString(s)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
_, err = serialPort.Write(b)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func cmdinit() {
|
||||||
|
commands = append(commands, Command{name: ".help", description: "帮助信息", function: cmdhelp})
|
||||||
|
commands = append(commands, Command{name: ".args", description: "参数信息", function: cmdargs})
|
||||||
|
commands = append(commands, Command{name: ".hex", description: "发送Hex", function: cmdhex})
|
||||||
|
commands = append(commands, Command{name: ".exit", description: "退出终端", function: cmdexit})
|
||||||
|
}
|
|
@ -0,0 +1,14 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
type Config struct {
|
||||||
|
portName string
|
||||||
|
baudRate int
|
||||||
|
dataBits int
|
||||||
|
stopBits int
|
||||||
|
outputCode string
|
||||||
|
inputCode string
|
||||||
|
endStr string
|
||||||
|
enableLog bool
|
||||||
|
logFilePath string
|
||||||
|
parityBit int
|
||||||
|
}
|
|
@ -0,0 +1,97 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"flag"
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ptrVal struct {
|
||||||
|
*string
|
||||||
|
*int
|
||||||
|
*bool
|
||||||
|
*float64
|
||||||
|
*float32
|
||||||
|
}
|
||||||
|
type Val struct {
|
||||||
|
string
|
||||||
|
int
|
||||||
|
bool
|
||||||
|
float64
|
||||||
|
float32
|
||||||
|
}
|
||||||
|
type Flag struct {
|
||||||
|
v ptrVal
|
||||||
|
sStr string
|
||||||
|
lStr string
|
||||||
|
dv Val
|
||||||
|
help string
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
portName = Flag{ptrVal{string: &config.portName}, "p", "port", Val{string: ""}, "要连接的串口\t(/dev/ttyUSB0、COMx)"}
|
||||||
|
baudRate = Flag{ptrVal{int: &config.baudRate}, "b", "baud", Val{int: 115200}, "波特率"}
|
||||||
|
dataBits = Flag{ptrVal{int: &config.dataBits}, "d", "data", Val{int: 8}, "数据位"}
|
||||||
|
stopBits = Flag{ptrVal{int: &config.stopBits}, "s", "stop", Val{int: 0}, "停止位停止位(0: 1停止 1:1.5停止 2:2停止)"}
|
||||||
|
outputCode = Flag{ptrVal{string: &config.outputCode}, "o", "out", Val{string: "UTF-8"}, "输出编码"}
|
||||||
|
inputCode = Flag{ptrVal{string: &config.inputCode}, "i", "in", Val{string: "UTF-8"}, "输入编码"}
|
||||||
|
endStr = Flag{ptrVal{string: &config.endStr}, "e", "end", Val{string: "\n"}, "终端换行符"}
|
||||||
|
enableLog = Flag{ptrVal{bool: &config.enableLog}, "l", "log", Val{bool: false}, "是否启用日志保存"}
|
||||||
|
logFilePath = Flag{ptrVal{string: &config.logFilePath}, "P", "Path", Val{string: "./Log.txt"}, "日志保存路径"}
|
||||||
|
parityBit = Flag{ptrVal{int: &config.parityBit}, "v", "verify", Val{int: 0}, "奇偶校验(0:无校验、1:奇校验、2:偶校验、3:1校验、4:0校验)"}
|
||||||
|
flags = []Flag{portName, baudRate, dataBits, stopBits, outputCode, inputCode, endStr, enableLog, logFilePath, parityBit}
|
||||||
|
)
|
||||||
|
|
||||||
|
type ValType int
|
||||||
|
|
||||||
|
const (
|
||||||
|
notVal ValType = iota
|
||||||
|
stringVal
|
||||||
|
intVal
|
||||||
|
boolVal
|
||||||
|
floatVal
|
||||||
|
)
|
||||||
|
|
||||||
|
func printUsage(ports []string) {
|
||||||
|
fmt.Printf("\n参数帮助:\n")
|
||||||
|
for _, f := range flags {
|
||||||
|
flagprint(f)
|
||||||
|
}
|
||||||
|
fmt.Printf("\n在线串口: %v\n", strings.Join(ports, ","))
|
||||||
|
}
|
||||||
|
func flagFindValue(v ptrVal) ValType {
|
||||||
|
if v.string != nil {
|
||||||
|
return stringVal
|
||||||
|
}
|
||||||
|
if v.bool != nil {
|
||||||
|
return boolVal
|
||||||
|
}
|
||||||
|
if v.int != nil {
|
||||||
|
return intVal
|
||||||
|
}
|
||||||
|
return notVal
|
||||||
|
}
|
||||||
|
func flagprint(f Flag) {
|
||||||
|
switch flagFindValue(f.v) {
|
||||||
|
case stringVal:
|
||||||
|
fmt.Printf("\t-%v -%v %T \n\t %v\t默认值:%q\n", f.sStr, f.lStr, f.dv.string, f.help, f.dv.string)
|
||||||
|
case intVal:
|
||||||
|
fmt.Printf("\t-%v -%v %T \n\t %v\t默认值:%v\n", f.sStr, f.lStr, f.dv.int, f.help, f.dv.int)
|
||||||
|
case boolVal:
|
||||||
|
fmt.Printf("\t-%v -%v %T \n\t %v\t默认值:%v\n", f.sStr, f.lStr, f.dv.bool, f.help, f.dv.bool)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func flagInit(f *Flag) {
|
||||||
|
if f.v.string != nil {
|
||||||
|
flag.StringVar(f.v.string, f.sStr, f.dv.string, "")
|
||||||
|
flag.StringVar(f.v.string, f.lStr, f.dv.string, f.help)
|
||||||
|
}
|
||||||
|
if f.v.bool != nil {
|
||||||
|
flag.BoolVar(f.v.bool, f.sStr, f.dv.bool, "")
|
||||||
|
flag.BoolVar(f.v.bool, f.lStr, f.dv.bool, f.help)
|
||||||
|
}
|
||||||
|
if f.v.int != nil {
|
||||||
|
flag.IntVar(f.v.int, f.sStr, f.dv.int, "")
|
||||||
|
flag.IntVar(f.v.int, f.lStr, f.dv.int, f.help)
|
||||||
|
}
|
||||||
|
}
|
139
main.go
139
main.go
|
@ -2,7 +2,6 @@ package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
"encoding/hex"
|
|
||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/zimolab/charsetconv"
|
"github.com/zimolab/charsetconv"
|
||||||
|
@ -13,24 +12,6 @@ import (
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Config struct {
|
|
||||||
enableLog bool
|
|
||||||
logFilePath string
|
|
||||||
portName string
|
|
||||||
endStr string
|
|
||||||
inputCode string
|
|
||||||
outputCode string
|
|
||||||
baudRate int
|
|
||||||
parityBit int
|
|
||||||
stopBits int
|
|
||||||
dataBits int
|
|
||||||
}
|
|
||||||
type Command struct {
|
|
||||||
name string
|
|
||||||
description string
|
|
||||||
function func()
|
|
||||||
}
|
|
||||||
|
|
||||||
var (
|
var (
|
||||||
config Config
|
config Config
|
||||||
commands []Command
|
commands []Command
|
||||||
|
@ -39,105 +20,37 @@ var (
|
||||||
args []string
|
args []string
|
||||||
)
|
)
|
||||||
|
|
||||||
func checkPortAvailability() {
|
func checkPortAvailability(name string) ([]string, error) {
|
||||||
ports, err := serial.GetPortsList()
|
ports, err := serial.GetPortsList()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
if len(ports) == 0 {
|
if len(ports) == 0 {
|
||||||
log.Fatal("未找到串口!")
|
return nil, fmt.Errorf("无串口")
|
||||||
|
}
|
||||||
|
if name == "" {
|
||||||
|
return ports, fmt.Errorf("串口未指定")
|
||||||
}
|
}
|
||||||
fmt.Printf("找到的串口: ")
|
|
||||||
for _, port := range ports {
|
for _, port := range ports {
|
||||||
fmt.Printf(" %v", port)
|
if strings.Compare(port, name) == 0 {
|
||||||
|
return ports, nil
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
return ports, fmt.Errorf("串口 " + name + " 未在线")
|
||||||
}
|
}
|
||||||
func printUsage() {
|
|
||||||
checkPortAvailability()
|
|
||||||
fmt.Printf("\n参数帮助:\n")
|
|
||||||
|
|
||||||
fmt.Printf("\t-%v -%v %T \n\t %v\t默认值:%v\n", "p", "port", "", "连接的串口(/dev/ttyUSB0、COMx)", "")
|
|
||||||
fmt.Printf("\t-%v -%v %T \n\t %v\t默认值:%v\n", "b", "baud", 115200, "波特率", 115200)
|
|
||||||
fmt.Printf("\t-%v -%v %T \n\t %v\t默认值:%v\n", "d", "data", 8, "数据位", 8)
|
|
||||||
fmt.Printf("\t-%v -%v %T \n\t %v\t默认值:%v\n", "s", "stop", 0, "停止位停止位(0: 1停止 1:1.5停止 2:2停止)", 0)
|
|
||||||
fmt.Printf("\t-%v -%v %T \n\t %v\t默认值:%v\n", "o", "out", "UTF-8", "输出编码", "UTF-8")
|
|
||||||
fmt.Printf("\t-%v -%v %T \n\t %v\t默认值:%v\n", "i", "in", "UTF-8", "输入编码", "UTF-8")
|
|
||||||
fmt.Printf("\t-%v -%v %T \n\t %v\t默认值:%v\n", "e", "end", "\n", "终端换行符", "\\n")
|
|
||||||
fmt.Printf("\t-%v -%v %T \n\t %v\t默认值:%v\n", "l", "log", false, "是否启用日志保存", false)
|
|
||||||
fmt.Printf("\t-%v -%v %T \n\t %v\t默认值:%v\n", "P", "Path", "./Log.txt", "日志保存路径", "./Log.txt")
|
|
||||||
fmt.Printf("\t-%v -%v %T \n\t %v\t默认值:%v\n", "v", "verify", 0, "奇偶校验(0:无校验、1:奇校验、2:偶校验、3:1校验、4:0校验)", 0)
|
|
||||||
}
|
|
||||||
func init() {
|
func init() {
|
||||||
log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile | log.Lmsgprefix)
|
log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile | log.Lmsgprefix)
|
||||||
flag.BoolVar(&config.enableLog, "log", false, "是否启用日志保存")
|
for _, f := range flags {
|
||||||
flag.BoolVar(&config.enableLog, "l", false, "")
|
flagInit(&f)
|
||||||
|
}
|
||||||
flag.StringVar(&config.logFilePath, "Path", "./Log.txt", "日志保存路径")
|
|
||||||
flag.StringVar(&config.logFilePath, "P", "./Log.txt", "")
|
|
||||||
|
|
||||||
flag.StringVar(&config.portName, "port", "", "连接的串口\t(/dev/ttyUSB0、COMx)")
|
|
||||||
flag.StringVar(&config.portName, "p", "", "")
|
|
||||||
|
|
||||||
flag.StringVar(&config.endStr, "end", "\n", "终端换行符")
|
|
||||||
flag.StringVar(&config.endStr, "e", "\n", "")
|
|
||||||
|
|
||||||
flag.IntVar(&config.baudRate, "baud", 115200, "波特率")
|
|
||||||
flag.IntVar(&config.baudRate, "b", 115200, "")
|
|
||||||
|
|
||||||
flag.IntVar(&config.parityBit, "verify", 0, "奇偶校验(0:无校验 1:奇校验 2:偶校验 3:1校验 4:0校验)")
|
|
||||||
flag.IntVar(&config.parityBit, "v", 0, "")
|
|
||||||
|
|
||||||
flag.IntVar(&config.stopBits, "stop", 0, "停止位(0: 1停止 1:1.5停止 2:2停止)")
|
|
||||||
flag.IntVar(&config.stopBits, "s", 0, "")
|
|
||||||
|
|
||||||
flag.IntVar(&config.dataBits, "data", 8, "数据位")
|
|
||||||
flag.IntVar(&config.dataBits, "d", 8, "")
|
|
||||||
|
|
||||||
flag.StringVar(&config.outputCode, "out", "UTF-8", "输出编码")
|
|
||||||
flag.StringVar(&config.outputCode, "o", "UTF-8", "")
|
|
||||||
|
|
||||||
flag.StringVar(&config.inputCode, "in", "UTF-8", "输入编码")
|
|
||||||
flag.StringVar(&config.inputCode, "i", "UTF-8", "")
|
|
||||||
cmdinit()
|
cmdinit()
|
||||||
}
|
}
|
||||||
func cmdhelp() {
|
|
||||||
var page = 0
|
|
||||||
fmt.Printf(">-------Help(%v)-------<\n", page)
|
|
||||||
for i := 0; i < len(commands); i++ {
|
|
||||||
strout(config.outputCode, fmt.Sprintf(" %-10v --%v\n", commands[i].name, commands[i].description))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
func cmdexit() {
|
|
||||||
os.Exit(0)
|
|
||||||
}
|
|
||||||
func cmdargs() {
|
|
||||||
fmt.Printf(">-------Args(%v)-------<\n", len(args)-1)
|
|
||||||
fmt.Printf("%q\n", args[1:])
|
|
||||||
}
|
|
||||||
func cmdhex() {
|
|
||||||
fmt.Printf(">-----Hex Send-----<\n")
|
|
||||||
fmt.Printf("%q\n", args[1:])
|
|
||||||
s := strings.Join(args[1:], "")
|
|
||||||
b, err := hex.DecodeString(s)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
_, err = serialPort.Write(b)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
func cmdinit() {
|
|
||||||
commands = append(commands, Command{name: ".help", description: "帮助信息", function: cmdhelp})
|
|
||||||
commands = append(commands, Command{name: ".args", description: "参数信息", function: cmdargs})
|
|
||||||
commands = append(commands, Command{name: ".hex", description: "发送Hex", function: cmdhex})
|
|
||||||
commands = append(commands, Command{name: ".exit", description: "退出终端", function: cmdexit})
|
|
||||||
}
|
|
||||||
func input() {
|
func input() {
|
||||||
input := bufio.NewScanner(os.Stdin)
|
input := bufio.NewScanner(os.Stdin)
|
||||||
var ok = false
|
var ok = false
|
||||||
for {
|
for input.Scan() {
|
||||||
input.Scan()
|
|
||||||
ok = false
|
ok = false
|
||||||
args = strings.Split(input.Text(), " ")
|
args = strings.Split(input.Text(), " ")
|
||||||
for _, cmd := range commands {
|
for _, cmd := range commands {
|
||||||
|
@ -147,29 +60,35 @@ func input() {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if !ok {
|
if !ok {
|
||||||
_, err := serialPort.Write(input.Bytes())
|
_, err := io.WriteString(serialPort, input.Text())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
_, err = io.WriteString(serialPort, config.endStr)
|
_, err = io.WriteString(serialPort, config.endStr)
|
||||||
//_, err = io.Copy(portName, os.Stdin)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
err := serialPort.Drain()
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func strout(cs, str string) {
|
func strout(cs, str string) {
|
||||||
err = charsetconv.EncodeWith(strings.NewReader(str), os.Stdout, charsetconv.Charset(cs), false)
|
err = charsetconv.EncodeWith(strings.NewReader(str), os.Stdout, charsetconv.Charset(cs), false)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func output(out io.Writer) {
|
func output(out io.Writer) {
|
||||||
if strings.Compare(config.inputCode, "hex") == 0 {
|
if strings.Compare(config.inputCode, "hex") == 0 {
|
||||||
_, err = io.Copy(hex.NewEncoder(out), serialPort)
|
b, _ := bufio.NewReader(io.LimitReader(serialPort, 16)).Peek(16)
|
||||||
|
_, err = fmt.Fprintf(out, "% X %q \n", b, b)
|
||||||
} else {
|
} else {
|
||||||
err = charsetconv.ConvertWith(serialPort, charsetconv.Charset(config.inputCode), out, charsetconv.Charset(config.outputCode), false)
|
err = charsetconv.ConvertWith(io.LimitReader(serialPort, 1024), charsetconv.Charset(config.inputCode), out, charsetconv.Charset(config.outputCode), false)
|
||||||
}
|
}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
|
@ -177,9 +96,10 @@ func output(out io.Writer) {
|
||||||
}
|
}
|
||||||
func main() {
|
func main() {
|
||||||
flag.Parse()
|
flag.Parse()
|
||||||
if config.portName == "" {
|
ports, err := checkPortAvailability(config.portName)
|
||||||
fmt.Println("端口未指定")
|
if err != nil {
|
||||||
printUsage()
|
fmt.Println(err)
|
||||||
|
printUsage(ports)
|
||||||
os.Exit(0)
|
os.Exit(0)
|
||||||
}
|
}
|
||||||
mode := &serial.Mode{
|
mode := &serial.Mode{
|
||||||
|
@ -195,10 +115,11 @@ func main() {
|
||||||
defer func(port serial.Port) {
|
defer func(port serial.Port) {
|
||||||
err := port.Close()
|
err := port.Close()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|
||||||
}
|
}
|
||||||
}(serialPort)
|
}(serialPort)
|
||||||
|
|
||||||
go input()
|
go input()
|
||||||
|
|
||||||
if config.enableLog {
|
if config.enableLog {
|
||||||
f, err := os.OpenFile(config.logFilePath, os.O_RDWR|os.O_APPEND|os.O_CREATE, 0666)
|
f, err := os.OpenFile(config.logFilePath, os.O_RDWR|os.O_APPEND|os.O_CREATE, 0666)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|
Loading…
Reference in New Issue