autogenpb/example/main.go

//go:build go1.20
// +build go1.20

package main

import (
	"fmt"
	"math/rand"
	"os"

	"go.wit.com/log"
)

// sent via -ldflags
var VERSION string
var BUILDTIME string

var sortmap map[string]string
var marshalKeys []string
var uniqueKeys []string

var pb *Fruits

func main() {
	pb = NewFruits()
	// attempt to load basket.pb
	if err := pb.loadBasket(); err != nil {
		log.Info("load basket failed. running this for the first time?")
		fruit := &Fruit{
			Brand: "mom",
			City:  "New NewYork",
			UPC:   "2000",
		}
		testAppend(fruit)
		appendByUPC()
	}
	if err := pb.saveBasket(); err != nil {
		badExit(err)
	}
	if pb == nil {
		badExit(fmt.Errorf("pb == nil This should not have happened"))
	}
	pb.printTable()
	pb.addThings()
	pb.printTable()
	pb.saveBasket()
	pb.sortTable(100)
}

func badExit(err error) {
	log.Info("autogenpb error:", err)
	os.Exit(-1)
}

/*
	x := new(Fruit)
	x = &Fruit{
		Brand: "fry",
		City:  "Germany",
	}
	pb.printTable()

	testAppend(x)
*/

func (pb *Fruits) printTable() {
	var count int
	all := pb.All()
	for all.Scan() {
		tmp := all.Next()
		count += 1
		log.Printf("found %d %s %s\n", count, tmp.Brand, tmp.City)
		if count > 5 {
			break
		}
	}
	log.Printf("Total entries: %d\n", pb.Len())
}

func (pb *Fruits) sortTable(i int) {
	var count int
	all := pb.SortByBrand()
	for all.Scan() {
		tmp := all.Next()
		count += 1
		log.Printf("found %d %s %s\n", count, tmp.Brand, tmp.City)
		if count > i {
			break
		}
	}
	log.Printf("Total entries: %d\n", pb.Len())
}

// adds 10 new entries
func (pb *Fruits) addThings() {
	var added int
	var count int = rand.Intn(10000)
	// var base string = "bender"
	for {
		count += rand.Intn(1000)
		var found bool
		name := fmt.Sprintf("bender%d", count)
		all := pb.All()
		for all.Scan() {
			tmp := all.Next()
			if tmp.Brand == name {
				// log.Printf("tmp name EQUAL %d %s vs %s city = %s len=%d\n", count, tmp.Brand, name, tmp.City, pb.Len())
				found = true
				break
			}
			// log.Printf("DID NOT EXIST %d %s vs %s city = %s len=%d\n", count, tmp.Brand, name, tmp.City, pb.Len())
		}
		if found {
			continue
		}
		// log.Printf("DID NOT EXIST %d %s len=%d\n", count, name, pb.Len())
		tmp := new(Fruit)
		tmp.Brand = name
		tmp.City = "paris"
		if pb.AppendUniqueByBrand(tmp) {
			// log.Printf("AppendUniqueBrand() ok   len=%s %s %d\n", tmp.Brand, tmp.City, pb.Len())
		} else {
			badExit(fmt.Errorf("AppendUniqueByBrand() ERROR len=%s %s %d\n", tmp.Brand, tmp.City, pb.Len()))
		}
		pb.Append(tmp)
		added += 1
		if added > 10 {
			return
		}
	}
}

func testAppend(fruit *Fruit) {
	if pb.AppendUnique(fruit) {
		log.Info("AppendUnique() test1 ok", fruit.Brand, fruit.City)
	} else {
		badExit(fmt.Errorf("AppendUnique test1 failed %s %s %d\n", fruit.Brand, fruit.City, pb.Len()))
	}
	if pb.AppendUnique(fruit) {
		badExit(fmt.Errorf("AppendUnique() test2 worked but should not have %s %s", fruit.Brand, fruit.City))
	} else {
		log.Info("AppendUnique() test2 failed ok", fruit.Brand, fruit.City)
	}
}

func appendByUPC() {
	fruit := new(Fruit)
	fruit.UPC = "blah331"
	fruit.City = "fry"
	fruit.City = "paris"
	if pb.AppendUniqueByUPC(fruit) {
		log.Info("AppendUnique() test1 ok", fruit.Brand, fruit.City)
	} else {
		badExit(fmt.Errorf("AppendUniqueByUPC test1 failed %s %s %d\n", fruit.Brand, fruit.City, pb.Len()))
	}
	if pb.AppendUniqueByUPC(fruit) {
		badExit(fmt.Errorf("AppendUniqueByUPC() test2 worked but should not have %s %s", fruit.Brand, fruit.City))
	} else {
		log.Info("AppendUnique() test2 failed ok", fruit.Brand, fruit.City)
	}
}

func (pb *Fruits) saveBasket() error {
	data, err := pb.Marshal()
	if err != nil {
		return err
	}

	w, err := os.OpenFile("basket.pb", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
	if err != nil {
		return err
	}
	w.Write(data)
	w.Close()
	log.Info("saved basket.pb ok")
	return nil
}

func (pb *Fruits) loadBasket() error {
	if pb == nil {
		pb = new(Fruits)
	}
	data, err := os.ReadFile("basket.pb")
	if err != nil {
		return err
	}

	if err := pb.Unmarshal(data); err != nil {
		return err
	}
	log.Info("loaded basket.pb ok")
	return nil
}