feat(plc4go): implement unsubscription requests

apache · Jul 31, 2023 · dcc5edb · dcc5edb
1 parent c9efef3
commit dcc5edb
Show file tree

Hide file tree

Showing 7 changed files with 174 additions and 20 deletions.
diff --git a/plc4go/pkg/api/model/plc_unsubscription.go b/plc4go/pkg/api/model/plc_unsubscription.go
@@ -21,21 +21,23 @@ package model
 
 import "context"
 
-type PlcUnsubscriptionRequestBuilder interface {
-	// TODO: Implement
-}
-
 type PlcUnsubscriptionRequestResult interface {
 	GetRequest() PlcUnsubscriptionRequest
 	GetResponse() PlcUnsubscriptionResponse
 	GetErr() error
 }
 
+type PlcUnsubscriptionRequestBuilder interface {
+	AddHandles(PlcSubscriptionHandle ...PlcSubscriptionHandle)
+	Build() (PlcUnsubscriptionRequest, error)
+}
+
 type PlcUnsubscriptionRequest interface {
 	Execute() <-chan PlcUnsubscriptionRequestResult
 	ExecuteWithContext(ctx context.Context) <-chan PlcUnsubscriptionRequestResult
 	PlcRequest
 }
 
 type PlcUnsubscriptionResponse interface {
+	GetRequest() PlcUnsubscriptionRequest
 }
diff --git a/plc4go/spi/model/DefaultPlcUnsubscriptionRequest.go b/plc4go/spi/model/DefaultPlcUnsubscriptionRequest.go
@@ -21,30 +21,50 @@ package model
 
 import (
 	"context"
+	"github.com/apache/plc4x/plc4go/spi"
 
 	apiModel "github.com/apache/plc4x/plc4go/pkg/api/model"
 )
 
+var _ apiModel.PlcUnsubscriptionRequestBuilder = &DefaultPlcUnsubscriptionRequestBuilder{}
+
+//go:generate go run ../../tools/plc4xgenerator/gen.go -type=DefaultPlcUnsubscriptionRequestBuilder
+type DefaultPlcUnsubscriptionRequestBuilder struct {
+	subscriptionHandles []apiModel.PlcSubscriptionHandle
+}
+
+func (d *DefaultPlcUnsubscriptionRequestBuilder) AddHandles(subscriptionHandles ...apiModel.PlcSubscriptionHandle) {
+	subscriptionHandles = append(subscriptionHandles, subscriptionHandles...)
+}
+
+func (d *DefaultPlcUnsubscriptionRequestBuilder) Build() (apiModel.PlcUnsubscriptionRequest, error) {
+	return NewDefaultPlcUnsubscriptionRequest(d.subscriptionHandles), nil
+}
+
 var _ apiModel.PlcUnsubscriptionRequest = &DefaultPlcUnsubscriptionRequest{}
 
 //go:generate go run ../../tools/plc4xgenerator/gen.go -type=DefaultPlcUnsubscriptionRequest
 type DefaultPlcUnsubscriptionRequest struct {
-	// TODO: implement
-	implementMe string
+	subscriber          spi.PlcSubscriber
+	subscriptionHandles []apiModel.PlcSubscriptionHandle
 }
 
-func NewDefaultPlcUnsubscriptionRequest() *DefaultPlcUnsubscriptionRequest {
-	return &DefaultPlcUnsubscriptionRequest{}
+func NewDefaultPlcUnsubscriptionRequest(subscriptionHandles []apiModel.PlcSubscriptionHandle) *DefaultPlcUnsubscriptionRequest {
+	return &DefaultPlcUnsubscriptionRequest{
+		subscriptionHandles: subscriptionHandles,
+	}
 }
 
 func (d *DefaultPlcUnsubscriptionRequest) Execute() <-chan apiModel.PlcUnsubscriptionRequestResult {
-	//TODO implement me
-	panic("implement me")
+	return d.ExecuteWithContext(context.Background())
 }
 
 func (d *DefaultPlcUnsubscriptionRequest) ExecuteWithContext(ctx context.Context) <-chan apiModel.PlcUnsubscriptionRequestResult {
-	//TODO implement me
-	panic("implement me")
+	return d.subscriber.Unsubscribe(ctx, d)
+}
+
+func (d *DefaultPlcUnsubscriptionRequest) GetSubscriptionHandles() []apiModel.PlcSubscriptionHandle {
+	return d.subscriptionHandles
 }
 
 func (d *DefaultPlcUnsubscriptionRequest) IsAPlcMessage() bool {

diff --git a/plc4go/spi/model/DefaultPlcUnsubscriptionRequestBuilder_plc4xgen.go b/plc4go/spi/model/DefaultPlcUnsubscriptionRequestBuilder_plc4xgen.go
diff --git a/plc4go/spi/model/DefaultPlcUnsubscriptionRequest_plc4xgen.go b/plc4go/spi/model/DefaultPlcUnsubscriptionRequest_plc4xgen.go
diff --git a/plc4go/spi/model/DefaultPlcUnsubscriptionResponse.go b/plc4go/spi/model/DefaultPlcUnsubscriptionResponse.go
@@ -25,10 +25,14 @@ var _ apiModel.PlcUnsubscriptionResponse = &DefaultPlcUnsubscriptionResponse{}
 
 //go:generate go run ../../tools/plc4xgenerator/gen.go -type=DefaultPlcUnsubscriptionResponse
 type DefaultPlcUnsubscriptionResponse struct {
-	// TODO: implement
-	implementMe string
+	request apiModel.PlcUnsubscriptionRequest `ignore:"true"`
+	none    string
 }
 
-func NewDefaultPlcUnsubscriptionResponse() *DefaultPlcUnsubscriptionResponse {
-	return &DefaultPlcUnsubscriptionResponse{}
+func (d *DefaultPlcUnsubscriptionResponse) GetRequest() apiModel.PlcUnsubscriptionRequest {
+	return d.request
+}
+
+func NewDefaultPlcUnsubscriptionResponse(request apiModel.PlcUnsubscriptionRequest) *DefaultPlcUnsubscriptionResponse {
+	return &DefaultPlcUnsubscriptionResponse{request: request}
 }
diff --git a/plc4go/spi/model/DefaultPlcUnsubscriptionResponse_plc4xgen.go b/plc4go/spi/model/DefaultPlcUnsubscriptionResponse_plc4xgen.go
diff --git a/plc4go/spi/model/render_test.go b/plc4go/spi/model/render_test.go
@@ -313,8 +313,8 @@ func TestRenderTestCustom(t *testing.T) {
 		},
 		{
 			sut: NewDefaultPlcUnsubscriptionRequestResult(
-				NewDefaultPlcUnsubscriptionRequest(),
-				NewDefaultPlcUnsubscriptionResponse(),
+				NewDefaultPlcUnsubscriptionRequest(nil),
+				NewDefaultPlcUnsubscriptionResponse(nil),
 				nil,
 			).(interface { // TODO: workaround
 				fmt.Stringer