parent
c0fd09be05
commit
212c86a784
@ -0,0 +1,48 @@ |
|||||||
|
package main |
||||||
|
|
||||||
|
import ( |
||||||
|
"bytes" |
||||||
|
"crypto/sha256" |
||||||
|
"encoding/gob" |
||||||
|
"log" |
||||||
|
"strconv" |
||||||
|
"strings" |
||||||
|
) |
||||||
|
|
||||||
|
// Transaction represents a Bitcoin transaction
|
||||||
|
type Transaction struct { |
||||||
|
ID []byte |
||||||
|
inputAddresss []string |
||||||
|
outputAddress []string |
||||||
|
value []int |
||||||
|
} |
||||||
|
|
||||||
|
func (tx *Transaction) Parse(data string) { |
||||||
|
items := strings.Split(data, ",") |
||||||
|
for _, value := range items { |
||||||
|
pair := strings.Split(value, " ") |
||||||
|
if len(pair) == 3 { |
||||||
|
intValue, err := strconv.Atoi(pair[2]) |
||||||
|
if err != nil { |
||||||
|
tx.inputAddress = append(tx.inputAddresss, strings.Trim(pair[0])) |
||||||
|
tx.outputAddress = append(tx.outputAddress, strings.Trim(pair[1])) |
||||||
|
tx.value = append(tx.value, intValue) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
return res |
||||||
|
} |
||||||
|
|
||||||
|
// SetID sets ID of a transaction
|
||||||
|
func (tx *Transaction) SetID() { |
||||||
|
var encoded bytes.Buffer |
||||||
|
var hash [32]byte |
||||||
|
|
||||||
|
enc := gob.NewEncoder(&encoded) |
||||||
|
err := enc.Encode(tx) |
||||||
|
if err != nil { |
||||||
|
log.Panic(err) |
||||||
|
} |
||||||
|
hash = sha256.Sum256(encoded.Bytes()) |
||||||
|
tx.ID = hash[:] |
||||||
|
} |
@ -0,0 +1,25 @@ |
|||||||
|
package main |
||||||
|
|
||||||
|
import "testing" |
||||||
|
|
||||||
|
func TestConvertIntoMap(t *testing.T) { |
||||||
|
data := "minh:3,mike:2" |
||||||
|
res := ConvertIntoMap(data) |
||||||
|
if len(res) != 2 { |
||||||
|
t.Errorf("Result should have 2 pairs (key, value)") |
||||||
|
} |
||||||
|
if val, ok := res["minh"]; !ok { |
||||||
|
t.Errorf("Result should contain key minh") |
||||||
|
} else { |
||||||
|
if res["minh"] != 3 { |
||||||
|
t.Errorf("Value of minh should be 3") |
||||||
|
} |
||||||
|
} |
||||||
|
if val, ok := res["mike"]; !ok { |
||||||
|
t.Errorf("Result should contain key mike") |
||||||
|
} else { |
||||||
|
if res["minh"] != 3 { |
||||||
|
t.Errorf("Value of minh should be 2") |
||||||
|
} |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue