commit 2021-12-16]17:46:11

change_permission.go

@@ -0,0 +1,20 @@
+package manager
+
+import (
+	"fmt"
+	"log"
+	"os/exec"
+)
+
+func ChangePermission(iface string) error {
+	log.Println("changing the mod of file")
+
+	cmd := exec.Command("chmod", "a+rw", iface)
+	b, err := cmd.CombinedOutput()
+	if err != nil {
+		return err
+	}
+
+	fmt.Println(string(b))
+	return nil
+}

cli.go

@@ -0,0 +1,94 @@
+package manager
+
+import (
+	"context"
+	"log"
+	"sync"
+
+	"git.godopu.net/lab/etri-smartfarm-poc-controller-serial/puserial"
+	"github.com/rjeczalik/notify"
+)
+
+var ctx context.Context
+var cancel context.CancelFunc
+var ch_discover chan notify.EventInfo
+var _managerObj *_manager
+var registerHandleFunc func(e Event)
+var removeHandleFunc func(e Event)
+
+func init() {
+	ctx, cancel = context.WithCancel(context.Background())
+	ch_discover = make(chan notify.EventInfo)
+	devWithUUID := map[string]*_device{}
+	devWithIface := map[interface{}]*_device{}
+	chanForSync := map[string]chan map[string]interface{}{}
+
+	_managerObj = &_manager{
+		devicesWithUUID:  devWithUUID,
+		devicesWithIface: devWithIface,
+		chanForSync:      chanForSync,
+		SyncListener:     &SyncHandler{devices: devWithUUID, chanForSync: chanForSync, mutex: &sync.Mutex{}, states: map[string]map[string]interface{}{}},
+		RecvListener:     &RecvHandler{devices: devWithIface, chanForSync: chanForSync},
+	}
+
+	registerHandleFunc = nil
+	removeHandleFunc = nil
+}
+
+func AddRegisterHandleFunc(h func(e Event)) {
+	registerHandleFunc = h
+}
+
+func AddRemoveHandleFunc(h func(e Event)) {
+	removeHandleFunc = h
+}
+
+func Close() {
+	cancel()
+}
+
+func Sync(key string, param map[string]interface{}) {
+	_managerObj.onSync(key, param)
+}
+
+func AddRecvListener(h EventHandler) {
+	_managerObj.addRecvListener(h)
+}
+
+// func SetDevicePropsToSync(uuid string, propsToSync []string) error {
+// 	device, ok := _managerObj.devicesWithUUID[uuid]
+// 	if !ok {
+// 		return errors.New("device not found")
+// 	}
+// 	device.propsToSync = propsToSync
+// 	return nil
+// }
+
+func Run() error {
+	ifaces, err := puserial.InitDevice()
+	if err != nil {
+		return err
+	}
+
+	for _, e := range ifaces {
+		go _managerObj.onAdded(e)
+	}
+
+	go puserial.WatchNewDevice(ctx, ch_discover)
+
+	for {
+		e, ok := <-ch_discover
+		if !ok {
+			log.Println("manager exit")
+			return nil
+		}
+		switch e.Event() {
+		case notify.Create:
+			go _managerObj.onAdded(e.Path())
+			// case notify.Remove:
+			// 	log.Println("USB Disconnected!!")
+			// 	_managerObj.onAdded(e.Path())
+		}
+
+	}
+}

communication.go

