Rewrite cache handling and JSON marshalling concept
Go / build (push) Failing after 30s Details
Go / lint (push) Failing after 33s Details

This commit is contained in:
Marcel M. Otte 2023-11-03 11:43:54 +01:00
parent 21c43afa18
commit e318067d90
5 changed files with 277 additions and 203 deletions

View File

@ -2,37 +2,88 @@ package cop
import (
"encoding/json"
"errors"
"fmt"
"io/fs"
"log"
"os"
"path"
"path/filepath"
"strings"
"github.com/go-git/go-git/v5"
"gitea.mmo.to/ppForge/ppforge/globals"
"gitea.mmo.to/ppForge/ppforge/protocol"
)
// GlobalCOP is the global variable holding the protocol collections,
var GlobalCOP *ProtocolCollectionList = &ProtocolCollectionList{[]COP{}}
var GlobalCOP *ProtocolCollectionList = &ProtocolCollectionList{[]COPer{}}
// CollectionTypes contains the functions to create actual objects from JSON
var CollectionTypes map[string](func([]byte) (COPer, error))
// InitCollectionTypes initializes the functions for each actual object
func InitCollectionTypes() {
CollectionTypes = map[string]func([]byte) (COPer, error){}
//CollectionTypes["COP"] = COPFromJSON
CollectionTypes["FileCOP"] = FileCOPFromJSON
//CollectionTypes["FileCOPFromGitRemote"] = FileCOPGitRemoteFromJSON
}
// ProtocolCollectionList is a list of all databases, with the most simple interface
type ProtocolCollectionList struct {
PCs []COP
PCs []COPer
}
// COPer interface defines the functions a CollectionOfProtocols should have at least.
type COPer interface {
Open(string) error
Close() error
GetProtocols() ([]ProtocolCollectionEntry, error)
AddOrUpdate(*protocol.Protocol) error
Sync() error
Get(string) (*protocol.Protocol, error)
ToJSON() (string, error)
}
// Init initializes the databases, opens the default one and checks for others if available
func init() {
InitCollectionTypes()
Init()
}
// Init the global COP
func Init() {
// initialize default db
fdb := FileCOP{COP{"FileCOP", map[string]ProtocolCollectionEntry{}, false}, globals.CollectionOfProtocolsDir}
err := fdb.Open(globals.CollectionOfProtocolsDir)
if err != nil {
log.Printf("Error opening global COP dir: %v", err)
return
}
err = fdb.Sync()
if err != nil {
log.Printf("Error on initial syncing of the global COP dir: %v", err)
return
}
GlobalCOP.PCs = append(GlobalCOP.PCs, &fdb)
log.Printf("Amount of databases available %d", len(GlobalCOP.PCs))
}
// GetCOP returns the global collection of Protocols
func GetCOP() *ProtocolCollectionList {
return GlobalCOP
}
// WriteCache updates the cache file of the available protocols
func (pcl *ProtocolCollectionList) WriteCache() error {
// [impl->dsn~protocol-collection-cache~0>>utest]
data, err := json.MarshalIndent(pcl, "", " ")
cops := []string{}
for _, pc := range pcl.PCs {
str, err := pc.ToJSON()
if err != nil {
return err
log.Println(err)
}
err = os.WriteFile(path.Join(globals.CollectionOfProtocolsDir, globals.COPCacheFileName), data, 0644)
cops = append(cops, str)
}
data := "{ \"PCs\": [" + strings.Join(cops, ",") + "]}"
err := os.WriteFile(path.Join(globals.CollectionOfProtocolsDir, globals.COPCacheFileName), []byte(data), 0644)
return err
}
@ -46,183 +97,20 @@ func (pcl *ProtocolCollectionList) ReadCache() error {
return err
}
//fmt.Println(string(data))
err = json.Unmarshal(data, pcl)
data, err = json.MarshalIndent(pcl, "", " ")
var raw map[string][]interface{}
err = json.Unmarshal(data, &raw)
PCs := raw
for _, rawcop := range PCs["PCs"] {
cop := rawcop.(map[string]interface{})
coptype := cop["Type"].(string)
copjson, err := json.Marshal(rawcop)
fmt.Println(string(copjson))
if err != nil {
log.Println(err)
}
coper, err := CollectionTypes[coptype](copjson)
pcl.PCs = append(pcl.PCs, coper)
}
fmt.Println(string(data))
return err
}
// ProtocolCollectionEntry is a single entry in the file database, additionally to the meta data the path is required
type ProtocolCollectionEntry struct {
Path string
Protocol protocol.ProtocolMeta
}
// Init initializes the databases, opens the default one and checks for others if available
func init() {
Init()
}
// Init the global COP
func Init() {
// initialize default db
fdb := FileCOP{COP{map[string]ProtocolCollectionEntry{}, false}, globals.CollectionOfProtocolsDir}
err := fdb.Open(globals.CollectionOfProtocolsDir)
if err != nil {
log.Printf("Error opening global COP dir: %v", err)
return
}
err = fdb.Sync()
if err != nil {
log.Printf("Error on initial syncing of the global COP dir: %v", err)
return
}
GlobalCOP.PCs = append(GlobalCOP.PCs, fdb.COP)
log.Printf("Amount of databases available %d", len(GlobalCOP.PCs))
}
// GetCOP returns the global collection of Protocols
func GetCOP() *ProtocolCollectionList {
return GlobalCOP
}
// COP represents a Collection of Protocols
type COP struct {
Protocols map[string]ProtocolCollectionEntry
closed bool
}
// COPer interface defines the functions a CollectionOfProtocols should have at least.
type COPer interface {
Open(string) error
Close() error
GetProtocols() ([]ProtocolCollectionEntry, error)
Add(*protocol.Protocol) error
Sync() error
Update(*protocol.Protocol) error
Get(string) (*protocol.Protocol, error)
}
// FileCOP implements
type FileCOP struct {
COP
Path string
}
// FileCOPFromGitRemote implements retrieving the latest stuff from an external git remote additionally to the FileDatabase functionality
type FileCOPFromGitRemote struct {
FileCOP
RemoteGitRepository string
gitRepo git.Repository
}
// NewFileCOP returns a new FileCOP instance from a given path
func NewFileCOP(path string) (*FileCOP, error) {
fCOP := &FileCOP{}
err := fCOP.Open(path)
if err != nil {
log.Printf("Error opening file COP at %s: %v", path, err)
return nil, err
}
err = fCOP.Sync()
return fCOP, err
}
// Open the database
func (fd *FileCOP) Open(path string) error {
fileinfo, err := os.Stat(path)
if err == os.ErrNotExist {
return err
}
if fileinfo == nil || !fileinfo.IsDir() {
return fmt.Errorf("Path %s is not a directory or does not exist", path)
}
fd.Path = path
fd.closed = false
return nil
}
// Close the database and free the data
func (fd *FileCOP) Close() error {
fd.closed = true
return nil
}
// GetProtocols returns the map of protocols or an error if the first read failed
func (fd *FileCOP) GetProtocols() ([]ProtocolCollectionEntry, error) {
if fd.closed {
return nil, errors.New("COP closed, no protocols to deliver")
}
entries := []ProtocolCollectionEntry{}
if !fd.closed && len(fd.Protocols) == 0 {
fd.Protocols = map[string]ProtocolCollectionEntry{}
err := fd.Sync()
if err != nil {
return nil, err
}
}
if entries == nil || len(entries) == 0 {
entries = make([]ProtocolCollectionEntry, 0)
for _, v := range fd.Protocols {
entries = append(entries, v)
}
}
return entries, nil
}
// Add puts a new protocol into the path of the file database
func (fd *FileCOP) Add(prot *protocol.Protocol) error {
if fd.closed {
return errors.New("Cannot add, FileCOP is closed")
}
p := path.Join(fd.Path, prot.Metadata.Name, ".protocoljson")
err := prot.Save(p)
if err == nil {
fd.Protocols[prot.Metadata.Name] = ProtocolCollectionEntry{p, prot.Metadata}
}
return err
}
// Update updates an existing entry through saving the contents of the protocol to the file, or just adding it anew
func (fd *FileCOP) Update(prot *protocol.Protocol) error {
if fd.closed {
return errors.New("Cannot update, FileCOP is closed")
}
if v, ok := fd.Protocols[prot.Metadata.Name]; ok {
v.Protocol = prot.Metadata
return prot.Save(v.Path)
}
return fd.Add(prot)
}
// Sync just rereads all files
func (fd *FileCOP) Sync() error {
if fd.closed {
return errors.New("Cannot sync, FileCOP is closed")
}
// recurse recursively through the path
if fd.Protocols == nil {
fd.Protocols = make(map[string]ProtocolCollectionEntry)
}
err := filepath.Walk(fd.Path, func(path string, info fs.FileInfo, err error) error {
if !info.IsDir() && strings.HasSuffix(info.Name(), ".protocoljson") {
prot, err := protocol.LoadNew(path)
if err == nil {
// add to map
// add to map
entry := ProtocolCollectionEntry{path, prot.Metadata}
fd.Protocols[prot.Metadata.Name] = entry
}
return err
}
return nil
})
return err
}
// Get a protocol by name
func (fd *FileCOP) Get(prot string) (*protocol.Protocol, error) {
entry := fd.Protocols[prot]
return protocol.LoadNew(entry.Path)
}

