From 3466f19db381027f416376808177f332413945ac Mon Sep 17 00:00:00 2001 From: zeaslity Date: Mon, 29 Apr 2024 11:56:58 +0800 Subject: [PATCH] =?UTF-8?q?[agent-operator]=20-=20=E5=A2=9E=E5=8A=A0?= =?UTF-8?q?=E9=95=9C=E5=83=8F=E5=90=8C=E6=AD=A5=E7=9A=84=E6=B6=88=E6=81=AF?= =?UTF-8?q?=E9=80=9A=E7=9F=A5=E5=86=85=E5=AE=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- agent-common/pusher/CmiiUpdateMessage.go | 2 +- agent-common/pusher/CmiiUpdateMessage_test.go | 7 +- agent-go/a_executor/K8sFunction.go | 68 ++-- message_pusher/cmii/ImageSyncMessage.go | 70 ++++ message_pusher/main.go | 1 + message_pusher/router/ImageSyncRouter.go | 38 ++ .../func/auto/beans/ImageFunctionEnum.java | 10 + .../wdd/func/auto/beans/ImageSyncContext.java | 4 + .../func/auto/service/ImageFuncScheduler.java | 40 +- .../java/io/wdd/pusher/MessagePusher.java | 80 ++++ .../main/java/io/wdd/pusher/NtfyClient.java | 33 -- .../wdd/pusher/beans/CmiiUpdateMessage.java | 27 ++ .../io/wdd/pusher/beans/ImageSyncMessage.java | 30 ++ .../wdd/pusher/beans/NtfyReceiveMessage.java | 43 ++ .../io/wdd/pusher/beans/NtfySendMessage.java | 55 +++ .../io/wdd/pusher/beans/PushMessageType.java | 28 ++ .../wdd/pusher/core/common/NtfyConstants.java | 14 - .../exception/NtfyConnectionException.java | 20 - .../pusher/core/exception/NtfyException.java | 24 -- .../exception/NtfyStreamingException.java | 24 -- .../io/wdd/pusher/core/model/ACTIONS.java | 19 - .../java/io/wdd/pusher/core/model/Action.java | 49 --- .../io/wdd/pusher/core/model/ClientType.java | 6 - .../io/wdd/pusher/core/model/NtfyRequest.java | 381 ------------------ .../io/wdd/pusher/core/model/PRIORITY.java | 27 -- .../pusher/core/model/PublishResponse.java | 5 - .../wdd/pusher/core/model/RequestModel.java | 117 ------ .../wdd/pusher/core/model/StreamRequest.java | 25 -- .../io/wdd/pusher/core/publish/PubClient.java | 9 - .../pusher/core/publish/PubClientImpl.java | 26 -- .../wdd/pusher/core/publish/PubService.java | 9 - .../pusher/core/publish/PubServiceImpl.java | 117 ------ .../core/stream/StreamingDataListener.java | 5 - .../pusher/core/stream/StreamingService.java | 63 --- .../io/wdd/pusher/example/PublishExample.java | 40 -- .../wdd/pusher/example/StreamingExample.java | 27 -- server/src/main/resources/application.yml | 4 +- .../test/java/io/wdd/pusher/PusherTest.java | 34 -- .../server/func/TestImageSyncScheduler.java | 2 +- .../java/io/wdd/server/pusher/PusherTest.java | 51 +++ 40 files changed, 525 insertions(+), 1109 deletions(-) create mode 100644 message_pusher/cmii/ImageSyncMessage.go create mode 100644 message_pusher/router/ImageSyncRouter.go create mode 100644 server/src/main/java/io/wdd/pusher/MessagePusher.java delete mode 100644 server/src/main/java/io/wdd/pusher/NtfyClient.java create mode 100644 server/src/main/java/io/wdd/pusher/beans/CmiiUpdateMessage.java create mode 100644 server/src/main/java/io/wdd/pusher/beans/ImageSyncMessage.java create mode 100644 server/src/main/java/io/wdd/pusher/beans/NtfyReceiveMessage.java create mode 100644 server/src/main/java/io/wdd/pusher/beans/NtfySendMessage.java create mode 100644 server/src/main/java/io/wdd/pusher/beans/PushMessageType.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/common/NtfyConstants.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/exception/NtfyConnectionException.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/exception/NtfyException.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/exception/NtfyStreamingException.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/model/ACTIONS.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/model/Action.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/model/ClientType.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/model/NtfyRequest.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/model/PRIORITY.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/model/PublishResponse.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/model/RequestModel.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/model/StreamRequest.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/publish/PubClient.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/publish/PubClientImpl.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/publish/PubService.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/publish/PubServiceImpl.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/stream/StreamingDataListener.java delete mode 100644 server/src/main/java/io/wdd/pusher/core/stream/StreamingService.java delete mode 100644 server/src/main/java/io/wdd/pusher/example/PublishExample.java delete mode 100644 server/src/main/java/io/wdd/pusher/example/StreamingExample.java delete mode 100644 server/src/test/java/io/wdd/pusher/PusherTest.java create mode 100644 server/src/test/java/io/wdd/server/pusher/PusherTest.java diff --git a/agent-common/pusher/CmiiUpdateMessage.go b/agent-common/pusher/CmiiUpdateMessage.go index 44568dd..aa60ea3 100644 --- a/agent-common/pusher/CmiiUpdateMessage.go +++ b/agent-common/pusher/CmiiUpdateMessage.go @@ -71,7 +71,7 @@ func (c *CmiiUpdateMessage) SendMessage() (message Message) { var m Message err = json.Unmarshal(bodyBytes, &m) if err != nil { - log.ErrorF("Error unmarshaling response body to JSON:", err) + log.ErrorF("Error unmarshaling response body to JSON: %s", err.Error()) return message } diff --git a/agent-common/pusher/CmiiUpdateMessage_test.go b/agent-common/pusher/CmiiUpdateMessage_test.go index 91af7e7..e3c4dff 100644 --- a/agent-common/pusher/CmiiUpdateMessage_test.go +++ b/agent-common/pusher/CmiiUpdateMessage_test.go @@ -1,6 +1,9 @@ package pusher -import "testing" +import ( + "testing" + "wdd.io/agent-common/utils" +) func TestCmiiUpdateMessage_SendMessage(t *testing.T) { c := &CmiiUpdateMessage{ @@ -12,5 +15,7 @@ func TestCmiiUpdateMessage_SendMessage(t *testing.T) { DeployStatus: false, } + utils.BeautifulPrint(c) + c.SendMessage() } diff --git a/agent-go/a_executor/K8sFunction.go b/agent-go/a_executor/K8sFunction.go index a4784ac..14e4fe0 100644 --- a/agent-go/a_executor/K8sFunction.go +++ b/agent-go/a_executor/K8sFunction.go @@ -14,7 +14,6 @@ import ( "sync" "time" image2 "wdd.io/agent-common/image" - "wdd.io/agent-common/utils" ) var k8sConfigFilePath = "/root/wdd/kube_config_cluster.yml" @@ -142,36 +141,51 @@ func K8sDeploymentUpdateTag(supreme, appName, newTag string) (bool, string) { updateResultLog := "" containers := deployment.Spec.Template.Spec.Containers - if len(containers) == 1 { - // only update this kind - container := containers[0] - oldName := container.Image + // 只支持container的数量为1的形式 + container := containers[0] - split := strings.Split(container.Image, ":") - if strings.HasPrefix(container.Image, image2.CmiiHarborPrefix) { - // harbor - container.Image = split[0] + ":" + newTag - } else if strings.Contains(container.Image, "8033") { - // 192.168.6.6:8033/rancher/k8s-dns-sidecar:v1.0.2 - container.Image = split[0] + ":" + split[1] + ":" + newTag + oldName := container.Image + split := strings.Split(container.Image, ":") + if strings.HasPrefix(container.Image, image2.CmiiHarborPrefix) { + // harbor + container.Image = split[0] + ":" + newTag + } else if strings.Contains(container.Image, "8033") { + // 192.168.6.6:8033/rancher/k8s-dns-sidecar:v1.0.2 + // 重新拼接 + container.Image = split[0] + ":" + split[1] + ":" + newTag + } + log.DebugF("[DeploymentUpdateTag] - update [%s] [%s] from [%s] to [%s]", deployment.Namespace, appName, oldName, container.Image) + + // 更新Cmii BIZ_GROUP + tagVersion := newTag + if strings.Contains(newTag, "-") { + tagVersion = strings.Split(newTag, "-")[0] + } + envList := container.Env + for _, envVar := range envList { + if envVar.Name == "IMAGE_VERSION" { + envVar.Value = tagVersion } - - updateResultLog = fmt.Sprintf(" [%s] [%s] [%s] from [%s] to [%s]", utils.TimeSplitFormatString(), supreme, appName, oldName, container.Image) - log.Info(updateResultLog) - - // re assign - deployment.Spec.Template.Spec.Containers[0] = container - - // update - _, err := k8sClient.AppsV1().Deployments(deployment.Namespace).Update(context.TODO(), deployment, metav1.UpdateOptions{}) - if err != nil { - log.ErrorF("[K8sDeploymentUpdateTag] - update [%s] [%s] from [%s] to [%s] error ! %s", supreme, appName, split[1], container.Image, err.Error()) - return false, "" + if envVar.Name == "BIZ_CONFIG_GROUP" { + envVar.Value = tagVersion } - } else if len(containers) == 2 { - log.ErrorF("[K8sDeploymentUpdateTag] - cant update app with 2 containers !") - return false, "" + if envVar.Name == "SYS_CONFIG_GROUP" { + envVar.Value = tagVersion + } + } + log.DebugF("[DeploymentUpdateTag] - update env IMAGE_VERSION to [%s]", tagVersion) + + // 赋值回去 很关键 + deployment.Spec.Template.Spec.Containers[0] = container + + // update + _, err := k8sClient.AppsV1().Deployments(deployment.Namespace).Update(context.TODO(), deployment, metav1.UpdateOptions{}) + if err != nil { + sprintf := fmt.Sprintf("[DeploymentUpdateTag] - update [%s] [%s] from [%s] to [%s] error ! %s", deployment.Namespace, appName, split[1], container.Image, err.Error()) + log.Error(sprintf) + return false, sprintf + } return true, updateResultLog diff --git a/message_pusher/cmii/ImageSyncMessage.go b/message_pusher/cmii/ImageSyncMessage.go new file mode 100644 index 0000000..8d90dcc --- /dev/null +++ b/message_pusher/cmii/ImageSyncMessage.go @@ -0,0 +1,70 @@ +package cmii + +import ( + "bytes" + "io.wdd.message_pusher/pusher" + "text/template" +) + +type ImageSyncMessage struct { + ImageFullName string `json:"imageFullName"` + ProjectName string `json:"projectName"` + ProjectNameSpace string `json:"projectNameSpace"` + ProjectMasterIP string `json:"projectMasterIP"` + ProjectMasterTopicName string `json:"projectMasterTopicName"` + InnerWorkTopicName string `json:"innerWorkTopicName"` + CurrentProcedure string `json:"currentProcedure"` + IsSyncFinished bool `json:"isSyncFinished"` +} + +const imageSyncTemplate = ` + {{if .IsSyncFinished}} + 镜像更新状态: 成功😍 + {{- else }} + 镜像更新状态: 失败👿👿👿 + {{- end}} +当前步骤: {{.CurrentProcedure}} +更新镜像: {{.ImageFullName}} +项目名称: {{.ProjectName}} +项目空间: {{.ProjectNameSpace}} +项目MasterIP: {{.ProjectMasterIP}} +项目Master名称: {{.ProjectMasterTopicName}} +中转节点名称: {{.InnerWorkTopicName}} +` + +var ImageSyncPushOptions = []pusher.PublishOption{ + pusher.WithTitle("镜像同步更新"), + pusher.WithPriority("3"), +} + +func (d *ImageSyncMessage) ParseTemplate() bytes.Buffer { + // 解析模板 + tmpl, err := template.New("imageSyncTemplate").Parse(imageSyncTemplate) + if err != nil { + panic(err) + } + + // 应用数据并打印结果 + var result bytes.Buffer + err = tmpl.Execute(&result, d) + if err != nil { + panic(err) + } + + return result +} + +// PublishMessage 使用默认的Client 发布更新的消息 +func (d *ImageSyncMessage) PublishMessage() *pusher.Message { + + parseTemplate := d.ParseTemplate() + + result, err := pusher.DefaultPusherClient.PublishDefault(parseTemplate, ImageSyncPushOptions) + if err != nil { + log.ErrorF("[ImageSyncMessage] - message push error ！ %s", err.Error()) + return result + } + + return result + +} diff --git a/message_pusher/main.go b/message_pusher/main.go index 54ec957..4d02754 100644 --- a/message_pusher/main.go +++ b/message_pusher/main.go @@ -45,6 +45,7 @@ func main() { { router.CMIIRouter(engine) router.OctopusRouter(engine) + router.ImageSyncRouter(engine) } engine.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler)) diff --git a/message_pusher/router/ImageSyncRouter.go b/message_pusher/router/ImageSyncRouter.go new file mode 100644 index 0000000..499fd3f --- /dev/null +++ b/message_pusher/router/ImageSyncRouter.go @@ -0,0 +1,38 @@ +package router + +import ( + "github.com/gin-gonic/gin" + "io.wdd.message_pusher/cmii" + "net/http" +) + +func ImageSyncRouter(r *gin.Engine) { + octopusGroup := r.Group("/image") + { + octopusGroup.POST("/sync", ImageSync) + } +} + +// ImageSync godoc +// @Summary 镜像同步消息 +// @Schemes +// @Description response to cmii update notification +// @Tags ImageSync +// @Accept json +// @Produce json +// @Param body body cmii.ImageSyncMessage true "请求体" +// @Success 200 {object} pusher.Message +// @Router /image/sync [post] +func ImageSync(c *gin.Context) { + // 获取请求中的参数 + var imageSyncMessage cmii.ImageSyncMessage + if err := c.ShouldBindJSON(&imageSyncMessage); err != nil { + c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid request body"}) + return + } + // 处理请求 + upgradeMessage := imageSyncMessage.PublishMessage() + + // *pusher.Message + c.JSON(http.StatusOK, upgradeMessage) +} diff --git a/server/src/main/java/io/wdd/func/auto/beans/ImageFunctionEnum.java b/server/src/main/java/io/wdd/func/auto/beans/ImageFunctionEnum.java index 51a6820..cea34fd 100644 --- a/server/src/main/java/io/wdd/func/auto/beans/ImageFunctionEnum.java +++ b/server/src/main/java/io/wdd/func/auto/beans/ImageFunctionEnum.java @@ -40,7 +40,17 @@ public enum ImageFunctionEnum { UPDATE_IMAGE_TAG( "UPDATE_IMAGE_TAG", "关闭防火墙" + ), + + FINISHED( + "FINISHED", + "同步更新完成" ); + + + /** + * 函数名 + */ String funcName; String desc; diff --git a/server/src/main/java/io/wdd/func/auto/beans/ImageSyncContext.java b/server/src/main/java/io/wdd/func/auto/beans/ImageSyncContext.java index 7396a91..0c07852 100644 --- a/server/src/main/java/io/wdd/func/auto/beans/ImageSyncContext.java +++ b/server/src/main/java/io/wdd/func/auto/beans/ImageSyncContext.java @@ -1,5 +1,6 @@ package io.wdd.func.auto.beans; +import io.wdd.pusher.beans.ImageSyncMessage; import io.wdd.server.beans.po.ProjectInfoPO; import io.wdd.server.beans.po.ServerInfoPO; import lombok.AllArgsConstructor; @@ -32,4 +33,7 @@ public class ImageSyncContext { // 我方的 机器 默认值为 192.168.35.71 String innerWorkerAgentName; + // 消息同步结构体 + ImageSyncMessage imageSyncMessage; + } diff --git a/server/src/main/java/io/wdd/func/auto/service/ImageFuncScheduler.java b/server/src/main/java/io/wdd/func/auto/service/ImageFuncScheduler.java index c70071b..6c67164 100644 --- a/server/src/main/java/io/wdd/func/auto/service/ImageFuncScheduler.java +++ b/server/src/main/java/io/wdd/func/auto/service/ImageFuncScheduler.java @@ -1,6 +1,10 @@ package io.wdd.func.auto.service; import io.wdd.func.auto.beans.ImageSyncContext; +import io.wdd.pusher.MessagePusher; +import io.wdd.pusher.beans.ImageSyncMessage; +import io.wdd.pusher.beans.NtfyReceiveMessage; +import io.wdd.pusher.beans.PushMessageType; import lombok.extern.slf4j.Slf4j; import org.apache.commons.collections.CollectionUtils; import org.springframework.stereotype.Service; @@ -22,6 +26,9 @@ public class ImageFuncScheduler { @Resource FuncService funcService; + @Resource + MessagePusher messagePusher; + public boolean runProcedure(ImageSyncContext imageSyncContext) { // before run @@ -55,16 +62,21 @@ public class ImageFuncScheduler { arrayList.add(imageSyncContext.getProjectInfoPO().getProjectNamespace());//namespace arrayList.add("");//targetImageFullName - imageSyncContext.setImageSyncFuncArgs(arrayList); + // 构建消息发送结构体 + imageSyncContext.setImageSyncMessage(new ImageSyncMessage()); + log.debug("beforeRunProcedure complete!"); } private void afterRunProcedure(ImageSyncContext imageSyncContext) { // message-pusher + ImageSyncMessage imageSyncMessage = imageSyncContext.getImageSyncMessage(); + NtfyReceiveMessage receiveMessage = messagePusher.Send(imageSyncMessage, PushMessageType.IMAGE_SYNC); + log.info("Image sync message push result => {}", receiveMessage.getRaw()); // 检查是否安装完成, 对安装环境进行判定 log.debug("afterRunProcedure complete!"); @@ -98,6 +110,7 @@ public class ImageFuncScheduler { log.info("prepare to update {} {} \n{}", imageSyncContext.getProjectInfoPO().getProjectNamespace(), imageSyncContext.getProjectMasterNode().getServerIpInV4(), realFullNameList); + String innerWorkerAgentName = imageSyncContext.getInnerWorkerAgentName(); Assert.notNull(innerWorkerAgentName, "inner worker agent name cant not be null !"); @@ -105,17 +118,30 @@ public class ImageFuncScheduler { String outsideAgentTopicName = imageSyncContext.getProjectMasterNode().getTopicName(); + // 设置消息体 + ImageSyncMessage imageSyncMessage = imageSyncContext.getImageSyncMessage(); + imageSyncMessage.setIsSyncFinished(Boolean.FALSE); + imageSyncMessage.setProjectNameSpace(imageSyncContext.getProjectInfoPO().getProjectNamespace()); + imageSyncMessage.setProjectName(imageSyncContext.getProjectInfoPO().getProjectName()); + imageSyncMessage.setInnerWorkTopicName(imageSyncContext.getInnerWorkerAgentName()); + imageSyncMessage.setProjectMasterTopicName(imageSyncContext.getProjectMasterNode().getTopicName()); + imageSyncMessage.setProjectMasterIP(imageSyncContext.getProjectMasterNode().getServerIpInV4()); + // List resultLog; - for (String fullName : realFullNameList) { + for (String imageFullName : realFullNameList) { + imageSyncMessage.setImageFullName(imageFullName); + // innerWorkerAgent - imageSyncFuncArgs.set(0, fullName); + imageSyncFuncArgs.set(0, imageFullName); resultLog = funcService.callImageFuncService(innerWorkerAgentName, DOWNLOAD_DOCKER_IMAGE, imageSyncFuncArgs); if (!JudgeSyncBaseCommandResult(resultLog)) { + imageSyncMessage.setCurrentProcedure(DOWNLOAD_DOCKER_IMAGE.getFuncName()); return; } resultLog = funcService.callImageFuncService(innerWorkerAgentName, COMPRESS_IMAGE_TO_GZIP, imageSyncFuncArgs); if (!JudgeSyncBaseCommandResult(resultLog)) { + imageSyncMessage.setCurrentProcedure(COMPRESS_IMAGE_TO_GZIP.getFuncName()); return; } String gzipFileName = resultLog.get(0); @@ -123,6 +149,7 @@ public class ImageFuncScheduler { imageSyncFuncArgs.set(2, gzipFileName); resultLog = funcService.callImageFuncService(innerWorkerAgentName, UPLOAD_GZIP_TO_OSS, imageSyncFuncArgs); if (!JudgeSyncBaseCommandResult(resultLog)) { + imageSyncMessage.setCurrentProcedure(UPLOAD_GZIP_TO_OSS.getFuncName()); return; } @@ -135,20 +162,27 @@ public class ImageFuncScheduler { // outside agent resultLog = funcService.callImageFuncService(outsideAgentTopicName, DOWNLOAD_GZIP_IMAGE_FILE, imageSyncFuncArgs); if (!JudgeSyncBaseCommandResult(resultLog)) { + imageSyncMessage.setCurrentProcedure(DOWNLOAD_GZIP_IMAGE_FILE.getFuncName()); return; } resultLog = funcService.callImageFuncService(outsideAgentTopicName, LOAD_DOCKER_IMAGE_FROM_GZIP, imageSyncFuncArgs); if (!JudgeSyncBaseCommandResult(resultLog)) { + imageSyncMessage.setCurrentProcedure(LOAD_DOCKER_IMAGE_FROM_GZIP.getFuncName()); return; } resultLog = funcService.callImageFuncService(outsideAgentTopicName, PUSH_IMAGE_TO_TARGET_HARBOR, imageSyncFuncArgs); if (!JudgeSyncBaseCommandResult(resultLog)) { + imageSyncMessage.setCurrentProcedure(PUSH_IMAGE_TO_TARGET_HARBOR.getFuncName()); return; } resultLog = funcService.callImageFuncService(outsideAgentTopicName, UPDATE_IMAGE_TAG, imageSyncFuncArgs); if (!JudgeSyncBaseCommandResult(resultLog)) { + imageSyncMessage.setCurrentProcedure(UPDATE_IMAGE_TAG.getFuncName()); return; } + + imageSyncMessage.setCurrentProcedure(FINISHED.getFuncName()); + imageSyncMessage.setIsSyncFinished(true); } } diff --git a/server/src/main/java/io/wdd/pusher/MessagePusher.java b/server/src/main/java/io/wdd/pusher/MessagePusher.java new file mode 100644 index 0000000..1081a29 --- /dev/null +++ b/server/src/main/java/io/wdd/pusher/MessagePusher.java @@ -0,0 +1,80 @@ +package io.wdd.pusher; + + +import com.fasterxml.jackson.core.JsonProcessingException; +import com.fasterxml.jackson.databind.ObjectMapper; +import io.wdd.pusher.beans.NtfyReceiveMessage; +import io.wdd.pusher.beans.PushMessageType; +import lombok.extern.slf4j.Slf4j; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.http.HttpHeaders; +import org.springframework.http.MediaType; +import org.springframework.http.ResponseEntity; +import org.springframework.stereotype.Service; +import org.springframework.web.client.RestTemplate; + +import javax.annotation.Resource; + +@Service +@Slf4j +public class MessagePusher { + + @Resource + ObjectMapper objectMapper; + @Value("${octopus.notify.pusher_url}") + String pushServerUrl; + + public NtfyReceiveMessage Send(Object sendMessage, PushMessageType pushMessageType) { + // Create a RestTemplate instance + RestTemplate restTemplate = new RestTemplate(); + +// Define the URL to send the request to + if (pushServerUrl.endsWith("/")) { + pushServerUrl.substring(0, pushServerUrl.length() - 1); + } + String url = pushServerUrl + pushMessageType.getUri(); + +// Define the JSON payload + String jsonPayload = ""; + try { + jsonPayload = objectMapper.writeValueAsString(sendMessage); + } catch (JsonProcessingException e) { + throw new RuntimeException(e); + } + +// Create a HttpHeaders instance to add authentication information + HttpHeaders headers = new HttpHeaders(); + headers.setContentType(MediaType.APPLICATION_JSON); + headers.add("Authorization", "Bearer tk_zvdb67fwj1hrjivkq3ga9z7u63av5"); + +// Create a MultiValueMap to add additional headers +// MultiValueMap additionalHeaders = new LinkedMultiValueMap<>(); +// additionalHeaders.add("X-Custom-Header", "Custom-Value"); +// headers.putAll(additionalHeaders); + +// Send the POST request + ResponseEntity response = restTemplate.postForEntity(url, jsonPayload, String.class, headers); + +// Check the response status code + NtfyReceiveMessage ntfyReceiveMessage = null; + if (response.getStatusCode().is2xxSuccessful()) { + // Process the response body (JSON) + String responseBody = response.getBody(); + + try { + ntfyReceiveMessage = objectMapper.readValue(responseBody, NtfyReceiveMessage.class); + } catch (JsonProcessingException e) { + throw new RuntimeException(e); + } + // ... + } else { + // Handle the error response + // ... + + log.error("Error sending message: {} with code: {}", response.getStatusCode(), jsonPayload); + } + + + return ntfyReceiveMessage; + } +} diff --git a/server/src/main/java/io/wdd/pusher/NtfyClient.java b/server/src/main/java/io/wdd/pusher/NtfyClient.java deleted file mode 100644 index 6cbdd28..0000000 --- a/server/src/main/java/io/wdd/pusher/NtfyClient.java +++ /dev/null @@ -1,33 +0,0 @@ -package io.wdd.pusher; - -import io.wdd.pusher.core.model.ClientType; -import io.wdd.pusher.core.publish.PubClient; -import io.wdd.pusher.core.publish.PubClientImpl; -import io.wdd.pusher.core.publish.PubServiceImpl; - -public final class NtfyClient { - - private PubClient pubClient; - - - public NtfyClient(ClientType type) { - switch (type) { - case PUB: - PubServiceImpl pubService = new PubServiceImpl(); - this.pubClient = new PubClientImpl(pubService); - break; - case SUB: - break; - - - default: - throw new IllegalArgumentException("Invalid client type"); - } - - } - - public PubClient getClient() { - return pubClient; - } -} - diff --git a/server/src/main/java/io/wdd/pusher/beans/CmiiUpdateMessage.java b/server/src/main/java/io/wdd/pusher/beans/CmiiUpdateMessage.java new file mode 100644 index 0000000..f1ee70a --- /dev/null +++ b/server/src/main/java/io/wdd/pusher/beans/CmiiUpdateMessage.java @@ -0,0 +1,27 @@ +package io.wdd.pusher.beans; + +import com.fasterxml.jackson.annotation.JsonProperty; +import lombok.AllArgsConstructor; +import lombok.Data; +import lombok.NoArgsConstructor; +import lombok.experimental.SuperBuilder; + +@Data +@AllArgsConstructor +@NoArgsConstructor +@SuperBuilder(toBuilder = true) +public class CmiiUpdateMessage { + + @JsonProperty("Namespace") + private String namespace; + @JsonProperty("AppName") + private String appName; + @JsonProperty("FromTag") + private String fromTag; + @JsonProperty("ToTag") + private String toTag; + @JsonProperty("Replicas") + private String replicas; + @JsonProperty("DeployStatus") + private Boolean deployStatus; +} diff --git a/server/src/main/java/io/wdd/pusher/beans/ImageSyncMessage.java b/server/src/main/java/io/wdd/pusher/beans/ImageSyncMessage.java new file mode 100644 index 0000000..f52152b --- /dev/null +++ b/server/src/main/java/io/wdd/pusher/beans/ImageSyncMessage.java @@ -0,0 +1,30 @@ +package io.wdd.pusher.beans; + +import lombok.AllArgsConstructor; +import lombok.Data; +import lombok.NoArgsConstructor; +import lombok.experimental.SuperBuilder; + +@Data +@AllArgsConstructor +@NoArgsConstructor +@SuperBuilder(toBuilder = true) +public class ImageSyncMessage { + + private String imageFullName; + + private String projectName; + + private String projectNameSpace; + + private String projectMasterIP; + + private String projectMasterTopicName; + + private String innerWorkTopicName; + + private String currentProcedure; + + private Boolean isSyncFinished; + +} diff --git a/server/src/main/java/io/wdd/pusher/beans/NtfyReceiveMessage.java b/server/src/main/java/io/wdd/pusher/beans/NtfyReceiveMessage.java new file mode 100644 index 0000000..01e9bcc --- /dev/null +++ b/server/src/main/java/io/wdd/pusher/beans/NtfyReceiveMessage.java @@ -0,0 +1,43 @@ +package io.wdd.pusher.beans; + +import com.fasterxml.jackson.annotation.JsonProperty; +import lombok.AllArgsConstructor; +import lombok.Data; +import lombok.NoArgsConstructor; +import lombok.experimental.SuperBuilder; + +@Data +@AllArgsConstructor +@NoArgsConstructor +@SuperBuilder(toBuilder = true) +public class NtfyReceiveMessage { + + @JsonProperty("ID") + private String id; + @JsonProperty("Event") + private String event; + @JsonProperty("Time") + private Integer time; + @JsonProperty("Topic") + private String topic; + @JsonProperty("Message") + private String message; + @JsonProperty("Title") + private String title; + @JsonProperty("Priority") + private Integer priority; + @JsonProperty("Tags") + private Object tags; + @JsonProperty("Click") + private String click; + @JsonProperty("Icon") + private String icon; + @JsonProperty("Attachment") + private Object attachment; + @JsonProperty("TopicURL") + private String topicURL; + @JsonProperty("SubscriptionID") + private String subscriptionID; + @JsonProperty("Raw") + private String raw; +} diff --git a/server/src/main/java/io/wdd/pusher/beans/NtfySendMessage.java b/server/src/main/java/io/wdd/pusher/beans/NtfySendMessage.java new file mode 100644 index 0000000..b0fe770 --- /dev/null +++ b/server/src/main/java/io/wdd/pusher/beans/NtfySendMessage.java @@ -0,0 +1,55 @@ +package io.wdd.pusher.beans; + + +import com.fasterxml.jackson.annotation.JsonProperty; +import lombok.AllArgsConstructor; +import lombok.Data; +import lombok.NoArgsConstructor; +import lombok.experimental.SuperBuilder; + +import java.util.List; + +@Data +@AllArgsConstructor +@NoArgsConstructor +@SuperBuilder(toBuilder = true) +public class NtfySendMessage { + + @JsonProperty("id") + private String id; + @JsonProperty("time") + private Integer time; + @JsonProperty("expires") + private Integer expires; + @JsonProperty("event") + private String event; + @JsonProperty("topic") + private String topic; + @JsonProperty("priority") + private Integer priority; + @JsonProperty("tags") + private List tags; + @JsonProperty("click") + private String click; + @JsonProperty("attachment") + private AttachmentDTO attachment; + @JsonProperty("title") + private String title; + @JsonProperty("message") + private String message; + + @NoArgsConstructor + @Data + public static class AttachmentDTO { + @JsonProperty("name") + private String name; + @JsonProperty("type") + private String type; + @JsonProperty("size") + private Integer size; + @JsonProperty("expires") + private Integer expires; + @JsonProperty("url") + private String url; + } +} diff --git a/server/src/main/java/io/wdd/pusher/beans/PushMessageType.java b/server/src/main/java/io/wdd/pusher/beans/PushMessageType.java new file mode 100644 index 0000000..15a29d5 --- /dev/null +++ b/server/src/main/java/io/wdd/pusher/beans/PushMessageType.java @@ -0,0 +1,28 @@ +package io.wdd.pusher.beans; + +import lombok.Getter; + +@Getter +public enum PushMessageType { + + + CMII_UPDATE( + "/cmii/update", + "更新CMII应用的消息通知类型" + ), + + IMAGE_SYNC( + "/image/sync", + "更新镜像同步的消类型" + ); + + String uri; + + String description; + + PushMessageType(String uri, String description) { + this.uri = uri; + this.description = description; + } + +} diff --git a/server/src/main/java/io/wdd/pusher/core/common/NtfyConstants.java b/server/src/main/java/io/wdd/pusher/core/common/NtfyConstants.java deleted file mode 100644 index 253a588..0000000 --- a/server/src/main/java/io/wdd/pusher/core/common/NtfyConstants.java +++ /dev/null @@ -1,14 +0,0 @@ -package io.wdd.pusher.core.common; - -public class NtfyConstants { - - public static final String DEFAULT_URL = "https://ntfy.sh/"; - public static final String POST = "POST"; - public static final String CONTENT_TYPE = "Content-Type"; - public static final String DEFAULT_HOST = "ntfy.sh"; - public static final String NTFY_CONNECTION_ERROR_MSG = "Unable to connect to ntfy host"; - public static final String CONNECTION_ERROR_MSG = "Unable to open connection"; - public static final String HTTPS = "https://"; - public static final String GET = "GET"; - public static final String JSON = "json"; -} diff --git a/server/src/main/java/io/wdd/pusher/core/exception/NtfyConnectionException.java b/server/src/main/java/io/wdd/pusher/core/exception/NtfyConnectionException.java deleted file mode 100644 index d00123d..0000000 --- a/server/src/main/java/io/wdd/pusher/core/exception/NtfyConnectionException.java +++ /dev/null @@ -1,20 +0,0 @@ -package io.wdd.pusher.core.exception; - -public class NtfyConnectionException extends Exception { - - public NtfyConnectionException() { - super(); - } - - public NtfyConnectionException(String message) { - super(message); - } - - public NtfyConnectionException(String message, Throwable cause) { - super(message, cause); - } - - public NtfyConnectionException(Throwable cause) { - super(cause); - } -} diff --git a/server/src/main/java/io/wdd/pusher/core/exception/NtfyException.java b/server/src/main/java/io/wdd/pusher/core/exception/NtfyException.java deleted file mode 100644 index 2b57dfd..0000000 --- a/server/src/main/java/io/wdd/pusher/core/exception/NtfyException.java +++ /dev/null @@ -1,24 +0,0 @@ -package io.wdd.pusher.core.exception; - - -public class NtfyException extends Exception { - public NtfyException() { - super(); - } - - public NtfyException(String message) { - super(message); - } - - public NtfyException(String message, Throwable cause) { - super(message, cause); - } - - public NtfyException(Throwable cause) { - super(cause); - } - - protected NtfyException(String message, Throwable cause, boolean enableSuppression, boolean writableStackTrace) { - super(message, cause, enableSuppression, writableStackTrace); - } -} diff --git a/server/src/main/java/io/wdd/pusher/core/exception/NtfyStreamingException.java b/server/src/main/java/io/wdd/pusher/core/exception/NtfyStreamingException.java deleted file mode 100644 index 3763483..0000000 --- a/server/src/main/java/io/wdd/pusher/core/exception/NtfyStreamingException.java +++ /dev/null @@ -1,24 +0,0 @@ -package io.wdd.pusher.core.exception; - - -public class NtfyStreamingException extends Exception { - public NtfyStreamingException() { - super(); - } - - public NtfyStreamingException(String message) { - super(message); - } - - public NtfyStreamingException(String message, Throwable cause) { - super(message, cause); - } - - public NtfyStreamingException(Throwable cause) { - super(cause); - } - - protected NtfyStreamingException(String message, Throwable cause, boolean enableSuppression, boolean writableStackTrace) { - super(message, cause, enableSuppression, writableStackTrace); - } -} diff --git a/server/src/main/java/io/wdd/pusher/core/model/ACTIONS.java b/server/src/main/java/io/wdd/pusher/core/model/ACTIONS.java deleted file mode 100644 index 3504c3f..0000000 --- a/server/src/main/java/io/wdd/pusher/core/model/ACTIONS.java +++ /dev/null @@ -1,19 +0,0 @@ -package io.wdd.pusher.core.model; - - -public enum ACTIONS { - - VIEW("view"), - BROADCAST("broadcast"), - HTTP("http"); - - private final String name; - - ACTIONS(String name) { - this.name = name; - } - - public String getName() { - return name; - } -} diff --git a/server/src/main/java/io/wdd/pusher/core/model/Action.java b/server/src/main/java/io/wdd/pusher/core/model/Action.java deleted file mode 100644 index 5a041d2..0000000 --- a/server/src/main/java/io/wdd/pusher/core/model/Action.java +++ /dev/null @@ -1,49 +0,0 @@ -package io.wdd.pusher.core.model; - -public class Action { - private ACTIONS action; - private String label; - private String url; - private boolean clear; - private String body; - - public ACTIONS getAction() { - return action; - } - - public void setAction(ACTIONS action) { - this.action = action; - } - - public String getLabel() { - return label; - } - - public void setLabel(String label) { - this.label = label; - } - - public String getUrl() { - return url; - } - - public void setUrl(String url) { - this.url = url; - } - - public boolean isClear() { - return clear; - } - - public void setClear(boolean clear) { - this.clear = clear; - } - - public String getBody() { - return body; - } - - public void setBody(String body) { - this.body = body; - } -} diff --git a/server/src/main/java/io/wdd/pusher/core/model/ClientType.java b/server/src/main/java/io/wdd/pusher/core/model/ClientType.java deleted file mode 100644 index d417e4a..0000000 --- a/server/src/main/java/io/wdd/pusher/core/model/ClientType.java +++ /dev/null @@ -1,6 +0,0 @@ -package io.wdd.pusher.core.model; - - -public enum ClientType { - PUB, SUB, STREAM; -} diff --git a/server/src/main/java/io/wdd/pusher/core/model/NtfyRequest.java b/server/src/main/java/io/wdd/pusher/core/model/NtfyRequest.java deleted file mode 100644 index 9ba2307..0000000 --- a/server/src/main/java/io/wdd/pusher/core/model/NtfyRequest.java +++ /dev/null @@ -1,381 +0,0 @@ -package io.wdd.pusher.core.model; - - -import java.util.Base64; -import java.util.List; - -/** - * This class represents a message object for publishing/ sending notification message. - * It contains various properties related to the structure of the notification. - */ - -public class NtfyRequest { - /** - * The topic to which you are publishing the message. - */ - private String topic; - - /** - * The actual message content to be published. It can contain markdown as well. - */ - private String message; - - /** - * The host or source to which you are publishing. - */ - private String host; - - /** - * The title or subject of the message. - */ - private String title; - - /** - * The priority level of the message. - */ - private PRIORITY priority; - - /** - * A list of tags associated with the message. - */ - private List tags; - - /** - * The URL associated with the message. - */ - private String url; - - /** - * Indicates whether the message content is in Markdown format. - */ - private boolean isMarkdown; - - /** - * A list of actions related to the message. - */ - private List actions; - - /** - * An attachment associated with the message. This will be the url linked of the attachment file. - */ - private String attach; - - /** - * This local file that needs to be attached with the message, - */ - private String fileName; - - /** - * The icon image url that needs to be used in the notification - */ - private String icon; - /** - * The email address to which the notification is to be sent. - */ - private String email; - - /** - * The phone number to which the notification will be read to via call. - */ - private String phone; - - /** - * Access token, generated in ntfy CLI - */ - private String accessToken; - /** - * Basic auth in format username:password - */ - private String authToken; - - /** - * Get the auth token to authenticate to Server - * - * @return auth token as String - */ - public String getAuthToken() { - return this.authToken; - } - - /** - * Sets the auth token (Base64 of username:password) - * - * @param userCombo username/password combo for generation of access token - */ - public void setAuthToken(String userCombo) { - this.authToken = Base64.getEncoder().encodeToString(userCombo.getBytes()); - } - - /** - * Get the accessToken for Authentication - * - * @return access token - */ - public String getAccessToken() { - return this.accessToken; - } - - /** - * @param accessToken The authToken from CLI - */ - public void setAccessToken(String accessToken) { - this.accessToken = accessToken; - } - - - /** - * Get the phone number to which the notification will be read via a call. - * - * @return The phone number as a String. - */ - public String getPhone() { - return phone; - } - - /** - * Set the phone number to which the notification will be read via a call. - * - * @param phone The phone number to set as a String. - */ - public void setPhone(String phone) { - this.phone = phone; - } - - - /** - * Get the email address to which the notification is to be sent. - * - * @return The email address as a String. - */ - public String getEmail() { - return email; - } - - /** - * Set the email address to which the notification is to be sent. - * - * @param email The email address to set as a String. - */ - public void setEmail(String email) { - this.email = email; - } - - /** - * Get the icon image URL that needs to be used in the notification. - * - * @return The icon image URL as a String. - */ - public String getIcon() { - return icon; - } - - /** - * Set the icon image URL that needs to be used in the notification. - * - * @param icon The icon image URL to set as a String. - */ - public void setIcon(String icon) { - this.icon = icon; - } - - - /** - * Get the name of the local file that needs to be attached with the message. - * - * @return The file name as a String. - */ - public String getFileName() { - return fileName; - } - - /** - * Set the name of the local file that needs to be attached with the message. - * - * @param fileName The file name to set as a String. - */ - public void setFileName(String fileName) { - this.fileName = fileName; - } - - - /** - * Get the attachment url associated with the message. - * - * @return The attachment url as a String. - */ - public String getAttach() { - return attach; - } - - /** - * Set the attachment url for the message. - * - * @param attach The attachment url to set as a String. - */ - public void setAttach(String attach) { - this.attach = attach; - } - - /** - * Get the list of actions related to the message. - * - * @return A list of Action objects. - */ - public List getActions() { - return actions; - } - - /** - * Set the list of actions for the message. - * - * @param actions A list of Action objects to set. - */ - public void setActions(List actions) { - this.actions = actions; - } - - /** - * Check if the message content is in Markdown format. - * - * @return True if the message content is in Markdown, false otherwise. - */ - public boolean isMarkdown() { - return isMarkdown; - } - - /** - * Set whether the message content is in Markdown format. - * - * @param markdown True to set the message content as Markdown, false otherwise. - */ - public void setMarkdown(boolean markdown) { - isMarkdown = markdown; - } - - /** - * Get the URL associated with the message. - * - * @return The URL as a String. - */ - public String getUrl() { - return url; - } - - /** - * Set the URL for the message. - * - * @param url The URL to set as a String. - */ - public void setUrl(String url) { - this.url = url; - } - - /** - * Get the actual message content. - * - * @return The message content as a String. - */ - public String getMessage() { - return message; - } - - /** - * Set the message content. - * - * @param message The message content to set as a String. - */ - public void setMessage(String message) { - this.message = message; - } - - /** - * Get the host or source of the message. - * - * @return The host as a String. - */ - public String getHost() { - return host; - } - - /** - * Set the host or source of the message. - * - * @param host The host to set as a String. - */ - public void setHost(String host) { - this.host = host; - } - - /** - * Get the title or subject of the message. - * - * @return The title as a String. - */ - public String getTitle() { - return title; - } - - /** - * Set the title or subject of the message. - * - * @param title The title to set as a String. - */ - public void setTitle(String title) { - this.title = title; - } - - /** - * Get the priority level of the message. - * - * @return The priority level as a PRIORITY enum value. - */ - public PRIORITY getPriority() { - return priority; - } - - /** - * Set the priority level of the message. - * - * @param priority The priority level to set as a PRIORITY enum value. - */ - public void setPriority(PRIORITY priority) { - this.priority = priority; - } - - /** - * Get the list of tags associated with the message. - * - * @return A list of tags as Strings. - */ - public List getTags() { - return tags; - } - - /** - * Set the list of tags associated with the message. - * - * @param tags A list of tags to set as Strings. - */ - public void setTags(List tags) { - this.tags = tags; - } - - /** - * Get the topic to which the message is published. - * - * @return The topic as a String. - */ - public String getTopic() { - return topic; - } - - /** - * Set the topic to which the message is published. - * - * @param topic The topic to set as a String. - */ - public void setTopic(String topic) { - this.topic = topic; - } -} - diff --git a/server/src/main/java/io/wdd/pusher/core/model/PRIORITY.java b/server/src/main/java/io/wdd/pusher/core/model/PRIORITY.java deleted file mode 100644 index 5aac035..0000000 --- a/server/src/main/java/io/wdd/pusher/core/model/PRIORITY.java +++ /dev/null @@ -1,27 +0,0 @@ -package io.wdd.pusher.core.model; - - -public enum PRIORITY { - MAX("Max", 5), - HIGH("High", 4), - DEFAULT("Default", 3), - LOW("Low", 2), - MIN("Min", 1); - - private final String name; - private final int level; - - PRIORITY(String name, int level) { - this.name = name; - this.level = level; - } - - public String getName() { - return name; - } - - public int getLevel() { - return level; - } - -} diff --git a/server/src/main/java/io/wdd/pusher/core/model/PublishResponse.java b/server/src/main/java/io/wdd/pusher/core/model/PublishResponse.java deleted file mode 100644 index 579b846..0000000 --- a/server/src/main/java/io/wdd/pusher/core/model/PublishResponse.java +++ /dev/null @@ -1,5 +0,0 @@ -package io.wdd.pusher.core.model; - - -public class PublishResponse { -} diff --git a/server/src/main/java/io/wdd/pusher/core/model/RequestModel.java b/server/src/main/java/io/wdd/pusher/core/model/RequestModel.java deleted file mode 100644 index 05ad80f..0000000 --- a/server/src/main/java/io/wdd/pusher/core/model/RequestModel.java +++ /dev/null @@ -1,117 +0,0 @@ -package io.wdd.pusher.core.model; - - -import java.util.List; - -public class RequestModel { - - private String topic; - private String message; - private String title; - private List tags; - private int priority; - private boolean markdown; - private List actions; - private String attach; - private String fileName; - private String icon; - private String email; - private String call; - - public String getCall() { - return call; - } - - public void setCall(String call) { - this.call = call; - } - - public String getEmail() { - return email; - } - - public void setEmail(String email) { - this.email = email; - } - - public String getIcon() { - return icon; - } - - public void setIcon(String icon) { - this.icon = icon; - } - - public String getFileName() { - return fileName; - } - - public void setFileName(String fileName) { - this.fileName = fileName; - } - - public String getAttach() { - return attach; - } - - public void setAttach(String attach) { - this.attach = attach; - } - - public List getActions() { - return actions; - } - - public void setActions(List actions) { - this.actions = actions; - } - - public List getTags() { - return tags; - } - - public void setTags(List tags) { - this.tags = tags; - } - - public int getPriority() { - return priority; - } - - public void setPriority(int priority) { - this.priority = priority; - } - - public boolean isMarkdown() { - return markdown; - } - - public void setMarkdown(boolean markdown) { - this.markdown = markdown; - } - - public String getTopic() { - return topic; - } - - public void setTopic(String topic) { - this.topic = topic; - } - - public String getMessage() { - return message; - } - - public void setMessage(String message) { - this.message = message; - } - - - public String getTitle() { - return title; - } - - public void setTitle(String title) { - this.title = title; - } -} diff --git a/server/src/main/java/io/wdd/pusher/core/model/StreamRequest.java b/server/src/main/java/io/wdd/pusher/core/model/StreamRequest.java deleted file mode 100644 index 6f8c825..0000000 --- a/server/src/main/java/io/wdd/pusher/core/model/StreamRequest.java +++ /dev/null @@ -1,25 +0,0 @@ -package io.wdd.pusher.core.model; - - -public class StreamRequest { - private String host; - private String topic; - - public String getHost() { - return host; - } - - public void setHost(String host) { - this.host = host; - } - - public String getTopic() { - return topic; - } - - public void setTopic(String topic) { - this.topic = topic; - } - - -} diff --git a/server/src/main/java/io/wdd/pusher/core/publish/PubClient.java b/server/src/main/java/io/wdd/pusher/core/publish/PubClient.java deleted file mode 100644 index 6aa7951..0000000 --- a/server/src/main/java/io/wdd/pusher/core/publish/PubClient.java +++ /dev/null @@ -1,9 +0,0 @@ -package io.wdd.pusher.core.publish; - - -import io.wdd.pusher.core.exception.NtfyException; -import io.wdd.pusher.core.model.NtfyRequest; - -public interface PubClient { - String sendNotification(NtfyRequest ntfyRequest) throws NtfyException; -} diff --git a/server/src/main/java/io/wdd/pusher/core/publish/PubClientImpl.java b/server/src/main/java/io/wdd/pusher/core/publish/PubClientImpl.java deleted file mode 100644 index 4b2fd11..0000000 --- a/server/src/main/java/io/wdd/pusher/core/publish/PubClientImpl.java +++ /dev/null @@ -1,26 +0,0 @@ -package io.wdd.pusher.core.publish; - - -import io.wdd.pusher.core.exception.NtfyException; -import io.wdd.pusher.core.model.NtfyRequest; - -import java.util.logging.Logger; - -public class PubClientImpl implements PubClient { - - private static final Logger logger = Logger.getLogger(PubClientImpl.class.getName()); - private final PubService pubService; - - public PubClientImpl(PubService pubService) { - this.pubService = pubService; - } - - @Override - public String sendNotification(NtfyRequest ntfyRequest) throws NtfyException { - String response = pubService.publish(ntfyRequest); - if (null != response) { - logger.info("Response from server : " + response); - } - return response; - } -} diff --git a/server/src/main/java/io/wdd/pusher/core/publish/PubService.java b/server/src/main/java/io/wdd/pusher/core/publish/PubService.java deleted file mode 100644 index 99b173f..0000000 --- a/server/src/main/java/io/wdd/pusher/core/publish/PubService.java +++ /dev/null @@ -1,9 +0,0 @@ -package io.wdd.pusher.core.publish; - - -import io.wdd.pusher.core.exception.NtfyException; -import io.wdd.pusher.core.model.NtfyRequest; - -public interface PubService { - String publish(NtfyRequest ntfyRequest) throws NtfyException; -} diff --git a/server/src/main/java/io/wdd/pusher/core/publish/PubServiceImpl.java b/server/src/main/java/io/wdd/pusher/core/publish/PubServiceImpl.java deleted file mode 100644 index 60f2b48..0000000 --- a/server/src/main/java/io/wdd/pusher/core/publish/PubServiceImpl.java +++ /dev/null @@ -1,117 +0,0 @@ -package io.wdd.pusher.core.publish; - -import com.fasterxml.jackson.databind.ObjectMapper; -import io.wdd.pusher.core.common.NtfyConstants; -import io.wdd.pusher.core.exception.NtfyConnectionException; -import io.wdd.pusher.core.exception.NtfyException; -import io.wdd.pusher.core.model.NtfyRequest; -import io.wdd.pusher.core.model.PRIORITY; -import io.wdd.pusher.core.model.RequestModel; - -import java.io.DataOutputStream; -import java.io.IOException; -import java.net.HttpURLConnection; -import java.net.URL; -import java.util.logging.Logger; - -public class PubServiceImpl implements PubService { - private static final Logger logger = Logger.getLogger(PubServiceImpl.class.getName()); - - @Override - public String publish(NtfyRequest request) throws NtfyException { - return publishMessage(request); - } - - private String publishMessage(NtfyRequest request) throws NtfyException { - String response = null; - try { - if (null == request.getHost()) { - request.setUrl(NtfyConstants.DEFAULT_URL); - request.setHost(NtfyConstants.DEFAULT_HOST); - } else { - request.setUrl(NtfyConstants.HTTPS + request.getHost() + "/"); - } - if (null == request.getPriority()) { - request.setPriority(PRIORITY.DEFAULT); - } - - response = sendPublishRequest(request); - - } catch (IOException e) { - logger.severe(NtfyConstants.CONNECTION_ERROR_MSG); - throw new NtfyException(e); - } catch (NtfyConnectionException e) { - logger.severe(NtfyConstants.NTFY_CONNECTION_ERROR_MSG); - throw new NtfyException(e); - } catch (Exception e) { - throw new RuntimeException(e); - } - return response; - - } - - private String sendPublishRequest(NtfyRequest request) throws Exception { - try { - URL obj = new URL(request.getUrl()); - HttpURLConnection con = (HttpURLConnection) obj.openConnection(); - - - con.setRequestMethod(NtfyConstants.POST); - con.setConnectTimeout(10000); - con.setRequestProperty(NtfyConstants.CONTENT_TYPE, "application/json"); - - //handle authentication (if supplied) - if (request.getAccessToken() != null) { - con.setRequestProperty("Authorization", "Bearer " + request.getAccessToken()); - } else if (request.getAuthToken() != null) { - con.setRequestProperty("Authorization", "Basic " + request.getAuthToken()); - } - // Enable input/output streams - con.setDoOutput(true); - - try (DataOutputStream wr = new DataOutputStream(con.getOutputStream())) { - wr.writeBytes(new ObjectMapper().writeValueAsString(createPublishRequest(request))); - wr.flush(); - } - // Get the response from the server - int responseCode = con.getResponseCode(); - - if (responseCode == HttpURLConnection.HTTP_OK) { - StringBuilder response = new StringBuilder(); - try (java.io.BufferedReader in = new java.io.BufferedReader(new java.io.InputStreamReader(con.getInputStream()))) { - String inputLine; - while ((inputLine = in.readLine()) != null) { - response.append(inputLine); - } - } - return response.toString(); - } else { - throw new NtfyConnectionException(NtfyConstants.NTFY_CONNECTION_ERROR_MSG + " : " + responseCode + " " + con.getResponseMessage()); - } - - - } catch (IOException e) { - throw new NtfyConnectionException(NtfyConstants.CONNECTION_ERROR_MSG, e); - } - - } - - private RequestModel createPublishRequest(NtfyRequest request) { - RequestModel model = new RequestModel(); - model.setTopic(request.getTopic()); - model.setMessage(request.getMessage()); - model.setTitle(request.getTitle()); - model.setTags(request.getTags()); - model.setMarkdown(request.isMarkdown()); - model.setPriority(request.getPriority().getLevel()); - model.setActions(request.getActions()); - model.setAttach(request.getAttach()); - model.setFileName(request.getFileName()); - model.setIcon(request.getIcon()); - model.setEmail(request.getEmail()); - model.setCall(request.getPhone()); - return model; - } - - -} diff --git a/server/src/main/java/io/wdd/pusher/core/stream/StreamingDataListener.java b/server/src/main/java/io/wdd/pusher/core/stream/StreamingDataListener.java deleted file mode 100644 index 1a9194e..0000000 --- a/server/src/main/java/io/wdd/pusher/core/stream/StreamingDataListener.java +++ /dev/null @@ -1,5 +0,0 @@ -package io.wdd.pusher.core.stream; - -public interface StreamingDataListener { - void onDataReceived(String data); -} \ No newline at end of file diff --git a/server/src/main/java/io/wdd/pusher/core/stream/StreamingService.java b/server/src/main/java/io/wdd/pusher/core/stream/StreamingService.java deleted file mode 100644 index 1145a03..0000000 --- a/server/src/main/java/io/wdd/pusher/core/stream/StreamingService.java +++ /dev/null @@ -1,63 +0,0 @@ -package io.wdd.pusher.core.stream; - -import io.wdd.pusher.core.common.NtfyConstants; -import io.wdd.pusher.core.exception.NtfyStreamingException; -import io.wdd.pusher.core.model.StreamRequest; - -import java.io.BufferedReader; -import java.io.IOException; -import java.io.InputStreamReader; -import java.net.HttpURLConnection; -import java.net.URL; - -public class StreamingService { - private final String streamingUrl; - private boolean running; - private StreamingDataListener dataListener; - - public StreamingService(StreamRequest request) { - this.streamingUrl = NtfyConstants.HTTPS + request.getHost() + "/" + - request.getTopic() + "/" + NtfyConstants.JSON; - } - - public void start() throws NtfyStreamingException { - try { - running = true; - Thread streamThread = new Thread(() -> { - try { - URL url = new URL(streamingUrl); - HttpURLConnection connection = (HttpURLConnection) url.openConnection(); - connection.setRequestMethod(NtfyConstants.GET); - BufferedReader reader = new BufferedReader(new InputStreamReader(connection.getInputStream())); - - while (running) { - String line = reader.readLine(); - if (line != null) { - if (dataListener != null) { - dataListener.onDataReceived(line); - } - } - } - reader.close(); - connection.disconnect(); - } catch (IOException e) { - System.err.println(e.getMessage()); - } - }); - streamThread.start(); - } catch (Exception e) { - throw new NtfyStreamingException(NtfyConstants.CONNECTION_ERROR_MSG, e); - } - - } - - public void stop() { - running = false; - } - - public void setDataListener(StreamingDataListener listener) { - this.dataListener = listener; - } - - -} diff --git a/server/src/main/java/io/wdd/pusher/example/PublishExample.java b/server/src/main/java/io/wdd/pusher/example/PublishExample.java deleted file mode 100644 index 1aedc0e..0000000 --- a/server/src/main/java/io/wdd/pusher/example/PublishExample.java +++ /dev/null @@ -1,40 +0,0 @@ -package io.wdd.pusher.example; - - -import io.wdd.pusher.NtfyClient; -import io.wdd.pusher.core.exception.NtfyException; -import io.wdd.pusher.core.model.*; -import io.wdd.pusher.core.publish.PubClient; - -import java.util.ArrayList; -import java.util.List; - -public class PublishExample { - - public static void main(String[] args) throws NtfyException { - PubClient client = new NtfyClient(ClientType.PUB).getClient(); - NtfyRequest request = new NtfyRequest(); - request.setTopic("test_ntfy2"); - request.setMessage("Look ma, **bold text**, *italics*, ..."); - request.setTitle("This is the obj msg"); - request.setPriority(PRIORITY.MAX); - request.setAttach("https://media.licdn.com/dms/image/D4E03AQEZTNXuX3kG7g/profile-displayphoto-shrink_400_400/0/1669618932666?e=1699488000&v=beta&t=q2z_UDFvwTZa02SligKZqgwk66BjuXQZxWtQF_K1Jqw"); - request.setFileName("Screenshot.png"); - request.setIcon("https://styles.redditmedia.com/t5_32uhe/styles/communityIcon_xnt6chtnr2j21.png"); - request.setEmail("mahesh.b.pec@gmail.com"); - request.setPhone(""); - - Action action = new Action(); - action.setAction(ACTIONS.VIEW); - action.setLabel("Open github"); - action.setUrl("https://github.com/MaheshBabu11/ntfy-java"); - action.setClear(true); - - List actions = new ArrayList<>(List.of(action)); - List tags = new ArrayList<>(List.of("+1", "warning")); - request.setTags(tags); - request.setMarkdown(true); - request.setActions(actions); - client.sendNotification(request); - } -} diff --git a/server/src/main/java/io/wdd/pusher/example/StreamingExample.java b/server/src/main/java/io/wdd/pusher/example/StreamingExample.java deleted file mode 100644 index 82b7fdb..0000000 --- a/server/src/main/java/io/wdd/pusher/example/StreamingExample.java +++ /dev/null @@ -1,27 +0,0 @@ -package io.wdd.pusher.example; - - -import io.wdd.pusher.core.exception.NtfyStreamingException; -import io.wdd.pusher.core.model.StreamRequest; -import io.wdd.pusher.core.stream.StreamingService; - -public class StreamingExample { - public static void main(String[] args) throws NtfyStreamingException { - - StreamRequest request = new StreamRequest(); - request.setHost("ntfy.sh"); - request.setTopic("test_ntfy2"); - - StreamingService streamingConnection = new StreamingService(request); - streamingConnection.setDataListener(data -> { - // Process the incoming data here - System.out.println("Received data: " + data); - }); - - // Start the streaming connection - streamingConnection.start(); - - // Keep the connection open and receive data indefinitely - // To stop the connection, call streamingConnection.stop() - } -} diff --git a/server/src/main/resources/application.yml b/server/src/main/resources/application.yml index 9013606..a4b7a35 100644 --- a/server/src/main/resources/application.yml +++ b/server/src/main/resources/application.yml @@ -96,8 +96,8 @@ octopus: notify: # 消息通知的URL地址 pusher_url: http://192.168.35.71:8080 - # 发送消息的密钥 - access_token: 123456 + # 发送消息的密钥 不需要使用 + access_token: tk_zvdb67fwj1hrjivkq3ga9z7u63av5 message: # agent boot up default common exchange init_exchange: InitExchange diff --git a/server/src/test/java/io/wdd/pusher/PusherTest.java b/server/src/test/java/io/wdd/pusher/PusherTest.java deleted file mode 100644 index 7d6fc99..0000000 --- a/server/src/test/java/io/wdd/pusher/PusherTest.java +++ /dev/null @@ -1,34 +0,0 @@ -package io.wdd.pusher; - -import io.wdd.pusher.core.exception.NtfyException; -import io.wdd.pusher.core.model.ClientType; -import io.wdd.pusher.core.model.NtfyRequest; -import io.wdd.pusher.core.model.PRIORITY; -import io.wdd.pusher.core.publish.PubClient; -import org.junit.Test; -import org.springframework.boot.test.context.SpringBootTest; - -@SpringBootTest -public class PusherTest { - - @Test - public void test() { - PubClient client = new NtfyClient(ClientType.PUB).getClient(); - - NtfyRequest ntfyRequest = new NtfyRequest(); - ntfyRequest.setHost("push.107421.xyz"); - ntfyRequest.setTopic("octopus"); - ntfyRequest.setTitle("来自Java的测试推送"); - ntfyRequest.setPriority(PRIORITY.HIGH); - ntfyRequest.setAccessToken("tk_zvdb67fwj1hrjivkq3ga9z7u63av5"); - ntfyRequest.setMessage("这是一条来自Java的测试推送"); - ntfyRequest.setAuthToken("tk_zvdb67fwj1hrjivkq3ga9z7u63av5"); - - - try { - client.sendNotification(ntfyRequest); - } catch (NtfyException e) { - throw new RuntimeException(e); - } - } -} diff --git a/server/src/test/java/io/wdd/server/func/TestImageSyncScheduler.java b/server/src/test/java/io/wdd/server/func/TestImageSyncScheduler.java index 5c54eb7..ff33bb7 100644 --- a/server/src/test/java/io/wdd/server/func/TestImageSyncScheduler.java +++ b/server/src/test/java/io/wdd/server/func/TestImageSyncScheduler.java @@ -44,7 +44,7 @@ public class TestImageSyncScheduler { ArrayList ImageFullNameList = new ArrayList<>(List.of( // "harbor.cdcyy.com.cn/cmii/cmii-live-operator:5.2.0", // "harbor.cdcyy.com.cn/cmii/cmii/srs:v5.0.195" - "harbor.cdcyy.com.cn/cmii/cmii-uav-industrial-portfolio:5.4.0-cqly-042802" + "harbor.cdcyy.com.cn/cmii/cmii-uav-mission:5.3.0-cqly-042901" )); Boolean downloadAndCompressOnly = false; diff --git a/server/src/test/java/io/wdd/server/pusher/PusherTest.java b/server/src/test/java/io/wdd/server/pusher/PusherTest.java new file mode 100644 index 0000000..e9c46ac --- /dev/null +++ b/server/src/test/java/io/wdd/server/pusher/PusherTest.java @@ -0,0 +1,51 @@ +package io.wdd.server.pusher; + +import com.fasterxml.jackson.databind.ObjectMapper; +import io.wdd.pusher.MessagePusher; +import io.wdd.pusher.beans.*; +import lombok.SneakyThrows; +import org.junit.jupiter.api.Test; +import org.springframework.boot.test.context.SpringBootTest; + +import javax.annotation.Resource; + +@SpringBootTest +public class PusherTest { + + @Resource + MessagePusher messagePusher; + + @Resource + ObjectMapper objectMapper; + + @Test + public void test() { + + NtfySendMessage ntfySendMessage = NtfySendMessage.builder() + .title("来自Java的测试内容") + .priority(1) + .message("Hello World") + .build(); + + CmiiUpdateMessage build = CmiiUpdateMessage.builder().appName("cmii-uav-gateway").namespace("uavcloud-dev").fromTag("1.0.0").toTag("5.0.0").replicas("1").deployStatus(Boolean.TRUE).build(); + + System.out.println("ntfySendMessage = " + ntfySendMessage); + + + NtfyReceiveMessage ntfyReceiveMessage = messagePusher.Send(build, PushMessageType.CMII_UPDATE); + + System.out.println("ntfyReceiveMessage = " + ntfyReceiveMessage); + + } + + @SneakyThrows + @Test + public void testImageSyncMessage() { + + ImageSyncMessage imageSyncMessage = new ImageSyncMessage(); + + String s = objectMapper.writeValueAsString(imageSyncMessage); + + System.out.println("imageSyncMessage = " + s); + } +}