composite

Author: coolpest

Composite/composite.go

@@ -0,0 +1,7 @@
+package Composite
+
+type Item interface {
+	Add(Item)
+	String() string
+	Price() int
+}

Composite/composite_test.go

@@ -0,0 +1,35 @@
+package Composite
+
+import (
+	"fmt"
+	"testing"
+)
+
+func TestComposite(t *testing.T) {
+	p := Project{name: "Upload images"}
+	t1 := Task{name: "Mockup", responsable: "UI Designer", price: 1000}
+
+	t2 := Task{name: "Markup", responsable: "Web Designer"}
+	st21 := SubTask{name: "HTML", price: 300}
+	st22 := SubTask{name: "CSS", price: 700}
+	t2.Add(&st21)
+	t2.Add(&st22)
+
+	t3 := Task{name: "JS", responsable: "Frontend Developer", price: 1000}
+
+	t4 := Task{name: "API Backend", responsable: "Backend Developer"}
+	st41 := SubTask{name: "Authentication", price: 100}
+	st42 := SubTask{name: "DB connection", price: 900}
+	t4.Add(&st41)
+	t4.Add(&st42)
+
+	t5 := Task{name: "Database", responsable: "DBA", price: 1000}
+
+	p.Add(&t1)
+	p.Add(&t2)
+	p.Add(&t3)
+	p.Add(&t4)
+	p.Add(&t5)
+
+	fmt.Println(&p)
+}

Composite/project.go

@@ -0,0 +1,36 @@
+package Composite
+
+import (
+	"strconv"
+	"strings"
+)
+
+type Project struct {
+	name string
+	tasks []Item
+}
+
+func (p *Project) Add(i Item){
+	p.tasks = append(p.tasks, i)
+}
+
+func (p *Project) String() string {
+	sb := strings.Builder{}
+	sb.WriteString(p.name)
+	sb.WriteString(" $")
+	sb.WriteString(strconv.Itoa(p.Price()))
+	sb.WriteString("\n")
+	for _, v := range p.tasks {
+		sb.WriteString(v.String())
+	}
+
+	return sb.String()
+}
+
+func (p *Project) Price() int {
+	price :=0
+	for _,v:=range p.tasks{
+		price+=v.Price()
+	}
+	return price
+}

Composite/subtask.go

@@ -0,0 +1,28 @@
+package Composite
+
+import (
+	"fmt"
+	"strings"
+)
+
+type SubTask struct {
+	name string
+	price int
+}
+
+func (s *SubTask) Add(i Item)  {
+	fmt.Println("You don't accept subtrareas")
+}
+
+func (s *SubTask) String() string{
+	sb := strings.Builder{}
+	sb.WriteString("\t\t|-- ")
+	sb.WriteString(s.name)
+	sb.WriteString("\n")
+
+	return sb.String()
+}
+
+func (s *SubTask) Price() int {
+	return s.price
+}

Composite/task.go

@@ -0,0 +1,40 @@
+package Composite
+
+import (
+	"strconv"
+	"strings"
+)
+
+type Task struct {
+	name string
+	responsable string
+	price int
+	subTasks []Item
+}
+
+func (t *Task) Add(i Item){
+	t.subTasks = append(t.subTasks,i)
+}
+
+func (t *Task) String() string  {
+	sb:=strings.Builder{}
+	sb.WriteString("\t|--")
+	sb.WriteString(t.name)
+	sb.WriteString(" - ")
+	sb.WriteString(t.responsable)
+	sb.WriteString(" $")
+	sb.WriteString(strconv.Itoa(t.Price()))
+	sb.WriteString("\n")
+	for _,v := range t.subTasks{
+		sb.WriteString(v.String())
+	}
+	return sb.String()
+}
+
+func (t *Task) Price() int {
+	price:= t.price
+	for _,v:=range t.subTasks{
+		price +=v.Price()
+	}
+	return price
+}