[WIP]kadai3 happylifetaka

kadai3-1/happylifetaka/typing.go

@@ -0,0 +1,12 @@
+package main
+
+import (
+	"os"
+
+	"github.com/happylifetaka/dojo4/kadai3-1/happylifetaka/typinggame"
+)
+
+func main() {
+	var t typinggame.TypingGame
+	t.Start(os.Stdin)
+}

kadai3-1/happylifetaka/typinggame/typinggame.go

@@ -0,0 +1,82 @@
+package typinggame
+
+import (
+	"bufio"
+	"fmt"
+	"io"
+	"math/rand"
+	"time"
+)
+
+//TypingGame 1 minute typing game.
+type TypingGame int
+
+//Start start 1 minute typing game.
+//Param
+//r:io.Reader example:os.Stdin
+func (t *TypingGame) Start(r io.Reader) {
+	ch1 := input(r)
+	ch2 := wait(60)
+	var words = []string{"apple", "banana", "cherry", "plum", "grape", "pineapple"}
+
+	shuffle(words)
+	i := 0
+	//success count
+	sucessCnt := 0
+	//fail count
+	failCnt := 0
+	fmt.Println("try typing.1 minute.")
+	fmt.Println(words[i])
+
+TIMEOUT_LABEL:
+	for {
+		select {
+		case msg := <-ch1:
+			if words[i] == msg {
+				if len(words) <= (i + 1) {
+					i = 0
+				} else {
+					i++
+				}
+				sucessCnt++
+			} else {
+				fmt.Println("miss.retyping words.")
+				failCnt++
+			}
+			fmt.Println(words[i])
+		case <-ch2:
+			fmt.Println("")
+			fmt.Println("time up.success count:", sucessCnt, " fail count:", failCnt)
+
+			break TIMEOUT_LABEL
+		}
+	}
+}
+
+func shuffle(a []string) {
+	for i := len(a) - 1; i >= 0; i-- {
+		j := rand.Intn(i + 1)
+		a[i], a[j] = a[j], a[i]
+	}
+}
+
+func input(r io.Reader) <-chan string {
+	ch := make(chan string)
+	go func() {
+		s := bufio.NewScanner(r)
+		defer close(ch)
+		for s.Scan() {
+			ch <- s.Text()
+		}
+	}()
+	return ch
+}
+
+func wait(sec int) <-chan bool {
+	ch := make(chan bool)
+	go func() {
+		time.Sleep(time.Duration(sec) * time.Second)
+		ch <- true
+	}()
+	return ch
+}

kadai3-2/happylifetaka/downloader.go

@@ -0,0 +1,41 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"os"
+
+	"github.com/happylifetaka/dojo4/kadai3-2/happylifetaka/downloader"
+)
+
+type downloadByteInfo struct {
+	start int64
+	end   int64
+}
+
+func main() {
+	div := flag.Int64("div", 3, "file download division")
+	usageMsg := "udage:downloader [-div] url saveFilePath"
+	flag.Usage = func() {
+		fmt.Println(usageMsg)
+		flag.PrintDefaults()
+		os.Exit(0)
+	}
+	flag.Parse()
+	args := flag.Args()
+	if len(args) != 2 {
+		fmt.Println("parameter error.")
+		fmt.Println(usageMsg)
+		flag.PrintDefaults()
+		os.Exit(0)
+	}
+	url := args[0]
+	saveFilePath := args[1]
+
+	var d downloader.Downloader
+	if err := d.Download(url, saveFilePath, *div); err != nil {
+		fmt.Println(err)
+		os.Exit(1)
+	}
+	os.Exit(0)
+}

kadai3-2/happylifetaka/downloader/downloader.go

@@ -0,0 +1,157 @@
+package downloader
+
+import (
+	"fmt"
+	"io"
+	"io/ioutil"
+	"net/http"
+	"os"
+	"strconv"
+
+	"golang.org/x/sync/errgroup"
+)
+
+type Downloader int
+
+type downloadInfo struct {
+	startByte int64
+	endByte   int64
+}
+
+// Download
+// description:Download specified URL.
+// parameter
+// url :download target url.
+// saveFilePath:save file path.
+// div:Number of divided downloads.
+func (d *Downloader) Download(url string, saveFilePath string, div int64) error {
+	res, err := http.Head(url)
+	if err != nil {
+		fmt.Println("http Head error")
+		return err
+	}
+	if res.StatusCode != 200 {
+		fmt.Println("bad status code")
+		fmt.Println("status code:", res.StatusCode)
+		return err
+	}
+
+	if !canRangeDownload(res.Header) {
+		fmt.Println("range download not support.")
+		return err
+	}
+
+	di := splitDownloadLength(res.ContentLength, div)
+	filenames := make([]string, div)
+
+	var eg errgroup.Group
+
+	i := 1
+	for _, d := range di {
+		j := i
+		eg.Go(func() error {
+			err := rangeDownload(j, url, d.startByte, d.endByte)
+			return err
+		})
+		filenames[i-1] = strconv.Itoa(j) + ".temp.download"
+		i++
+	}
+	if err := eg.Wait(); err != nil {
+		fmt.Println("download error")
+		return err
+	}
+
+	if err := joinFiles(filenames, saveFilePath); err != nil {
+		fmt.Println("join file error")
+		return err
+	}
+
+	if err := deleteFiles(filenames); err != nil {
+		fmt.Println("join file error")
+		return err
+	}
+	fmt.Println("download finish.")
+	return nil
+}
+
+func canRangeDownload(h http.Header) bool {
+	f := false
+	for k, v := range h {
+		if k == "Accept-Ranges" && len(v) > 0 && v[0] == "bytes" {
+			f = true
+			break
+		}
+	}
+	return f
+}
+
+func splitDownloadLength(length int64, div int64) []downloadInfo {
+
+	divLength := length / div
+
+	a := make([]downloadInfo, div)
+
+	var i int64
+	for i = 0; i < div; i++ {
+		s := i * divLength
+		e := (i+1)*divLength - 1
+
+		a[i] = downloadInfo{s, e}
+	}
+	return a
+}
+
+func rangeDownload(index int, url string, s int64, e int64) error {
+	req, err := http.NewRequest("GET", url, nil)
+	if err != nil {
+		return err
+	}
+
+	req.Header.Set("Range", fmt.Sprintf("bytes=%d-%d", s, e))
+
+	res, err := http.DefaultClient.Do(req)
+	defer res.Body.Close()
+	if err != nil {
+		return err
+	}
+
+	out_path := strconv.Itoa(index) + ".temp.download"
+	out, err := os.Create(out_path)
+	defer out.Close()
+	if err != nil {
+		return err
+	}
+
+	io.Copy(out, res.Body)
+	return nil
+}
+
+func joinFiles(filenames []string, saveFilePath string) error {
+	files := make([]io.Reader, len(filenames))
+
+	for i, filename := range filenames {
+		files[i], _ = os.Open(filename)
+	}
+
+	reader := io.MultiReader(files...)
+	b, _ := ioutil.ReadAll(reader)
+
+	file, err := os.Create(saveFilePath)
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+
+	file.Write(([]byte)(b))
+	return nil
+}
+
+func deleteFiles(filenames []string) error {
+	for _, f := range filenames {
+		err := os.Remove(f)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}