feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

backend/domain/workflow/crossdomain/database/database.go

@@ -0,0 +1,143 @@
+/*
+ * Copyright 2025 coze-dev Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package database
+
+import (
+	"context"
+)
+
+type SQLParam struct {
+	Value  string
+	IsNull bool
+}
+type CustomSQLRequest struct {
+	DatabaseInfoID int64
+	SQL            string
+	Params         []SQLParam
+	IsDebugRun     bool
+	UserID         int64
+}
+
+type Object = map[string]any
+
+type Response struct {
+	RowNumber *int64
+	Objects   []Object
+}
+
+type Operator string
+type ClauseRelation string
+
+const (
+	ClauseRelationAND ClauseRelation = "and"
+	ClauseRelationOR  ClauseRelation = "or"
+)
+
+const (
+	OperatorEqual          Operator = "="
+	OperatorNotEqual       Operator = "!="
+	OperatorGreater        Operator = ">"
+	OperatorLesser         Operator = "<"
+	OperatorGreaterOrEqual Operator = ">="
+	OperatorLesserOrEqual  Operator = "<="
+	OperatorIn             Operator = "in"
+	OperatorNotIn          Operator = "not_in"
+	OperatorIsNull         Operator = "is_null"
+	OperatorIsNotNull      Operator = "is_not_null"
+	OperatorLike           Operator = "like"
+	OperatorNotLike        Operator = "not_like"
+)
+
+type ClauseGroup struct {
+	Single *Clause
+	Multi  *MultiClause
+}
+type Clause struct {
+	Left     string
+	Operator Operator
+}
+type MultiClause struct {
+	Clauses  []*Clause
+	Relation ClauseRelation
+}
+
+type Condition struct {
+	Left     string
+	Operator Operator
+	Right    any
+}
+
+type ConditionGroup struct {
+	Conditions []*Condition
+	Relation   ClauseRelation
+}
+
+type DeleteRequest struct {
+	DatabaseInfoID int64
+	ConditionGroup *ConditionGroup
+	IsDebugRun     bool
+	UserID         int64
+}
+
+type QueryRequest struct {
+	DatabaseInfoID int64
+	SelectFields   []string
+	Limit          int64
+	ConditionGroup *ConditionGroup
+	OrderClauses   []*OrderClause
+	IsDebugRun     bool
+	UserID         int64
+}
+
+type OrderClause struct {
+	FieldID string
+	IsAsc   bool
+}
+type UpdateRequest struct {
+	DatabaseInfoID int64
+	ConditionGroup *ConditionGroup
+	Fields         map[string]any
+	IsDebugRun     bool
+	UserID         int64
+}
+
+type InsertRequest struct {
+	DatabaseInfoID int64
+	Fields         map[string]any
+	IsDebugRun     bool
+	UserID         int64
+}
+
+func GetDatabaseOperator() DatabaseOperator {
+	return databaseOperatorImpl
+}
+func SetDatabaseOperator(d DatabaseOperator) {
+	databaseOperatorImpl = d
+}
+
+var (
+	databaseOperatorImpl DatabaseOperator
+)
+
+//go:generate  mockgen -destination databasemock/database_mock.go --package databasemock -source database.go
+type DatabaseOperator interface {
+	Execute(ctx context.Context, request *CustomSQLRequest) (*Response, error)
+	Query(ctx context.Context, request *QueryRequest) (*Response, error)
+	Update(context.Context, *UpdateRequest) (*Response, error)
+	Insert(ctx context.Context, request *InsertRequest) (*Response, error)
+	Delete(context.Context, *DeleteRequest) (*Response, error)
+}

backend/domain/workflow/crossdomain/database/databasemock/database_mock.go

@@ -0,0 +1,133 @@
+/*
+ * Copyright 2025 coze-dev Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+// Code generated by MockGen. DO NOT EDIT.
+// Source: database.go
+//
+// Generated by this command:
+//
+//	mockgen -destination databasemock/database_mock.go --package databasemock -source database.go
+//
+
+// Package databasemock is a generated GoMock package.
+package databasemock
+
+import (
+	context "context"
+	reflect "reflect"
+
+	database "github.com/coze-dev/coze-studio/backend/domain/workflow/crossdomain/database"
+	gomock "go.uber.org/mock/gomock"
+)
+
+// MockDatabaseOperator is a mock of DatabaseOperator interface.
+type MockDatabaseOperator struct {
+	ctrl     *gomock.Controller
+	recorder *MockDatabaseOperatorMockRecorder
+	isgomock struct{}
+}
+
+// MockDatabaseOperatorMockRecorder is the mock recorder for MockDatabaseOperator.
+type MockDatabaseOperatorMockRecorder struct {
+	mock *MockDatabaseOperator
+}
+
+// NewMockDatabaseOperator creates a new mock instance.
+func NewMockDatabaseOperator(ctrl *gomock.Controller) *MockDatabaseOperator {
+	mock := &MockDatabaseOperator{ctrl: ctrl}
+	mock.recorder = &MockDatabaseOperatorMockRecorder{mock}
+	return mock
+}
+
+// EXPECT returns an object that allows the caller to indicate expected use.
+func (m *MockDatabaseOperator) EXPECT() *MockDatabaseOperatorMockRecorder {
+	return m.recorder
+}
+
+// Delete mocks base method.
+func (m *MockDatabaseOperator) Delete(arg0 context.Context, arg1 *database.DeleteRequest) (*database.Response, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "Delete", arg0, arg1)
+	ret0, _ := ret[0].(*database.Response)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// Delete indicates an expected call of Delete.
+func (mr *MockDatabaseOperatorMockRecorder) Delete(arg0, arg1 any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockDatabaseOperator)(nil).Delete), arg0, arg1)
+}
+
+// Execute mocks base method.
+func (m *MockDatabaseOperator) Execute(ctx context.Context, request *database.CustomSQLRequest) (*database.Response, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "Execute", ctx, request)
+	ret0, _ := ret[0].(*database.Response)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// Execute indicates an expected call of Execute.
+func (mr *MockDatabaseOperatorMockRecorder) Execute(ctx, request any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Execute", reflect.TypeOf((*MockDatabaseOperator)(nil).Execute), ctx, request)
+}
+
+// Insert mocks base method.
+func (m *MockDatabaseOperator) Insert(ctx context.Context, request *database.InsertRequest) (*database.Response, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "Insert", ctx, request)
+	ret0, _ := ret[0].(*database.Response)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// Insert indicates an expected call of Insert.
+func (mr *MockDatabaseOperatorMockRecorder) Insert(ctx, request any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Insert", reflect.TypeOf((*MockDatabaseOperator)(nil).Insert), ctx, request)
+}
+
+// Query mocks base method.
+func (m *MockDatabaseOperator) Query(ctx context.Context, request *database.QueryRequest) (*database.Response, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "Query", ctx, request)
+	ret0, _ := ret[0].(*database.Response)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// Query indicates an expected call of Query.
+func (mr *MockDatabaseOperatorMockRecorder) Query(ctx, request any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Query", reflect.TypeOf((*MockDatabaseOperator)(nil).Query), ctx, request)
+}
+
+// Update mocks base method.
+func (m *MockDatabaseOperator) Update(arg0 context.Context, arg1 *database.UpdateRequest) (*database.Response, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "Update", arg0, arg1)
+	ret0, _ := ret[0].(*database.Response)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// Update indicates an expected call of Update.
+func (mr *MockDatabaseOperatorMockRecorder) Update(arg0, arg1 any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockDatabaseOperator)(nil).Update), arg0, arg1)
+}