diff options
Diffstat (limited to 'DAL/DB')
| -rw-r--r-- | DAL/DB/CRUD.go | 36 | ||||
| -rw-r--r-- | DAL/DB/CRUD_test.go | 25 |
2 files changed, 61 insertions, 0 deletions
diff --git a/DAL/DB/CRUD.go b/DAL/DB/CRUD.go new file mode 100644 index 0000000..a3b4a24 --- /dev/null +++ b/DAL/DB/CRUD.go @@ -0,0 +1,36 @@ +package db + +import ( + "reflect" + "strings" +) + +type DataTable[T any] struct { + TableName string +} + +// Forms a query to get all the fields from database +func (self *DataTable[T]) formSelectRequest() string { + var sb = strings.Builder{} + sb.WriteString("SELECT (") + var fields = self.getFields() + for i, v := range fields { + if i != 0 { + sb.WriteString(", ") + } + sb.WriteString(v) + } + sb.WriteString(") FROM ") + sb.WriteString(self.TableName) + sb.WriteString(";") + return sb.String() +} + +func (self *DataTable[T]) getFields() []string { + var zero T + var outp = make([]string, 0) + for _, f := range reflect.VisibleFields(reflect.TypeOf(zero)) { + outp = append(outp, f.Name) + } + return outp +} diff --git a/DAL/DB/CRUD_test.go b/DAL/DB/CRUD_test.go new file mode 100644 index 0000000..de6fd40 --- /dev/null +++ b/DAL/DB/CRUD_test.go @@ -0,0 +1,25 @@ +package db + +import ( + "testing" + "time" +) + +type testDataType struct { + Id uint64 + Name string + Data time.Time +} + +var testDataTable = DataTable[testDataType] { + TableName: "DataTable", +} + +func TestFormSelectRequest(t *testing.T) { + var expected = "SELECT (Id, Name, Data) FROM DataTable;" + var request = testDataTable.formSelectRequest() + if request != expected { + t.Errorf("Incorrect select query. \n Expected: '%s' \n Got: '%s'", expected, request) + } +} + |
