add: io reader composition
This commit is contained in:
21
interfaces/15-io-reader-composition/book.go
Normal file
21
interfaces/15-io-reader-composition/book.go
Normal file
@ -0,0 +1,21 @@
|
|||||||
|
// For more tutorials: https://blog.learngoprogramming.com
|
||||||
|
//
|
||||||
|
// Copyright © 2018 Inanc Gumus
|
||||||
|
// Learn Go Programming Course
|
||||||
|
// License: https://creativecommons.org/licenses/by-nc-sa/4.0/
|
||||||
|
//
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
type book struct {
|
||||||
|
product
|
||||||
|
Published timestamp
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *book) String() string {
|
||||||
|
return fmt.Sprintf("%s - (%s)", &b.product, b.Published)
|
||||||
|
}
|
35
interfaces/15-io-reader-composition/counter.go
Normal file
35
interfaces/15-io-reader-composition/counter.go
Normal file
@ -0,0 +1,35 @@
|
|||||||
|
// For more tutorials: https://blog.learngoprogramming.com
|
||||||
|
//
|
||||||
|
// Copyright © 2018 Inanc Gumus
|
||||||
|
// Learn Go Programming Course
|
||||||
|
// License: https://creativecommons.org/licenses/by-nc-sa/4.0/
|
||||||
|
//
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import "io"
|
||||||
|
|
||||||
|
// counter counts the total number of bytes read.
|
||||||
|
type counter struct {
|
||||||
|
r io.Reader
|
||||||
|
n int
|
||||||
|
}
|
||||||
|
|
||||||
|
// newCounter wraps `r` and returns a *counter.
|
||||||
|
func newCounter(r io.Reader) *counter {
|
||||||
|
return &counter{r: r}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Total bytes read.
|
||||||
|
func (c *counter) total() int {
|
||||||
|
return c.n
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read counts the number of bytes read from the underlying reader.
|
||||||
|
func (c *counter) Read(p []byte) (n int, err error) {
|
||||||
|
n, err = c.r.Read(p) // read from the underlying reader
|
||||||
|
if n > 0 {
|
||||||
|
c.n += n // sum the number of bytes read with the total bytes counted
|
||||||
|
}
|
||||||
|
return n, err // return the number of bytes read and an error
|
||||||
|
}
|
112
interfaces/15-io-reader-composition/database.go
Normal file
112
interfaces/15-io-reader-composition/database.go
Normal file
@ -0,0 +1,112 @@
|
|||||||
|
// For more tutorials: https://blog.learngoprogramming.com
|
||||||
|
//
|
||||||
|
// Copyright © 2018 Inanc Gumus
|
||||||
|
// Learn Go Programming Course
|
||||||
|
// License: https://creativecommons.org/licenses/by-nc-sa/4.0/
|
||||||
|
//
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"reflect"
|
||||||
|
)
|
||||||
|
|
||||||
|
type database struct {
|
||||||
|
list *list
|
||||||
|
types map[string]item
|
||||||
|
}
|
||||||
|
|
||||||
|
// load the list by decoding the data from any kind of data source.
|
||||||
|
func (db *database) load(r io.Reader) error {
|
||||||
|
return json.NewDecoder(r).Decode(db)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (db *database) MarshalJSON() ([]byte, error) {
|
||||||
|
type encodable struct {
|
||||||
|
Type string
|
||||||
|
Item item
|
||||||
|
}
|
||||||
|
|
||||||
|
var e []encodable
|
||||||
|
|
||||||
|
for _, it := range *db.list {
|
||||||
|
t := reflect.TypeOf(it).Elem().Name()
|
||||||
|
e = append(e, encodable{t, it})
|
||||||
|
}
|
||||||
|
|
||||||
|
return json.Marshal(e)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (db *database) UnmarshalJSON(data []byte) error {
|
||||||
|
var decodables []struct {
|
||||||
|
Type string
|
||||||
|
Item json.RawMessage
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := json.Unmarshal(data, &decodables); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, d := range decodables {
|
||||||
|
it, err := db.newItem(d.Item, d.Type)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
*db.list = append(*db.list, it)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (db *database) newItem(data []byte, typ string) (item, error) {
|
||||||
|
it, ok := db.types[typ]
|
||||||
|
if !ok {
|
||||||
|
return nil, fmt.Errorf("newItem: type (%q) does not exist", typ)
|
||||||
|
}
|
||||||
|
|
||||||
|
t := reflect.TypeOf(it)
|
||||||
|
e := t.Elem()
|
||||||
|
v := reflect.New(e)
|
||||||
|
it = v.Interface().(item)
|
||||||
|
|
||||||
|
return it, json.Unmarshal(data, &it)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (db *database) register(typ string, it item) {
|
||||||
|
if db.types == nil {
|
||||||
|
db.types = make(map[string]item)
|
||||||
|
}
|
||||||
|
db.types[typ] = it
|
||||||
|
}
|
||||||
|
|
||||||
|
// I put it here as a reference implementation of manually reading from a reader.
|
||||||
|
// func (db *database) loadX(r io.Reader) error {
|
||||||
|
// buf := make([]byte, 64)
|
||||||
|
//
|
||||||
|
// var (
|
||||||
|
// data []byte
|
||||||
|
// n int
|
||||||
|
// err error
|
||||||
|
// )
|
||||||
|
//
|
||||||
|
// for {
|
||||||
|
// n, err = r.Read(buf)
|
||||||
|
//
|
||||||
|
// if n > 0 {
|
||||||
|
// data = append(data, buf[:n]...)
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// if err != nil {
|
||||||
|
// break
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// if err != io.EOF {
|
||||||
|
// return err
|
||||||
|
// }
|
||||||
|
// return json.Unmarshal(data, db)
|
||||||
|
// }
|
54
interfaces/15-io-reader-composition/database.json
Normal file
54
interfaces/15-io-reader-composition/database.json
Normal file
@ -0,0 +1,54 @@
|
|||||||
|
[
|
||||||
|
{
|
||||||
|
"Type": "book",
|
||||||
|
"Item": {
|
||||||
|
"Title": "moby dick",
|
||||||
|
"Price": 10,
|
||||||
|
"Published": 118281600
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"Type": "book",
|
||||||
|
"Item": {
|
||||||
|
"Title": "odyssey",
|
||||||
|
"Price": 15,
|
||||||
|
"Published": 733622400
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"Type": "book",
|
||||||
|
"Item": {
|
||||||
|
"Title": "hobbit",
|
||||||
|
"Price": 25,
|
||||||
|
"Published": -62135596800
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"Type": "puzzle",
|
||||||
|
"Item": {
|
||||||
|
"Title": "rubik's cube",
|
||||||
|
"Price": 5
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"Type": "game",
|
||||||
|
"Item": {
|
||||||
|
"Title": "minecraft",
|
||||||
|
"Price": 20
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"Type": "game",
|
||||||
|
"Item": {
|
||||||
|
"Title": "tetris",
|
||||||
|
"Price": 5
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"Type": "toy",
|
||||||
|
"Item": {
|
||||||
|
"Title": "yoda",
|
||||||
|
"Price": 150
|
||||||
|
}
|
||||||
|
}
|
||||||
|
]
|
BIN
interfaces/15-io-reader-composition/database.json.gz
Normal file
BIN
interfaces/15-io-reader-composition/database.json.gz
Normal file
Binary file not shown.
12
interfaces/15-io-reader-composition/game.go
Normal file
12
interfaces/15-io-reader-composition/game.go
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
// For more tutorials: https://blog.learngoprogramming.com
|
||||||
|
//
|
||||||
|
// Copyright © 2018 Inanc Gumus
|
||||||
|
// Learn Go Programming Course
|
||||||
|
// License: https://creativecommons.org/licenses/by-nc-sa/4.0/
|
||||||
|
//
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
type game struct {
|
||||||
|
product
|
||||||
|
}
|
40
interfaces/15-io-reader-composition/list.go
Normal file
40
interfaces/15-io-reader-composition/list.go
Normal file
@ -0,0 +1,40 @@
|
|||||||
|
// For more tutorials: https://blog.learngoprogramming.com
|
||||||
|
//
|
||||||
|
// Copyright © 2018 Inanc Gumus
|
||||||
|
// Learn Go Programming Course
|
||||||
|
// License: https://creativecommons.org/licenses/by-nc-sa/4.0/
|
||||||
|
//
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type item interface {
|
||||||
|
discount(float64)
|
||||||
|
fmt.Stringer
|
||||||
|
}
|
||||||
|
|
||||||
|
type list []item
|
||||||
|
|
||||||
|
func (l list) String() string {
|
||||||
|
if len(l) == 0 {
|
||||||
|
return "Sorry. We're waiting for delivery 🚚."
|
||||||
|
}
|
||||||
|
|
||||||
|
var str strings.Builder
|
||||||
|
for _, it := range l {
|
||||||
|
str.WriteString(it.String())
|
||||||
|
str.WriteRune('\n')
|
||||||
|
}
|
||||||
|
|
||||||
|
return str.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l list) discount(ratio float64) {
|
||||||
|
for _, it := range l {
|
||||||
|
it.discount(ratio)
|
||||||
|
}
|
||||||
|
}
|
97
interfaces/15-io-reader-composition/main.go
Normal file
97
interfaces/15-io-reader-composition/main.go
Normal file
@ -0,0 +1,97 @@
|
|||||||
|
// For more tutorials: https://blog.learngoprogramming.com
|
||||||
|
//
|
||||||
|
// Copyright © 2018 Inanc Gumus
|
||||||
|
// Learn Go Programming Course
|
||||||
|
// License: https://creativecommons.org/licenses/by-nc-sa/4.0/
|
||||||
|
//
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"compress/gzip"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
/*
|
||||||
|
store := list{
|
||||||
|
&book{product{"moby dick", 10}, toTimestamp(118281600)},
|
||||||
|
&book{product{"odyssey", 15}, toTimestamp("733622400")},
|
||||||
|
&book{product{"hobbit", 25}, unknown},
|
||||||
|
&puzzle{product{"rubik's cube", 5}},
|
||||||
|
&game{product{"minecraft", 20}},
|
||||||
|
&game{product{"tetris", 5}},
|
||||||
|
&toy{product{"yoda", 150}},
|
||||||
|
}
|
||||||
|
|
||||||
|
db := database{list: &store}
|
||||||
|
|
||||||
|
out, err := json.MarshalIndent(&db, "", "\t")
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalln(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Println(string(out))
|
||||||
|
|
||||||
|
// store.discount(.5)
|
||||||
|
// fmt.Print(store)
|
||||||
|
*/
|
||||||
|
|
||||||
|
var store list
|
||||||
|
|
||||||
|
db := database{list: &store}
|
||||||
|
db.register("book", new(book))
|
||||||
|
db.register("book", new(book))
|
||||||
|
db.register("game", new(game))
|
||||||
|
db.register("puzzle", new(puzzle))
|
||||||
|
db.register("toy", new(toy))
|
||||||
|
|
||||||
|
// load from a file
|
||||||
|
// f, _ := os.Open("database.json")
|
||||||
|
// db.load(f)
|
||||||
|
// f.Close()
|
||||||
|
|
||||||
|
// load from a string
|
||||||
|
// const data = `[
|
||||||
|
// { "Type": "book", "Item": { "Title": "1984", "Price": 8, "Published": -649641600 } },
|
||||||
|
// { "Type": "game", "Item": { "Title": "paperboy", "Price": 20 } }]`
|
||||||
|
// r := strings.NewReader(data)
|
||||||
|
// db.load(r)
|
||||||
|
|
||||||
|
// load from a web server
|
||||||
|
// res, _ := http.Get("https://inancgumus.github.io/x/database.json")
|
||||||
|
// db.load(res.Body)
|
||||||
|
// res.Body.Close()
|
||||||
|
|
||||||
|
// decompress the reader while reading from it
|
||||||
|
// res, _ := http.Get("https://inancgumus.github.io/x/database.json.gz")
|
||||||
|
// gr, _ := gzip.NewReader(res.Body)
|
||||||
|
// db.load(gr)
|
||||||
|
// gr.Close()
|
||||||
|
// res.Body.Close()
|
||||||
|
|
||||||
|
// decompress and count the number of compressed bytes read
|
||||||
|
// res, _ := http.Get("https://inancgumus.github.io/x/database.json.gz")
|
||||||
|
// c := newCounter(res.Body) // count the bytes read
|
||||||
|
// gr, _ := gzip.NewReader(c)
|
||||||
|
|
||||||
|
// db.load(gr)
|
||||||
|
// fmt.Printf("%d bytes read.\n\n", c.total())
|
||||||
|
|
||||||
|
// gr.Close()
|
||||||
|
// res.Body.Close()
|
||||||
|
|
||||||
|
// count the number of compressed bytes read from the web server then decompress
|
||||||
|
res, _ := http.Get("https://inancgumus.github.io/x/database.json.gz")
|
||||||
|
gr, _ := gzip.NewReader(res.Body)
|
||||||
|
c := newCounter(gr)
|
||||||
|
|
||||||
|
db.load(c)
|
||||||
|
fmt.Printf("%d bytes read.\n\n", c.total())
|
||||||
|
|
||||||
|
gr.Close()
|
||||||
|
res.Body.Close()
|
||||||
|
|
||||||
|
fmt.Print(store)
|
||||||
|
}
|
16
interfaces/15-io-reader-composition/money.go
Normal file
16
interfaces/15-io-reader-composition/money.go
Normal file
@ -0,0 +1,16 @@
|
|||||||
|
// For more tutorials: https://blog.learngoprogramming.com
|
||||||
|
//
|
||||||
|
// Copyright © 2018 Inanc Gumus
|
||||||
|
// Learn Go Programming Course
|
||||||
|
// License: https://creativecommons.org/licenses/by-nc-sa/4.0/
|
||||||
|
//
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
type money float64
|
||||||
|
|
||||||
|
func (m money) String() string {
|
||||||
|
return fmt.Sprintf("$%.2f", m)
|
||||||
|
}
|
23
interfaces/15-io-reader-composition/product.go
Normal file
23
interfaces/15-io-reader-composition/product.go
Normal file
@ -0,0 +1,23 @@
|
|||||||
|
// For more tutorials: https://blog.learngoprogramming.com
|
||||||
|
//
|
||||||
|
// Copyright © 2018 Inanc Gumus
|
||||||
|
// Learn Go Programming Course
|
||||||
|
// License: https://creativecommons.org/licenses/by-nc-sa/4.0/
|
||||||
|
//
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
type product struct {
|
||||||
|
Title string
|
||||||
|
Price money
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *product) String() string {
|
||||||
|
return fmt.Sprintf("%-15s: %s", p.Title, p.Price)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *product) discount(ratio float64) {
|
||||||
|
p.Price *= money(1 - ratio)
|
||||||
|
}
|
12
interfaces/15-io-reader-composition/puzzle.go
Normal file
12
interfaces/15-io-reader-composition/puzzle.go
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
// For more tutorials: https://blog.learngoprogramming.com
|
||||||
|
//
|
||||||
|
// Copyright © 2018 Inanc Gumus
|
||||||
|
// Learn Go Programming Course
|
||||||
|
// License: https://creativecommons.org/licenses/by-nc-sa/4.0/
|
||||||
|
//
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
type puzzle struct {
|
||||||
|
product
|
||||||
|
}
|
77
interfaces/15-io-reader-composition/timestamp.go
Normal file
77
interfaces/15-io-reader-composition/timestamp.go
Normal file
@ -0,0 +1,77 @@
|
|||||||
|
// For more tutorials: https://blog.learngoprogramming.com
|
||||||
|
//
|
||||||
|
// Copyright © 2018 Inanc Gumus
|
||||||
|
// Learn Go Programming Course
|
||||||
|
// License: https://creativecommons.org/licenses/by-nc-sa/4.0/
|
||||||
|
//
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strconv"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Mon Jan 2 15:04:05 -0700 MST 2006
|
||||||
|
const layout = "2006/01"
|
||||||
|
|
||||||
|
// unknown is the zero value of a timestamp.
|
||||||
|
var unknown = timestamp(time.Time{})
|
||||||
|
|
||||||
|
// timestamp prints timestamps, it's a stringer.
|
||||||
|
// It is useful even if it's zero.
|
||||||
|
type timestamp time.Time
|
||||||
|
|
||||||
|
// MarshalJSON from timestamp.
|
||||||
|
func (ts timestamp) MarshalJSON() (out []byte, err error) {
|
||||||
|
u := time.Time(ts).Unix()
|
||||||
|
return strconv.AppendInt(out, u, 10), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// UnmarshalJSON data to timestamp.
|
||||||
|
func (ts *timestamp) UnmarshalJSON(data []byte) error {
|
||||||
|
s := string(data)
|
||||||
|
|
||||||
|
// Let the ParseInt parse quoted strings.
|
||||||
|
us, err := strconv.Unquote(s)
|
||||||
|
if err == nil {
|
||||||
|
s = us
|
||||||
|
}
|
||||||
|
|
||||||
|
// Always overwrite the timestamp when decoding.
|
||||||
|
*ts = unknown
|
||||||
|
|
||||||
|
// Handle the numeric case.
|
||||||
|
if n, err := strconv.ParseInt(s, 10, 64); err == nil {
|
||||||
|
*ts = timestamp(time.Unix(n, 0))
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// String representation of the timestamp.
|
||||||
|
func (ts timestamp) String() string {
|
||||||
|
t := time.Time(ts)
|
||||||
|
|
||||||
|
if t.IsZero() {
|
||||||
|
return "unknown"
|
||||||
|
}
|
||||||
|
|
||||||
|
return t.Format(layout)
|
||||||
|
}
|
||||||
|
|
||||||
|
// toTimestamp from an unknown value.
|
||||||
|
func toTimestamp(v interface{}) timestamp {
|
||||||
|
var t int
|
||||||
|
|
||||||
|
switch v := v.(type) {
|
||||||
|
case int:
|
||||||
|
t = v
|
||||||
|
case string:
|
||||||
|
t, _ = strconv.Atoi(v)
|
||||||
|
default:
|
||||||
|
return unknown
|
||||||
|
}
|
||||||
|
|
||||||
|
return timestamp(time.Unix(int64(t), 0))
|
||||||
|
}
|
12
interfaces/15-io-reader-composition/toy.go
Normal file
12
interfaces/15-io-reader-composition/toy.go
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
// For more tutorials: https://blog.learngoprogramming.com
|
||||||
|
//
|
||||||
|
// Copyright © 2018 Inanc Gumus
|
||||||
|
// Learn Go Programming Course
|
||||||
|
// License: https://creativecommons.org/licenses/by-nc-sa/4.0/
|
||||||
|
//
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
type toy struct {
|
||||||
|
product
|
||||||
|
}
|
Reference in New Issue
Block a user