doorgo/door/write_linux.go

package door

import (
	"log"
	"strings"
	"syscall"
	"sync/atomic"
)

// https://go101.org/article/channel-closing.html

// This will get renamed back to Writer, once I get
// it working...
func Writer2(handle int, d *Door) {
	log.Println("Writer2")
	defer d.wg.Done()
	for {
		select {
		case <- d.closeChannel:
		close(d.writerChannel)
		log.Println("~Writer2")
		return
		default:
		}

		select {
                case <- d.closeChannel:
		close(d.writerChannel)
		log.Println("~Writer2")
		return
		case output, ok := <-d.writerChannel:
			if !ok {
				log.Println("closeChannel")
				if ! d.Disconnect() {
				// d.Disconnected = true
				atomic.StoreInt32(&d.Disconnected, 1)
				// if !ok, the channel is already closed...
				// close(d.writerChannel)
}
				log.Println("~Writer2")
				return

//				d.closeChannel <- true
//				continue
			} else {
				// log.Println("output")
				/* Handle cases where we're updating a portion of the screen.
				When we SavePos, also save/restore the last color set.
				*/
				if strings.HasSuffix(output, RestorePos) {
					output += Color(d.LastColor...)

				} else {
					d.UpdateLastColor(output, &d.LastColor)
				}
				// log.Println("write output")
				buffer := []byte(output)
				n, err := syscall.Write(handle, buffer)
				if (err != nil) || (n != len(buffer)) {
					log.Println("closeChannel")
					if ! d.Disconnect() {
					// d.Disconnected = true
					atomic.StoreInt32(&d.Disconnected, 1)
					close(d.writerChannel)
					}
					log.Println("~Writer2")
					return
				}
			}
		}
	}
}

func Writer(handle int, writerChannel *chan string) {
	for output := range *writerChannel {
		buffer := []byte(output)
		n, err := syscall.Write(handle, buffer)
		if (err != nil) || (n != len(buffer)) {
			close(*writerChannel)
			break
		}
	}
}