@@ -0,0 +1,65 @@
+package manager
+
+import (
+	"bufio"
+	"encoding/json"
+	"io"
+	"log"
+)
+
+func recv(port io.Reader, r Receiver) {
+	reader := bufio.NewReader(port)
+
+	for {
+		b, _, err := reader.ReadLine()
+		if err != nil {
+			if err == io.EOF {
+				log.Println("USB is disconnected")
+				_managerObj.onRemoved(port)
+				return
+			}
+		}
+		recvObj := map[string]interface{}{}
+		err = json.Unmarshal(b, &recvObj)
+
+		if err == nil {
+			r.onRecv(port, recvObj)
+		}
+		// data = string(b)
+	}
+}
+
+// for {
+// 	fmt.Print("> ")
+// 	cmd, _, _ := cmdReader.ReadLine()
+// 	cmdTkns := strings.Split(string(cmd), " ")
+
+// 	if cmdTkns[0] == "light" {
+// 		command["code"] = 1
+// 		if cmdTkns[1] == "on" {
+// 			command["light"] = 100
+// 		} else {
+// 			command["light"] = 0
+// 		}
+// 	} else if cmdTkns[0] == "fan" {
+// 		command["code"] = 2
+// 		if cmdTkns[1] == "on" {
+// 			command["status"] = 1
+// 		} else {
+// 			command["status"] = 0
+// 		}
+// 	} else if cmdTkns[0] == "servo" {
+// 		command["code"] = 3
+// 		angle, err := strconv.Atoi(cmdTkns[1])
+// 		if err != nil {
+// 			continue
+// 		}
+// 		command["angle"] = angle
+// 	} else if cmdTkns[0] == "print" {
+// 		fmt.Println(data)
+// 	}
+// 	err := encoder.Encode(command)
+// 	if err != nil {
+// 		panic(err)
+// 	}
+// }

examples/main.go

@@ -0,0 +1,51 @@
+package main
+
+import (
+	"bufio"
+	"fmt"
+	"os"
+	"strconv"
+	"strings"
+
+	manager "git.godopu.net/lab/etri-smartfarm-poc-controller-serial"
+)
+
+func main() {
+	manager.AddRecvListener(manager.NewEventHandler(func(e manager.Event) {
+		fmt.Println("RECV: ", e.Params())
+	}))
+
+	manager.AddRegisterHandleFunc(func(e manager.Event) {
+		param := e.Params()
+		fmt.Println(param["uuid"].(string), " is registered!! ] ", param["sname"])
+	})
+
+	manager.AddRemoveHandleFunc(func(e manager.Event) {
+		param := e.Params()
+		fmt.Println(param["uuid"].(string), " is removed!!")
+	})
+
+	go manager.Run()
+	defer manager.Close()
+
+	param := map[string]interface{}{}
+	for {
+		fmt.Print("> ")
+		reader := bufio.NewReader(os.Stdin)
+		b, _, _ := reader.ReadLine()
+		cmd := string(b)
+		fmt.Println("cmd: ", cmd)
+		if cmd == "exit" {
+			return
+		}
+
+		tkns := strings.Split(cmd, " ")
+		key := tkns[0]
+		value, err := strconv.Atoi(tkns[2])
+		if err != nil {
+			continue
+		}
+		param[tkns[1]] = value
+		manager.Sync(key, param)
+	}
+}

go.mod

@@ -2,6 +2,9 @@ module git.godopu.net/lab/etri-smartfarm-poc-controller-serial
 
 go 1.17
 
-require github.com/rjeczalik/notify v0.9.2
+require (
+	github.com/jacobsa/go-serial v0.0.0-20180131005756-15cf729a72d4
+	github.com/rjeczalik/notify v0.9.2
+)
 
 require golang.org/x/sys v0.0.0-20180926160741-c2ed4eda69e7 // indirect

go.sum

@@ -1,3 +1,5 @@
+github.com/jacobsa/go-serial v0.0.0-20180131005756-15cf729a72d4 h1:G2ztCwXov8mRvP0ZfjE6nAlaCX2XbykaeHdbT6KwDz0=
+github.com/jacobsa/go-serial v0.0.0-20180131005756-15cf729a72d4/go.mod h1:2RvX5ZjVtsznNZPEt4xwJXNJrM3VTZoQf7V6gk0ysvs=
 github.com/rjeczalik/notify v0.9.2 h1:MiTWrPj55mNDHEiIX5YUSKefw/+lCQVoAFmD6oQm5w8=
 github.com/rjeczalik/notify v0.9.2/go.mod h1:aErll2f0sUX9PXZnVNyeiObbmTlk5jnMoCa4QEjJeqM=
 golang.org/x/sys v0.0.0-20180926160741-c2ed4eda69e7 h1:bit1t3mgdR35yN0cX0G8orgLtOuyL9Wqxa1mccLB0ig=

