Fixes to allow caching of the flat file for faster access.

IPQualityScore · Aug 23, 2024 · 6a516ff · 6a516ff
1 parent 51a8e76
commit 6a516ff
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 12 deletions.
diff --git a/FileReader.go b/FileReader.go
@@ -1,18 +1,18 @@
 package GoIPQSDBReader;
 
 import (
-	"os";
-	"net";
-	"fmt";
-	"math";
-	"strconv";
-	"errors";
-	"strings";
-	"encoding/binary";
+	"bytes"
+	"encoding/binary"
+	"errors"
+	"fmt"
+	"math"
+	"net"
+	"strconv"
+	"strings"
 );
 
 type FileReader struct {
-	Handler *os.File;
+	Handler bytes.Reader;
 	TotalBytes uint64;
 	RecordBytes uint64;
 

diff --git a/GoIPQSDBReader.go b/GoIPQSDBReader.go
@@ -13,10 +13,12 @@ func Open(filename string) (*FileReader, error) {
 	file := &FileReader{Columns: make(map[int]*Column)};
 
 	var ferr error;
-	file.Handler, ferr = os.Open(filename);
-	if(ferr != nil){
-		return file, ferr;
+	fileBytes, ferr := os.ReadFile(filename)
+	if ferr != nil {
+		return file, ferr
 	}
+
+	file.Handler = *bytes.NewReader(fileBytes)
 
 	header := make([]byte, 11);
 	bl, err := file.Handler.Read(header);