[feat] add new api station in swagger

server/assets/protobuf/protogen.go

@@ -0,0 +1,3 @@
+package type_
+
+//go:generate protoc --go_out=path=source_relative:. types/type.proto

server/assets/protobuf/types/type.proto

@@ -0,0 +1,12 @@
+syntax = "proto3";
+
+package type_;
+
+enum Type {
+    UNSPECIFIED = 0;
+    LIMITARY_HOUR = 1;
+    STATION = 2;
+    STATION_GROUP = 3;
+    SUBSTITUTION = 4;
+    MATERIAL_RESOURCE = 5;
+}

server/go.mod

@@ -26,6 +26,7 @@ require (
 	gopkg.in/yaml.v3 v3.0.1
 	gorm.io/driver/postgres v1.5.2
 	gorm.io/gorm v1.25.3
+	sigs.k8s.io/yaml v1.3.0
 )
 
 require (
@@ -55,4 +56,5 @@ require (
 	golang.org/x/sys v0.11.0 // indirect
 	golang.org/x/text v0.12.0 // indirect
 	google.golang.org/protobuf v1.31.0 // indirect
+	gopkg.in/yaml.v2 v2.4.0 // indirect
 )

server/go.sum

@@ -261,3 +261,5 @@ gorm.io/driver/postgres v1.5.2 h1:ytTDxxEv+MplXOfFe3Lzm7SjG09fcdb3Z/c056DTBx0=
 gorm.io/driver/postgres v1.5.2/go.mod h1:fmpX0m2I1PKuR7mKZiEluwrP3hbs+ps7JIGMUBpCgl8=
 gorm.io/gorm v1.25.3 h1:zi4rHZj1anhZS2EuEODMhDisGy+Daq9jtPrNGgbQYD8=
 gorm.io/gorm v1.25.3/go.mod h1:L4uxeKpfBml98NYqVqwAdmV1a2nBtAec/cf3fpucW/k=
+sigs.k8s.io/yaml v1.3.0 h1:a2VclLzOGrwOHDiV8EfBGhvjHvP46CtW5j6POvhYGGo=
+sigs.k8s.io/yaml v1.3.0/go.mod h1:GeOyir5tyXNByN85N/dRIT9es5UQNerPYEKK56eTBm8=

server/internal/repositories/orm/models/model.go

@@ -9,5 +9,6 @@ type Models interface {
 func GetModelList() []Models {
 	return []Models{
 		&Account{},
+		&DataStorage{},
 	}
 }

server/internal/repositories/orm/models/storage.go

@@ -0,0 +1,17 @@
+package models
+
+import (
+	type_ "github.com/quocbang/data-flow-sync/server/assets/protobuf/types"
+)
+
+type DataStorage struct {
+	ID        string     `gorm:"column:id;type:text;primaryKey"`
+	Type      type_.Type `gorm:"type:int;not null"`
+	Content   []byte     `gorm:"type:bytea;not null"`
+	CreatedBy string     `gorm:"type:text;not null"`
+	CreatedAt int64      `gorm:"type:bigint;autoCreateTime"`
+}
+
+func (d *DataStorage) TableName() string {
+	return "data_storage"
+}

server/internal/repositories/repo_list.go

@@ -3,7 +3,10 @@ package repositories
 import (
 	"context"
 
+	"github.com/go-openapi/runtime/middleware"
 	"github.com/quocbang/data-flow-sync/server/internal/repositories/orm/models"
+	mod "github.com/quocbang/data-flow-sync/server/swagger/models"
+	"github.com/quocbang/data-flow-sync/server/swagger/restapi/operations/station"
 )
 
 type Services interface {
@@ -13,6 +16,7 @@ type Services interface {
 }
 
 type StationServices interface {
+	UpsertStationDataStorage(params station.CreateStationDataStorageParams, principal *mod.Principal) middleware.Responder
 }
 
 type StationGroupServices interface {

server/internal/repositories/services/station/impl.go

@@ -1,10 +1,19 @@
 package station
 
 import (
+	"fmt"
+	"net/http"
+
+	"github.com/go-openapi/runtime/middleware"
 	"github.com/go-redis/redis/v9"
 	"gorm.io/gorm"
+	"sigs.k8s.io/yaml"
 
+	type_ "github.com/quocbang/data-flow-sync/server/assets/protobuf/types"
 	"github.com/quocbang/data-flow-sync/server/internal/repositories"
+	repo "github.com/quocbang/data-flow-sync/server/internal/repositories/orm/models"
+	"github.com/quocbang/data-flow-sync/server/swagger/models"
+	"github.com/quocbang/data-flow-sync/server/swagger/restapi/operations/station"
 )
 
 type service struct {
@@ -18,3 +27,34 @@ func NewService(pg *gorm.DB, redis *redis.Client) repositories.StationServices {
 		redis: redis,
 	}
 }
+
+func (s service) UpsertStationDataStorage(params station.CreateStationDataStorageParams, principal *models.Principal) middleware.Responder {
+	errorMessage := ""
+	data, err := yaml.YAMLToJSON([]byte(params.Body.Content))
+	if err != nil {
+		errorMessage += fmt.Sprintf("%s station is not a valid YAML: %v\n", params.Body.ID, err)
+	}
+
+	if errorMessage != "" {
+		// return repositories.CreateStationDataStorageReply{}, fmt.Errorf(errorMessage)
+		return station.NewCreateStationDataStorageDefault(http.StatusBadRequest).WithPayload(&station.CreateStationDataStorageDefaultBody{
+			Details: errorMessage,
+		})
+	}
+
+	reply := s.pg.Create(&repo.DataStorage{
+		ID:        params.Body.ID,
+		Type:      type_.Type_STATION,
+		Content:   data,
+		CreatedBy: "tester_AI",
+		CreatedAt: 454545,
+	})
+
+	if reply.Error != nil {
+		return station.NewCreateStationDataStorageDefault(http.StatusBadRequest).WithPayload(&station.CreateStationDataStorageDefaultBody{
+			Details: reply.Error.Error(),
+		})
+	}
+
+	return station.NewCreateStationDataStorageOK()
+}

server/internal/repositories/station.go

@@ -0,0 +1,8 @@
+package repositories
+
+type CreateStationDataStorageReply struct {
+	RowsAffected RowsAffected
+}
+
+type GetStationReply struct {
+}

server/internal/services/service.go

@@ -19,6 +19,7 @@ func RegisterService(account AccountServices) *Services {
 }
 
 type StationServices interface {
+	CreateStationDataStorage()
 }
 
 type StationGroupServices interface {

server/swagger/restapi/configure_data_flow_sync.go

@@ -26,6 +26,7 @@ import (
 	"github.com/quocbang/data-flow-sync/server/swagger/restapi/operations"
 	"github.com/quocbang/data-flow-sync/server/swagger/restapi/operations/account"
 	"github.com/quocbang/data-flow-sync/server/swagger/restapi/operations/limitary_hour"
+	"github.com/quocbang/data-flow-sync/server/swagger/restapi/operations/station"
 )
 
 //go:generate swagger generate server --target ..\..\swagger --name DataFlowSync --spec ..\..\..\swagger.yml --principal models.Principal
@@ -138,6 +139,28 @@ func configureAPI(api *operations.DataFlowSyncAPI) http.Handler {
 		})
 	}
 
+	// station api handler
+	if api.StationCreateMergeRequestStationHandler == nil {
+		api.StationCreateMergeRequestStationHandler = station.CreateMergeRequestStationHandlerFunc(func(params station.CreateMergeRequestStationParams, principal *models.Principal) middleware.Responder {
+			return middleware.NotImplemented("operation station.CreateMergeRequestStation has not yet been implemented")
+		})
+	}
+	if api.StationGetStationDiffHandler == nil {
+		api.StationGetStationDiffHandler = station.GetStationDiffHandlerFunc(func(params station.GetStationDiffParams, principal *models.Principal) middleware.Responder {
+			return middleware.NotImplemented("operation station.GetStationDiff has not yet been implemented")
+		})
+	}
+	if api.StationListMergeRequestsStationHandler == nil {
+		api.StationListMergeRequestsStationHandler = station.ListMergeRequestsStationHandlerFunc(func(params station.ListMergeRequestsStationParams, principal *models.Principal) middleware.Responder {
+			return middleware.NotImplemented("operation station.ListMergeRequestsStation has not yet been implemented")
+		})
+	}
+	if api.StationUploadStationHandler == nil {
+		api.StationUploadStationHandler = station.UploadStationHandlerFunc(func(params station.UploadStationParams, principal *models.Principal) middleware.Responder {
+			return middleware.NotImplemented("operation station.UploadStation has not yet been implemented")
+		})
+	}
+
 	api.PreServerShutdown = func() {}
 
 	api.ServerShutdown = func() {}