Add dbo.Set

dbo/set.go

@@ -0,0 +1,44 @@
+package dbo
+
+import (
+	"database/sql/driver"
+	"encoding/json"
+	"fmt"
+
+	"github.com/Cyberpull/gokit"
+)
+
+type Set[T gokit.SetConstraint] struct {
+	Data []T
+}
+
+func (x *Set[T]) Scan(value any) (err error) {
+	switch data := value.(type) {
+	case []T:
+		x.Data = data
+
+	default:
+		x.Data, err = gokit.Split[T](fmt.Sprint(value), ",")
+	}
+
+	return
+}
+
+func (x Set[T]) Value() (value driver.Value, err error) {
+	value = gokit.Join(",", x.Data...)
+	return
+}
+
+func (x *Set[T]) UnmarshalJSON(b []byte) error {
+	var value T
+
+	if err := json.Unmarshal(b, &value); err != nil {
+		return err
+	}
+
+	return x.Scan(value)
+}
+
+func (x Set[T]) MarshalJSON() ([]byte, error) {
+	return json.Marshal(x.Data)
+}

set.go

@@ -27,7 +27,7 @@ func In[T comparable](value T, entries ...T) bool {
 	return false
 }
 
-func JoinFunc(delim string, entries []any, callbacks ...JoinEntryFunc) string {
+func JoinFunc[T any](delim string, entries []T, callbacks ...JoinEntryFunc) string {
 	var buff bytes.Buffer
 
 	for _, entry := range entries {
@@ -56,7 +56,7 @@ func JoinFunc(delim string, entries []any, callbacks ...JoinEntryFunc) string {
 	return buff.String()
 }
 
-func Join(delim string, entries ...any) string {
+func Join[T any](delim string, entries ...T) string {
 	return JoinFunc(delim, entries)
 }
 

tests/dbo_test.go

@@ -0,0 +1,31 @@
+package tests
+
+import (
+	"testing"
+
+	"github.com/Cyberpull/gokit/dbo"
+	"github.com/stretchr/testify/require"
+	"github.com/stretchr/testify/suite"
+)
+
+type DBOTestSuite struct {
+	suite.Suite
+}
+
+func (x *DBOTestSuite) TestSet() {
+	var data dbo.Set[string]
+
+	err := data.Scan("a,b,c,d,e")
+	require.NoError(x.T(), err)
+	require.EqualValues(x.T(), []string{"a", "b", "c", "d", "e"}, data.Data)
+
+	value, err := data.Value()
+	require.NoError(x.T(), err)
+	require.EqualValues(x.T(), "a,b,c,d,e", value)
+}
+
+// ===============================
+
+func TestDBO(t *testing.T) {
+	suite.Run(t, new(DBOTestSuite))
+}