View File

@ -12,6 +12,7 @@ import (
"gitea.mmo.to/ppForge/ppforge/cop"
"gitea.mmo.to/ppForge/ppforge/globals"
"gitea.mmo.to/ppForge/ppforge/protocol"
)
func getCurrentDir() string {
@ -94,14 +95,9 @@ func TestClosed(t *testing.T) {
fmt.Println("Protocols did not deliver error")
t.Fail()
}
err = fcop.Add(nil)
err = fcop.AddOrUpdate(nil)
if err == nil {
fmt.Println("Add did not deliver error")
t.Fail()
}
err = fcop.Update(nil)
if err == nil {
fmt.Println("Update did not deliver error")
fmt.Println("AddOrUpdate did not deliver error")
t.Fail()
}
err = fcop.Sync()
@ -128,7 +124,7 @@ func TestGlobalCOP(t *testing.T) {
fmt.Println("Error on loading new FileCOP", err)
t.Fail()
}
gcop.PCs = append(gcop.PCs, fcop.COP)
gcop.PCs = append(gcop.PCs, fcop)
fmt.Printf("Length of all Protocol Collections: %d\n", len(gcop.PCs))
if len(gcop.PCs) != 2 {
@ -142,15 +138,15 @@ func TestCOPCache(t *testing.T) {
defer td(t)
gcop := cop.GlobalCOP
// reset to zero because of other tests
gcop.PCs = []cop.COP{}
gcop.PCs = []cop.COPer{}
cop.Init()
c := gcop.PCs[0]
protos := c.Protocols
protos, err := c.GetProtocols()
if len(protos) != 4 {
t.Fail()
}
err := gcop.WriteCache()
err = gcop.WriteCache()
if err != nil {
fmt.Println("Writing the cache file failed")
t.Fail()
@ -162,17 +158,44 @@ func TestCOPCache(t *testing.T) {
}
// reset again and check if reading the cache works
gcop.PCs = []cop.COP{}
gcop.PCs = []cop.COPer{}
err = gcop.ReadCache()
if err != nil {
fmt.Println("Reading the cache file failed")
fmt.Println("Reading the cache file failed", err)
t.Fail()
}
c = gcop.PCs[0]
protos = c.Protocols
protos, err = c.GetProtocols()
if len(protos) != 4 {
t.Fail()
}
}
func TestAddOrUpdate(t *testing.T) {
td := setupSuite(t)
defer td(t)
gcop := cop.GlobalCOP
gcop.PCs = []cop.COPer{}
cop.Init()
prot := protocol.NewProtocolStructure()
prot.Metadata.Name = "TestAddOrUpdate"
prot.Metadata.Description = "A protocol of the unit tests to test AddOrUpdate of a protocol collection"
prot.Metadata.Version = "1.0"
prot.Metadata.Revision = 0
prot.Metadata.TCPIPLayer = 1
f := protocol.NewField(
"Testfield",
"A field for the unit tests of protocol collections...",
"",
32,
nil,
false,
false,
)
prot.AddField(0, f)
col := gcop.PCs[0]
col.AddOrUpdate(prot)
}

16
cop/cop.go Normal file
View File

@ -0,0 +1,16 @@
package cop
import "gitea.mmo.to/ppForge/ppforge/protocol"
// COP represents a Collection of Protocols
type COP struct {
Type string
Protocols map[string]ProtocolCollectionEntry
closed bool
}
// ProtocolCollectionEntry is a single entry in the file database, additionally to the meta data the path is required
type ProtocolCollectionEntry struct {
Path string
Protocol protocol.ProtocolMeta
}

137
cop/filecop.go Normal file
View File

@ -0,0 +1,137 @@
package cop
import (
"encoding/json"
"errors"
"fmt"
"io/fs"
"log"
"os"
"path"
"path/filepath"
"strings"
"gitea.mmo.to/ppForge/ppforge/protocol"
)
// FileCOP implements
type FileCOP struct {
COP
Path string
}
// FileCOPFromJSON creates a FileCOP
func FileCOPFromJSON(data []byte) (COPer, error) {
fcop := &FileCOP{}
err := json.Unmarshal(data, fcop)
return fcop, err
}
// NewFileCOP returns a new FileCOP instance from a given path
func NewFileCOP(path string) (*FileCOP, error) {
fCOP := &FileCOP{}
err := fCOP.Open(path)
if err != nil {
log.Printf("Error opening file COP at %s: %v", path, err)
return nil, err
}
err = fCOP.Sync()
return fCOP, err
}
// Open the database
func (fd *FileCOP) Open(path string) error {
fileinfo, err := os.Stat(path)
if err == os.ErrNotExist {
return err
}
if fileinfo == nil || !fileinfo.IsDir() {
return fmt.Errorf("Path %s is not a directory or does not exist", path)
}
fd.Path = path
fd.closed = false
return nil
}
// Close the database and free the data
func (fd *FileCOP) Close() error {
fd.closed = true
return nil
}
// GetProtocols returns the map of protocols or an error if the first read failed
func (fd *FileCOP) GetProtocols() ([]ProtocolCollectionEntry, error) {
if fd.closed {
return nil, errors.New("COP closed, no protocols to deliver")
}
entries := []ProtocolCollectionEntry{}
if !fd.closed && len(fd.Protocols) == 0 {
fd.Protocols = map[string]ProtocolCollectionEntry{}
err := fd.Sync()
if err != nil {
return nil, err
}
}
if entries == nil || len(entries) == 0 {
entries = make([]ProtocolCollectionEntry, 0)
for _, v := range fd.Protocols {
entries = append(entries, v)
}
}
return entries, nil
}
// AddOrUpdate puts a new protocol into the path of the file database
func (fd *FileCOP) AddOrUpdate(prot *protocol.Protocol) error {
if fd.closed {
return errors.New("Cannot add or update, FileCOP is closed")
}
if v, ok := fd.Protocols[prot.Metadata.Name]; ok {
v.Protocol = prot.Metadata
return prot.Save(v.Path)
}
p := path.Join(fd.Path, prot.Metadata.Name, ".protocoljson")
err := prot.Save(p)
if err == nil {
fd.Protocols[prot.Metadata.Name] = ProtocolCollectionEntry{p, prot.Metadata}
}
return err
}
// Sync just rereads all files
func (fd *FileCOP) Sync() error {
if fd.closed {
return errors.New("Cannot sync, FileCOP is closed")
}
// recurse recursively through the path
if fd.Protocols == nil {
fd.Protocols = make(map[string]ProtocolCollectionEntry)
}
err := filepath.Walk(fd.Path, func(path string, info fs.FileInfo, err error) error {
if !info.IsDir() && strings.HasSuffix(info.Name(), ".protocoljson") {
prot, err := protocol.LoadNew(path)
if err == nil {
// add to map
// add to map
entry := ProtocolCollectionEntry{path, prot.Metadata}
fd.Protocols[prot.Metadata.Name] = entry
}
return err
}
return nil
})
return err
}
// Get a protocol by name
func (fd *FileCOP) Get(prot string) (*protocol.Protocol, error) {
entry := fd.Protocols[prot]
return protocol.LoadNew(entry.Path)
}
// ToJSON marshals the FileCOP to JSON
func (fd *FileCOP) ToJSON() (string, error) {
by, err := json.MarshalIndent(fd, "", " ")
return string(by), err
}

10
cop/filecopgitremote.go Normal file
View File

@ -0,0 +1,10 @@
package cop
import "github.com/go-git/go-git/v5"
// FileCOPGitRemote implements retrieving the latest stuff from an external git remote additionally to the FileDatabase functionality
type FileCOPGitRemote struct {
FileCOP
RemoteGitRepository string
gitRepo git.Repository
}