virtbuf/configfile/main.go

77 lines
1.5 KiB
Go

package main
import "log"
import "bytes"
import "os"
import "bufio"
import "io/ioutil"
import "google.golang.org/protobuf/proto"
import pb "go.wit.com/lib/protobuf/virtbuf"
//
// saves entries in a config file
//
func main() {
TestWriteDroplet()
in, err := ioutil.ReadFile("/tmp/testing4.protobuf")
if err != nil {
log.Fatalln("Error reading file:", err)
}
var aDroplet pb.Droplet
if err := proto.Unmarshal(in, &aDroplet); err != nil {
log.Fatalln("Failed to parse droplet:", err)
}
log.Println(aDroplet)
}
func marshalWriteToFile(myWriter *bufio.Writer, e *pb.Droplet) {
buf, err := proto.Marshal(e)
if err != nil {
log.Fatal("marshaling error: ", err)
}
tmp, err := myWriter.Write(buf)
myWriter.Flush()
log.Println("bufio.Write() tmp, err = ", tmp, err)
buf, err = proto.Marshal(e)
tmp2, err := myWriter.Write(buf)
myWriter.Flush()
log.Println("bufio.Write() tmp2, err = ", tmp2, err)
}
func TestWriteDroplet() {
buf := new(bytes.Buffer)
e := pb.CreateSampleDroplet()
got := buf.String()
log.Println(got)
newfile, _ := os.Create("/tmp/testing4.protobuf")
myWriter := bufio.NewWriter(newfile)
marshalWriteToFile(myWriter, e)
marshalUnmarshal()
}
func marshalUnmarshal() {
test := pb.CreateSampleDroplet()
data, err := proto.Marshal(test)
if err != nil {
log.Fatal("marshaling error: ", err)
}
newTest := &pb.Droplet{}
err = proto.Unmarshal(data, newTest)
if err != nil {
log.Fatal("unmarshaling error: ", err)
} else {
log.Println("proto.Marshal() and proto.Unmarshal() worked")
}
}