interface.go

@@ -0,0 +1,21 @@
+package manager
+
+type Event interface {
+	Params() map[string]interface{}
+	Key() interface{}
+}
+type EventHandler interface {
+	Handle(e Event)
+}
+
+type Receiver interface {
+	onRecv(key interface{}, params map[string]interface{})
+}
+
+type Syncronizer interface {
+	onSync(key interface{}, params map[string]interface{})
+}
+
+
+
+

listener.go

@@ -0,0 +1,147 @@
+package manager
+
+import (
+	"encoding/json"
+	"fmt"
+	"log"
+	"reflect"
+	"sync"
+)
+
+type SyncHandler struct {
+	devices     map[string]*_device
+	mutex       *sync.Mutex
+	chanForSync map[string]chan map[string]interface{}
+	states      map[string]map[string]interface{}
+}
+
+func compareMap(src map[string]interface{}, dst map[string]interface{}) bool {
+
+	for key, value := range src {
+		if key == "code" {
+			continue
+		}
+
+		if reflect.TypeOf(value).String() == "string" {
+			dstV, ok := dst[key].(string)
+			if !ok || value != dstV {
+				return false
+			}
+			continue
+		}
+
+		srcV, ok := value.(float64)
+		if !ok {
+			srcV = float64(value.(int))
+		}
+
+		dstV, ok := dst[key].(float64)
+		if !ok {
+			dstV = float64(dst[key].(int))
+		}
+
+		if srcV != dstV {
+			fmt.Println("diff ", key, "] ", srcV, " vs ", dstV)
+			return false
+		}
+	}
+
+	return true
+}
+
+func (sh *SyncHandler) Handle(e Event) {
+	device, ok := sh.devices[e.Key().(string)]
+	// fmt.Println("sync] ", device.IfaceName)
+	// fmt.Println(sh.devices)
+	if !ok {
+		return
+	}
+
+	params := e.Params()
+	if params == nil {
+		return
+	}
+
+	go func() {
+		encoder := json.NewEncoder(device.Iface)
+		origin := map[string]interface{}{}
+		origin["code"] = 200
+		for key, value := range params {
+			origin[key] = value
+		}
+
+		sh.states[device.UUID] = origin
+		// props := []string{"fan", "light", "servo"}
+
+		err := encoder.Encode(origin)
+		if err != nil {
+			return
+		}
+
+		_, ok := sh.chanForSync[device.IfaceName]
+		if ok {
+			sh.mutex.Lock()
+			close(sh.chanForSync[device.IfaceName])
+			delete(sh.chanForSync, device.IfaceName)
+			sh.mutex.Unlock()
+		}
+
+		chanForSync := make(chan map[string]interface{})
+		sh.mutex.Lock()
+		sh.chanForSync[device.IfaceName] = chanForSync
+		sh.mutex.Unlock()
+
+		for state := range chanForSync {
+			if compareMap(sh.states[device.UUID], state) {
+				sh.mutex.Lock()
+				close(sh.chanForSync[device.IfaceName])
+				delete(sh.chanForSync, device.IfaceName)
+				sh.mutex.Unlock()
+				return
+			}
+			log.Println("wrong: ", state)
+			log.Println("resend: ", sh.states[device.UUID])
+			err := encoder.Encode(sh.states[device.UUID])
+			if err != nil {
+				return
+			}
+		}
+
+	}()
+}
+
+type RecvHandler struct {
+	devices     map[interface{}]*_device
+	chanForSync map[string]chan map[string]interface{}
+	next        EventHandler
+}
+
+func (rh *RecvHandler) Handle(e Event) {
+	defer func() {
+		if r := recover(); r != nil {
+			fmt.Println("panic recover - ", r)
+		}
+	}()
+	device, ok := rh.devices[e.Key()]
+	if !ok {
+		return
+	}
+
+	param := e.Params()
+	code, _ := param["code"].(float64)
+	if int(code) == 100 {
+		return
+	}
+
+	device.states = param
+	channel, ok := rh.chanForSync[device.IfaceName]
+	if ok {
+		channel <- device.states
+	}
+
+	if rh.next != nil {
+		rh.next.Handle(e)
+	}
+
+	// fmt.Println("recv] ", device.states)
+}

manager.go

@@ -1,27 +1,131 @@
 package manager
 
 import (
-	"context"
+	"bufio"
+	"encoding/json"
 	"fmt"
-	"time"
+	"io"
+	"log"
 
-	"git.godopu.net/lab/etri-smartfarm-poc-controller-serial/puserial"
+	"github.com/jacobsa/go-serial/serial"
-	"github.com/rjeczalik/notify"
 )
 
-func Run() {
+type _manager struct {
-	fmt.Println("Hello world")
+	devicesWithUUID  map[string]*_device
+	devicesWithIface map[interface{}]*_device
+	chanForSync      map[string]chan map[string]interface{}
+	RegisterListener EventHandler
+	SyncListener     EventHandler
+	RecvListener     EventHandler
+}
 
-	ctx, _ := context.WithTimeout(context.Background(), time.Second*10)
+func (m *_manager) onRegistered(dev *_device) {
-	ch_discover := make(chan notify.EventInfo)
+	// if m.RegisterListener != nil {
-	go puserial.WatchNewDevice(ctx, ch_discover)
+	// 	m.RegisterListener.Handle(&EventStruct{key: })
+	// }
+	if registerHandleFunc != nil {
+		param := map[string]interface{}{}
+		param["uuid"] = dev.UUID
+		param["sname"] = dev.Sname
 
-	for {
+		registerHandleFunc(&EventStruct{key: dev.UUID, params: param})
-		e, ok := <-ch_discover
+	}
-		if !ok {
+	go recv(dev.Iface, m)
-			fmt.Println("byebye~!")
+}
+
+func (m *_manager) onRemoved(port io.Reader) {
+	dev := _managerObj.devicesWithIface[port]
+	delete(m.devicesWithUUID, dev.UUID)
+	delete(m.devicesWithIface, port)
+	ch, ok := m.chanForSync[dev.UUID]
+	if ok {
+		close(ch)
+		delete(m.chanForSync, dev.UUID)
+	}
+
+	if removeHandleFunc != nil {
+		param := map[string]interface{}{}
+		param["uuid"] = dev.UUID
+
+		removeHandleFunc(&EventStruct{key: dev.UUID, params: param})
+	}
+	// log.Println(m.devicesWithUUID)
+	// log.Println(m.devicesWithIface)
+}
+
+func (m *_manager) onAdded(iface string) {
+	err := ChangePermission(iface)
+	if err != nil {
+		panic(err)
+	}
+
+	// Set up options.
+	options := serial.OpenOptions{
+		PortName:        iface,
+		BaudRate:        9600,
+		DataBits:        8,
+		StopBits:        1,
+		MinimumReadSize: 16,
+	}
+
+	// Open the port.
+	go func() {
+		port, err := serial.Open(options)
+		if err != nil {
+			log.Printf("serial.Open: %v", err)
 			return
 		}
-		fmt.Println(e)
+
+		reader := bufio.NewReader(port)
+		encoder := json.NewEncoder(port)
+
+		sndMsg := map[string]interface{}{}
+		sndMsg["code"] = 100
+
+		for {
+			b, _, _ := reader.ReadLine()
+			rcvMsg := map[string]interface{}{}
+			err := json.Unmarshal(b, &rcvMsg)
+
+			if err != nil {
+				continue
+			}
+
+			code, ok := rcvMsg["code"].(float64)
+			if ok && code == 100 {
+				newDevice := &_device{
+					UUID:      rcvMsg["uuid"].(string),
+					IfaceName: iface,
+					Iface:     port,
+					Sname:     rcvMsg["sname"].(string),
+					states:    map[string]interface{}{},
+				}
+				m.devicesWithUUID[rcvMsg["uuid"].(string)] = newDevice
+				m.devicesWithIface[port] = newDevice
+
+				m.onRegistered(newDevice)
+				fmt.Println("onAdded sub-routine is died")
+				return
+			}
+			encoder.Encode(sndMsg)
+		}
+	}()
+
+	fmt.Println("onAdded main-routine is died")
+}
+
+func (m *_manager) onSync(key interface{}, params map[string]interface{}) {
+	if m.SyncListener != nil {
+		m.SyncListener.Handle(&EventStruct{key: key, params: params})
 	}
 }
+
+func (m *_manager) onRecv(key interface{}, params map[string]interface{}) {
+	if m.RecvListener != nil {
+		m.RecvListener.Handle(&EventStruct{key: key, params: params})
+	}
+}
+
+func (m *_manager) addRecvListener(h EventHandler) {
+	m.RecvListener = &RecvHandler{next: h, devices: m.devicesWithIface, chanForSync: m.chanForSync}
+}

model.go

@@ -0,0 +1,15 @@
+package manager
+
+import "io"
+
+type _device struct {
+	UUID      string
+	IfaceName string
+	Sname     string
+	Iface     io.ReadWriter
+	states    map[string]interface{}
+}
+
+type RecvEvent struct {
+	Params map[string]interface{}
+}

puserial/serial.go

@@ -3,16 +3,34 @@ package puserial
 import (
 	"context"
 	"fmt"
+	"io/ioutil"
+	"path/filepath"
 	"strings"
 
 	"github.com/rjeczalik/notify"
 )
 
+func InitDevice() ([]string, error) {
+	fs, err := ioutil.ReadDir("/dev")
+	if err != nil {
+		return nil, err
+	}
+
+	var result []string = nil
+	for _, f := range fs {
+		if strings.Contains(f.Name(), "ttyACM") {
+			result = append(result, filepath.Join("/dev", f.Name()))
+		}
+	}
+
+	return result, nil
+}
+
 func WatchNewDevice(ctx context.Context, ch_discover chan<- notify.EventInfo) error {
 	defer close(ch_discover)
 
 	filter := make(chan notify.EventInfo, 1)
-	if err := notify.Watch("/dev", filter, notify.Create, notify.Remove); err != nil {
+	if err := notify.Watch("/dev", filter, notify.Create); err != nil {
 		return err
 	}
 	defer notify.Stop(filter)
@@ -20,7 +38,6 @@ func WatchNewDevice(ctx context.Context, ch_discover chan<- notify.EventInfo) er
 	for {
 		select {
 		case <-ctx.Done():
-			fmt.Println("routine bye")
 			return nil
 		case e := <-filter:
 			if strings.Contains(e.Path(), "/dev/ttyACM") {

structure.go

@@ -0,0 +1,28 @@
+package manager
+
+type EventStruct struct {
+	Event
+	params map[string]interface{}
+	key    interface{}
+}
+
+func (es *EventStruct) Params() map[string]interface{} {
+	return es.params
+}
+
+func (es *EventStruct) Key() interface{} {
+	return es.key
+}
+
+type EventHandlerStruct struct {
+	EventHandler
+	HandleFunc func(e Event)
+}
+
+func (ehs *EventHandlerStruct) Handle(e Event) {
+	ehs.HandleFunc(e)
+}
+
+func NewEventHandler(h func(e Event)) *EventHandlerStruct {
+	return &EventHandlerStruct{HandleFunc: h}
+}