压缩历史至 95081aff

internal/cmd/api.go

@@ -0,0 +1,30 @@
+/*
+Copyright 2025 linux.do
+
+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 cmd
+
+import (
+	"github.com/linux-do/credit/internal/router"
+	"github.com/spf13/cobra"
+)
+
+var apiCmd = &cobra.Command{
+	Use:   "api",
+	Short: "credit API",
+	Run: func(cmd *cobra.Command, args []string) {
+		router.Serve()
+	},
+}

internal/cmd/root.go

@@ -0,0 +1,57 @@
+/*
+Copyright 2025 linux.do
+
+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 cmd
+
+import (
+	"log"
+
+	"github.com/linux-do/credit/internal/db/migrator"
+	"github.com/spf13/cobra"
+)
+
+var rootCmd = &cobra.Command{
+	Use: "linux-do-credit",
+	PreRun: func(cmd *cobra.Command, args []string) {
+		migrator.Migrate()
+	},
+	Run: func(cmd *cobra.Command, args []string) {
+		if len(args) == 0 {
+			log.Fatalf("[CMD] please provide a command\n")
+		}
+		appMode := args[0]
+		switch appMode {
+		case "api":
+			apiCmd.Run(apiCmd, args)
+		case "scheduler":
+			schedulerCmd.Run(schedulerCmd, args)
+		case "worker":
+			workerCmd.Run(workerCmd, args)
+		default:
+			log.Fatal("[CMD] unknown app mode\n")
+		}
+	},
+}
+
+func init() {
+	rootCmd.CompletionOptions.DisableDefaultCmd = true
+}
+
+func Execute() {
+	if err := rootCmd.Execute(); err != nil {
+		log.Fatalf("[CMD] execute failed; %s\n", err)
+	}
+}

internal/cmd/scheduler.go

@@ -0,0 +1,36 @@
+/*
+Copyright 2025 linux.do
+
+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 cmd
+
+import (
+	"log"
+
+	"github.com/linux-do/credit/internal/task/scheduler"
+
+	"github.com/spf13/cobra"
+)
+
+var schedulerCmd = &cobra.Command{
+	Use:   "scheduler",
+	Short: "credit Scheduler",
+	Run: func(cmd *cobra.Command, args []string) {
+		log.Println("[Scheduler] 启动定时任务调度服务")
+		if err := scheduler.StartScheduler(); err != nil {
+			log.Fatalf("[调度器] 启动失败: %v", err)
+		}
+	},
+}

internal/cmd/worker.go

@@ -0,0 +1,36 @@
+/*
+Copyright 2025 linux.do
+
+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 cmd
+
+import (
+	"log"
+
+	"github.com/linux-do/credit/internal/task/worker"
+
+	"github.com/spf13/cobra"
+)
+
+var workerCmd = &cobra.Command{
+	Use:   "worker",
+	Short: "credit Worker",
+	Run: func(cmd *cobra.Command, args []string) {
+		log.Println("[Worker] 启动任务处理服务")
+		if err := worker.StartWorker(); err != nil {
+			log.Fatalf("[工作器] 启动失败: %v", err)
+		}
+	},
+}